ai-LiaoWangweb-app/migrate_add_is_default.py

"""Add is_default column to ai_model table if not exists."""
from app import create_app, db
from sqlalchemy import text, inspect


def add_is_default_column():
    app = create_app()
    with app.app_context():
        inspector = inspect(db.engine)
        columns = [col['name'] for col in inspector.get_columns('ai_model')]

        if 'is_default' not in columns:
            print("Adding 'is_default' column to ai_model table...")
            db.session.execute(
                text("ALTER TABLE ai_model ADD COLUMN is_default BOOLEAN DEFAULT FALSE")
            )
            db.session.commit()
            print("Done. Column added successfully.")
        else:
            print("Column 'is_default' already exists. No action needed.")


if __name__ == '__main__':
    add_is_default_column()