add_intro_banner.py 505 B

12345678910111213
  1. from app import create_app, db
  2. import sqlalchemy
  3. app = create_app()
  4. with app.app_context():
  5. inspector = sqlalchemy.inspect(db.engine)
  6. columns = [col['name'] for col in inspector.get_columns('company_info')]
  7. if 'intro_banner_url' not in columns:
  8. with db.engine.begin() as conn:
  9. conn.execute(sqlalchemy.text('ALTER TABLE company_info ADD COLUMN intro_banner_url VARCHAR(256)'))
  10. print("Added intro_banner_url column")
  11. else:
  12. print("Column already exists")