Initial commit

xuexiao/scripts/test_alter.py

@@ -0,0 +1,23 @@
+import sys
+from pathlib import Path
+sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
+
+from app import create_app
+from models import db, Course
+from utils import _migrate_col_type, is_mysql
+from db_bootstrap import is_doris
+
+app = create_app()
+with app.app_context():
+    print('is_mysql', is_mysql(), 'is_doris', is_doris())
+    col = Course.__table__.columns['course_code']
+    col_type = _migrate_col_type(col)
+    sql = f"ALTER TABLE `courses` ADD COLUMN `course_code` {col_type} DEFAULT ''"
+    print('SQL:', sql)
+    try:
+        db.session.execute(db.text(sql))
+        db.session.commit()
+        print('SUCCESS')
+    except Exception as e:
+        db.session.rollback()
+        print('FAIL:', e)