From b5526d6e2f54fe7383d950ba09ed6d0cb75c9536 Mon Sep 17 00:00:00 2001 From: jessikitty Date: Wed, 28 Jan 2026 17:22:23 +1100 Subject: [PATCH] Add migration script for discord_id and profile_picture columns --- backend/migrations/add_user_fields.py | 37 +++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 backend/migrations/add_user_fields.py diff --git a/backend/migrations/add_user_fields.py b/backend/migrations/add_user_fields.py new file mode 100644 index 0000000..376abdd --- /dev/null +++ b/backend/migrations/add_user_fields.py @@ -0,0 +1,37 @@ +"""Add discord_id and profile_picture to users table.""" +import sqlite3 +from pathlib import Path + +def migrate(): + """Add new columns to users table.""" + db_path = Path("/app/data/familyhub.db") + + conn = sqlite3.connect(db_path) + cursor = conn.cursor() + + try: + # Add discord_id column + cursor.execute("ALTER TABLE users ADD COLUMN discord_id VARCHAR(100)") + print("✓ Added discord_id column") + except sqlite3.OperationalError as e: + if "duplicate column name" in str(e): + print("✓ discord_id column already exists") + else: + raise + + try: + # Add profile_picture column + cursor.execute("ALTER TABLE users ADD COLUMN profile_picture VARCHAR(500)") + print("✓ Added profile_picture column") + except sqlite3.OperationalError as e: + if "duplicate column name" in str(e): + print("✓ profile_picture column already exists") + else: + raise + + conn.commit() + conn.close() + print("✓ Migration complete!") + +if __name__ == "__main__": + migrate()