File size: 1,010 Bytes
81e8bb6 | 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 |
import sqlite3
import os
db_path = "backend/atom_dev.db"
if not os.path.exists(db_path):
print("DB not found at backend/atom_dev.db, trying atom_dev.db")
db_path = "atom_dev.db"
conn = sqlite3.connect(db_path)
cursor = conn.cursor()
def add_column_if_not_exists(table, col_name, col_type):
try:
cursor.execute(f"ALTER TABLE {table} ADD COLUMN {col_name} {col_type}")
print(f"Added {col_name}")
except sqlite3.OperationalError as e:
if "duplicate column name" in str(e):
print(f"Column {col_name} already exists")
else:
print(f"Error adding {col_name}: {e}")
print(f"Migrating {db_path}...")
add_column_if_not_exists("users", "skills", "TEXT")
add_column_if_not_exists("users", "capacity_hours", "FLOAT DEFAULT 40.0")
add_column_if_not_exists("users", "hourly_cost_rate", "FLOAT DEFAULT 0.0")
add_column_if_not_exists("users", "metadata_json", "TEXT") # SQLite uses TEXT for JSON
conn.commit()
conn.close()
print("Migration done.")
|