BakoAI / run_migration.py
icanedit2's picture
Deploy backend to HF with player video visibility fix
32bc095
Raw
History Blame Contribute Delete
743 Bytes
import os
import psycopg2
from dotenv import load_dotenv
load_dotenv()
def run_migration():
db_url = os.getenv("DATABASE_URL")
if not db_url:
print("DATABASE_URL not found in environment")
return
try:
conn = psycopg2.connect(db_url)
conn.autocommit = True
cur = conn.cursor()
print("Running migration from migrations/007_add_detailed_stats.sql...")
with open("migrations/007_add_detailed_stats.sql", "r") as f:
sql = f.read()
cur.execute(sql)
print("Migration successful!")
cur.close()
conn.close()
except Exception as e:
print(f"Migration failed: {e}")
if __name__ == "__main__":
run_migration()