135 lines
5.9 KiB
Python
135 lines
5.9 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
修复license表结构
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
import sqlite3
|
||
|
||
# 添加项目路径
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
from app import create_app
|
||
|
||
def fix_license_table():
|
||
"""修复license表结构"""
|
||
app = create_app()
|
||
|
||
with app.app_context():
|
||
try:
|
||
# 获取数据库路径
|
||
db_uri = app.config['SQLALCHEMY_DATABASE_URI']
|
||
if db_uri.startswith('sqlite:///'):
|
||
db_path = db_uri[10:] # 移除 'sqlite:///' 前缀
|
||
print(f"数据库路径: {db_path}")
|
||
|
||
# 连接到SQLite数据库
|
||
conn = sqlite3.connect(db_path)
|
||
cursor = conn.cursor()
|
||
|
||
# 列出所有表
|
||
cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
|
||
tables = cursor.fetchall()
|
||
print("=== 数据库中的所有表 ===")
|
||
for table in tables:
|
||
print(f"表名: {table[0]}")
|
||
|
||
# 检查是否有license表
|
||
if any(table[0] == 'license' for table in tables):
|
||
print("\n找到license表，检查表结构...")
|
||
|
||
# 检查当前表结构
|
||
cursor.execute("PRAGMA table_info(license)")
|
||
columns = cursor.fetchall()
|
||
|
||
print("=== 当前License表结构 ===")
|
||
for column in columns:
|
||
print(f"列名: {column[1]}, 类型: {column[2]}, 是否可为空: {column[3]}, 默认值: {column[4]}, 是否为主键: {column[5]}")
|
||
|
||
# 检查license_key列
|
||
license_key_info = None
|
||
for column in columns:
|
||
if column[1] == 'license_key':
|
||
license_key_info = column
|
||
break
|
||
|
||
if license_key_info:
|
||
current_type = license_key_info[2]
|
||
print(f"\n当前license_key列类型: {current_type}")
|
||
|
||
if "VARCHAR(32)" in current_type.upper():
|
||
print("需要更新license_key列类型为VARCHAR(35)")
|
||
|
||
# SQLite修改表结构的方法：
|
||
# 1. 创建新表
|
||
# 2. 复制数据
|
||
# 3. 删除旧表
|
||
# 4. 重命名新表
|
||
|
||
# 开始事务
|
||
cursor.execute("BEGIN TRANSACTION")
|
||
|
||
# 1. 创建新表（修改license_key列长度）
|
||
cursor.execute("""
|
||
CREATE TABLE license_new (
|
||
license_id INTEGER NOT NULL PRIMARY KEY,
|
||
license_key VARCHAR(35) NOT NULL,
|
||
product_id VARCHAR(32) NOT NULL,
|
||
type INTEGER NOT NULL,
|
||
status INTEGER NOT NULL,
|
||
valid_days INTEGER NOT NULL,
|
||
bind_machine_code VARCHAR(64),
|
||
activate_time DATETIME,
|
||
expire_time DATETIME,
|
||
last_verify_time DATETIME,
|
||
unbind_count INTEGER,
|
||
create_time DATETIME,
|
||
update_time DATETIME
|
||
)
|
||
""")
|
||
|
||
# 2. 复制数据
|
||
cursor.execute("""
|
||
INSERT INTO license_new
|
||
SELECT license_id, license_key, product_id, type, status, valid_days,
|
||
bind_machine_code, activate_time, expire_time, last_verify_time,
|
||
unbind_count, create_time, update_time
|
||
FROM license
|
||
""")
|
||
|
||
# 3. 删除旧表
|
||
cursor.execute("DROP TABLE license")
|
||
|
||
# 4. 重命名新表
|
||
cursor.execute("ALTER TABLE license_new RENAME TO license")
|
||
|
||
# 5. 重新创建索引
|
||
cursor.execute("CREATE UNIQUE INDEX ix_license_license_key ON license (license_key)")
|
||
cursor.execute("CREATE INDEX ix_license_product_id ON license (product_id)")
|
||
|
||
# 提交事务
|
||
conn.commit()
|
||
|
||
print("✓ 成功更新license_key列类型为VARCHAR(35)")
|
||
|
||
elif "VARCHAR(35)" in current_type.upper():
|
||
print("✓ license_key列已经是正确的长度")
|
||
else:
|
||
print(f"⚠ license_key列类型未知: {current_type}")
|
||
else:
|
||
print("✗ 未找到license_key列!")
|
||
else:
|
||
print("✗ 未找到license表!")
|
||
|
||
# 关闭连接
|
||
conn.close()
|
||
|
||
except Exception as e:
|
||
print(f"修复数据库结构时出错: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
|
||
if __name__ == "__main__":
|
||
fix_license_table() |