Spaces:
Sleeping
Sleeping
File size: 1,427 Bytes
4a693cf | 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 32 33 34 35 36 37 38 39 40 | import asyncio
from sqlalchemy import text
from app.db.session import engine
async def run_migration():
async with engine.begin() as conn:
print("Checking database columns...")
# Check if remarks column exists
result = await conn.execute(text("""
SELECT column_name
FROM information_schema.columns
WHERE table_name='study_tasks' AND column_name='remarks';
"""))
exists = result.fetchone()
if not exists:
print("Adding 'remarks' column to 'study_tasks'...")
await conn.execute(text("ALTER TABLE study_tasks ADD COLUMN remarks TEXT;"))
print("Column 'remarks' added.")
else:
print("Column 'remarks' already exists.")
# Also check for is_reminder
result = await conn.execute(text("""
SELECT column_name
FROM information_schema.columns
WHERE table_name='study_tasks' AND column_name='is_reminder';
"""))
exists = result.fetchone()
if not exists:
print("Adding 'is_reminder' column to 'study_tasks'...")
await conn.execute(text("ALTER TABLE study_tasks ADD COLUMN is_reminder BOOLEAN DEFAULT FALSE;"))
print("Column 'is_reminder' added.")
else:
print("Column 'is_reminder' already exists.")
if __name__ == "__main__":
asyncio.run(run_migration())
|