File size: 743 Bytes
32bc095 | 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 | 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()
|