run_visibility_migration.py 481 B

123456789101112131415161718
  1. """执行visibility字段迁移"""
  2. import sys
  3. sys.path.insert(0, '.')
  4. from sqlalchemy import text
  5. from app.database import engine
  6. def run_migration():
  7. with engine.connect() as conn:
  8. conn.execute(text("""
  9. ALTER TABLE aigcspace.models
  10. ADD COLUMN IF NOT EXISTS visibility VARCHAR(20) DEFAULT 'user'
  11. """))
  12. conn.commit()
  13. print("Migration completed: visibility column added")
  14. if __name__ == "__main__":
  15. run_migration()