42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
|
import pymysql
|
|||
|
|
|
|||
|
|
# 数据库连接信息
|
|||
|
|
host = 'localhost'
|
|||
|
|
user = 'root'
|
|||
|
|
password = 'taiyi1224'
|
|||
|
|
database = 'kamaxitong'
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 连接数据库
|
|||
|
|
connection = pymysql.connect(
|
|||
|
|
host=host,
|
|||
|
|
user=user,
|
|||
|
|
password=password,
|
|||
|
|
database=database,
|
|||
|
|
charset='utf8mb4'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
with connection.cursor() as cursor:
|
|||
|
|
# 先检查当前版本
|
|||
|
|
cursor.execute("SELECT * FROM alembic_version")
|
|||
|
|
current_version = cursor.fetchone()
|
|||
|
|
print(f"Current alembic version: {current_version}")
|
|||
|
|
|
|||
|
|
# 更新alembic_version到最新版本（使用较短的版本号）
|
|||
|
|
cursor.execute("UPDATE alembic_version SET version_num = %s WHERE version_num = %s", ("20241119", current_version[0]))
|
|||
|
|
print("Updated alembic_version to 20241119")
|
|||
|
|
|
|||
|
|
connection.commit()
|
|||
|
|
|
|||
|
|
# 验证alembic版本
|
|||
|
|
cursor.execute("SELECT * FROM alembic_version")
|
|||
|
|
version_result = cursor.fetchone()
|
|||
|
|
print(f"New alembic version: {version_result}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"Error: {e}")
|
|||
|
|
if 'connection' in locals():
|
|||
|
|
connection.rollback()
|
|||
|
|
finally:
|
|||
|
|
if 'connection' in locals():
|
|||
|
|
connection.close()
|