53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Check if WaveSpeed migration is needed for user database
|
|
"""
|
|
|
|
import sqlite3
|
|
import os
|
|
|
|
# Database path from error logs
|
|
db_path = r'c:\Users\diksha rawat\Desktop\ALwrity_github\windsurf\ALwrity\workspace\workspace_user_33Gz1FPI86VDXhRY8QN4ragRFGN\db\alwrity_user_33Gz1FPI86VDXhRY8QN4ragRFGN.db'
|
|
|
|
print(f"Checking database: {db_path}")
|
|
print(f"Database exists: {os.path.exists(db_path)}")
|
|
|
|
if os.path.exists(db_path):
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
try:
|
|
# Check if usage_summaries table exists
|
|
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='usage_summaries'")
|
|
table_exists = cursor.fetchone()
|
|
|
|
if table_exists:
|
|
print("✅ usage_summaries table found")
|
|
|
|
# Check current columns
|
|
cursor.execute('PRAGMA table_info(usage_summaries)')
|
|
columns = [col[1] for col in cursor.fetchall()]
|
|
|
|
wavespeed_cols = [col for col in columns if 'wavespeed' in col]
|
|
print(f"Current WaveSpeed columns: {wavespeed_cols}")
|
|
|
|
if not wavespeed_cols:
|
|
print("\n❌ WaveSpeed columns are MISSING!")
|
|
print("\nTo fix this, run these SQL commands:")
|
|
print(f"sqlite3 \"{db_path}\"")
|
|
print("ALTER TABLE usage_summaries ADD COLUMN wavespeed_calls INTEGER DEFAULT 0;")
|
|
print("ALTER TABLE usage_summaries ADD COLUMN wavespeed_tokens INTEGER DEFAULT 0;")
|
|
print("ALTER TABLE usage_summaries ADD COLUMN wavespeed_cost REAL DEFAULT 0.0;")
|
|
print(".quit")
|
|
else:
|
|
print("✅ WaveSpeed columns already exist!")
|
|
else:
|
|
print("❌ usage_summaries table not found")
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
finally:
|
|
conn.close()
|
|
else:
|
|
print("❌ Database file not found")
|