28 lines
917 B
Python
28 lines
917 B
Python
"""
|
||
Migration: Add monthly_budget column to users table
|
||
Run this with: python migrations/add_monthly_budget.py
|
||
"""
|
||
|
||
from app import create_app, db
|
||
|
||
def migrate():
|
||
app = create_app()
|
||
with app.app_context():
|
||
try:
|
||
# Check if column exists
|
||
from sqlalchemy import inspect
|
||
inspector = inspect(db.engine)
|
||
columns = [col['name'] for col in inspector.get_columns('users')]
|
||
|
||
if 'monthly_budget' not in columns:
|
||
db.engine.execute('ALTER TABLE users ADD COLUMN monthly_budget FLOAT DEFAULT 0.0')
|
||
print("✅ Successfully added monthly_budget column to users table")
|
||
else:
|
||
print("ℹ️ Column monthly_budget already exists")
|
||
|
||
except Exception as e:
|
||
print(f"❌ Migration failed: {e}")
|
||
raise
|
||
|
||
if __name__ == '__main__':
|
||
migrate()
|