47 lines
1.4 KiB
Python
47 lines
1.4 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:
|
|||
|
|
# 添加ip_address字段到device表
|
|||
|
|
cursor.execute("ALTER TABLE device ADD COLUMN ip_address VARCHAR(45) NULL")
|
|||
|
|
print("Added ip_address column to device table")
|
|||
|
|
|
|||
|
|
# 更新alembic_version到最新版本（使用较短的版本号）
|
|||
|
|
cursor.execute("UPDATE alembic_version SET version_num = %s", ("20241119",))
|
|||
|
|
print("Updated alembic_version to 20241119")
|
|||
|
|
|
|||
|
|
connection.commit()
|
|||
|
|
|
|||
|
|
# 验证字段已添加
|
|||
|
|
cursor.execute("DESCRIBE device")
|
|||
|
|
device_columns = cursor.fetchall()
|
|||
|
|
has_ip_address = any(col[0] == 'ip_address' for col in device_columns)
|
|||
|
|
print(f"Device table has ip_address column: {has_ip_address}")
|
|||
|
|
|
|||
|
|
# 验证alembic版本
|
|||
|
|
cursor.execute("SELECT * FROM alembic_version")
|
|||
|
|
version_result = cursor.fetchone()
|
|||
|
|
print(f"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()
|