60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
import sqlite3
|
|
|
|
# Database path
|
|
db_path = r'c:\Users\diksha rawat\Desktop\ALwrity_github\windsurf\ALwrity\workspace\workspace_user_33Gz1FPI86VDXhRY8QN4ragRFGN\db\alwrity_user_33Gz1FPI86VDXhRY8QN4ragRFGN.db'
|
|
|
|
print(f"Running WaveSpeed migration on: {db_path}")
|
|
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
try:
|
|
# Check current columns
|
|
cursor.execute('PRAGMA table_info(usage_summaries)')
|
|
columns = [col[1] for col in cursor.fetchall()]
|
|
|
|
print(f"Current columns with 'wavespeed': {[col for col in columns if 'wavespeed' in col]}")
|
|
|
|
# Add wavespeed_calls if missing
|
|
if 'wavespeed_calls' not in columns:
|
|
print("Adding wavespeed_calls...")
|
|
cursor.execute('ALTER TABLE usage_summaries ADD COLUMN wavespeed_calls INTEGER DEFAULT 0')
|
|
print("✅ wavespeed_calls added")
|
|
else:
|
|
print("wavespeed_calls already exists")
|
|
|
|
# Add wavespeed_tokens if missing
|
|
if 'wavespeed_tokens' not in columns:
|
|
print("Adding wavespeed_tokens...")
|
|
cursor.execute('ALTER TABLE usage_summaries ADD COLUMN wavespeed_tokens INTEGER DEFAULT 0')
|
|
print("✅ wavespeed_tokens added")
|
|
else:
|
|
print("wavespeed_tokens already exists")
|
|
|
|
# Add wavespeed_cost if missing
|
|
if 'wavespeed_cost' not in columns:
|
|
print("Adding wavespeed_cost...")
|
|
cursor.execute('ALTER TABLE usage_summaries ADD COLUMN wavespeed_cost REAL DEFAULT 0.0')
|
|
print("✅ wavespeed_cost added")
|
|
else:
|
|
print("wavespeed_cost already exists")
|
|
|
|
conn.commit()
|
|
|
|
# Verify
|
|
cursor.execute('PRAGMA table_info(usage_summaries)')
|
|
updated_columns = [col[1] for col in cursor.fetchall()]
|
|
wavespeed_cols = [col for col in updated_columns if 'wavespeed' in col]
|
|
|
|
print(f"\n✅ Migration completed!")
|
|
print(f"WaveSpeed columns now available: {wavespeed_cols}")
|
|
|
|
except Exception as e:
|
|
print(f"❌ Error: {e}")
|
|
conn.rollback()
|
|
finally:
|
|
conn.close()
|
|
|
|
print("\n🎉 WaveSpeed migration completed successfully!")
|
|
print("The subscription dashboard should now work without errors.")
|