51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
import sqlite3
|
|
import os
|
|
|
|
# Find database
|
|
db_paths = ["backend/database.db", "backend/data/database.db", "database.db", "data/database.db"]
|
|
db_path = None
|
|
|
|
for path in db_paths:
|
|
if os.path.exists(path):
|
|
db_path = path
|
|
break
|
|
|
|
if not db_path:
|
|
print("Database not found!")
|
|
exit(1)
|
|
|
|
print(f"Using database: {db_path}")
|
|
|
|
try:
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
# Check table
|
|
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='usage_summaries'")
|
|
if not cursor.fetchone():
|
|
print("Table usage_summaries not found!")
|
|
exit(1)
|
|
|
|
# Get columns
|
|
cursor.execute("PRAGMA table_info(usage_summaries)")
|
|
columns = [row[1] for row in cursor.fetchall()]
|
|
print(f"Columns: {columns}")
|
|
|
|
# Check for wavespeed_calls
|
|
if "wavespeed_calls" in columns:
|
|
print("✅ wavespeed_calls column exists")
|
|
else:
|
|
print("❌ wavespeed_calls column missing")
|
|
|
|
# Add the column
|
|
print("Adding wavespeed_calls column...")
|
|
cursor.execute("ALTER TABLE usage_summaries ADD COLUMN wavespeed_calls INTEGER DEFAULT 0")
|
|
conn.commit()
|
|
print("✅ wavespeed_calls column added")
|
|
|
|
conn.close()
|
|
print("Migration completed!")
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|