File size: 1,026 Bytes
5e90518 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import sqlite3
def init_db():
conn = sqlite3.connect("curato.db")
cursor = conn.cursor()
# Create table if it doesn't exist
cursor.execute("""
CREATE TABLE IF NOT EXISTS artworks (
filename TEXT PRIMARY KEY,
style TEXT,
tags TEXT,
caption TEXT
)
""")
# Add cloud_url column if not exists
cursor.execute("PRAGMA table_info(artworks)")
columns = [col[1] for col in cursor.fetchall()]
if 'cloud_url' not in columns:
cursor.execute("ALTER TABLE artworks ADD COLUMN cloud_url TEXT")
conn.commit()
conn.close()
def save_metadata(filename, style, tags, caption, cloud_url):
conn = sqlite3.connect("curato.db")
cursor = conn.cursor()
cursor.execute("""
INSERT OR REPLACE INTO artworks (filename, style, tags, caption, cloud_url)
VALUES (?, ?, ?, ?, ?)
""", (filename, style, ",".join(tags), caption, cloud_url))
conn.commit()
conn.close()
|