24 lines
742 B
Python
24 lines
742 B
Python
|
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)
|