import sqlite3, os

BASE_DIR = os.path.dirname(os.path.abspath(__file__))
DB_PATH  = os.path.join(BASE_DIR, 'memorial.db')

conn = sqlite3.connect(DB_PATH)

conn.execute('''
    CREATE TABLE IF NOT EXISTS gallery (
        id          INTEGER PRIMARY KEY AUTOINCREMENT,
        filename    TEXT NOT NULL,
        caption     TEXT,
        uploader    TEXT,
        uploaded_at TEXT NOT NULL
    )
''')

conn.execute('''
    CREATE TABLE IF NOT EXISTS admin_sessions (
        token      TEXT PRIMARY KEY,
        expires_at TEXT NOT NULL
    )
''')

# Add uploader column if upgrading old gallery table
try:
    conn.execute('ALTER TABLE gallery ADD COLUMN uploader TEXT')
except Exception:
    pass

conn.commit()
conn.close()
print('All tables created successfully')