可以给我一个sqlite存储照片的样例嘛
时间: 2024-04-29 21:23:51 浏览: 14
以下是一个使用SQLite存储照片的Python示例:
```python
import sqlite3
from sqlite3 import Error
import base64
def create_connection():
conn = None;
try:
conn = sqlite3.connect('photos.db')
print(f"Connected to SQLite version {sqlite3.version}")
except Error as e:
print(e)
return conn
def create_table(conn):
try:
cursor = conn.cursor()
cursor.execute('''CREATE TABLE IF NOT EXISTS photos
(id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
data BLOB NOT NULL);''')
conn.commit()
print("Table created successfully")
except Error as e:
print(e)
def insert_photo(conn, name, data):
try:
cursor = conn.cursor()
cursor.execute("INSERT INTO photos (name, data) VALUES (?, ?)", (name, data))
conn.commit()
print("Photo inserted successfully")
except Error as e:
print(e)
def retrieve_photo(conn, id):
try:
cursor = conn.cursor()
cursor.execute("SELECT * FROM photos WHERE id=?", (id,))
row = cursor.fetchone()
if row:
name = row[1]
data = row[2]
with open(name, "wb") as f:
f.write(base64.b64decode(data))
print("Photo retrieved successfully")
else:
print("Photo not found")
except Error as e:
print(e)
def main():
conn = create_connection()
create_table(conn)
with open("example.jpg", "rb") as f:
data = base64.b64encode(f.read())
insert_photo(conn, "example.jpg", data)
retrieve_photo(conn, 1)
if __name__ == '__main__':
main()
```
解释:
1. `create_connection()`函数创建一个连接到SQLite数据库的对象。如果连接成功,函数将返回一个连接对象。
2. `create_table(conn)`函数创建一个名为“photos”的表,其中包含“id”(主键)、“name”和“data”列。其中,“name”列用于存储照片的文件名,“data”列用于存储照片的二进制数据。
3. `insert_photo(conn, name, data)`函数将照片存储在数据库中。它接受两个参数:照片的文件名和二进制数据。
4. `retrieve_photo(conn, id)`函数从数据库中检索照片。它接受一个参数:照片的ID。如果找到该照片,则将其保存在文件系统中。
5. `main()`函数创建一个连接对象,创建表并插入一张照片。然后检索该照片并将其保存在文件系统中。
注意:在实际的应用程序中,您可能需要更复杂的查询来检索和管理照片。此示例仅用于演示如何使用SQLite存储和检索照片。