550 lines
19 KiB
Python
550 lines
19 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
数据库初始化脚本
|
||
KaMiXiTong 软件授权管理系统
|
||
|
||
功能：
|
||
1. 创建数据库表结构
|
||
2. 插入初始数据
|
||
3. 创建默认管理员账号
|
||
4. 插入示例产品和许可证
|
||
|
||
使用方法：
|
||
python init_db.py
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
from datetime import datetime, timedelta
|
||
|
||
# 添加项目根目录到Python路径
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
# 导入应用创建函数（在文件顶部是安全的）
|
||
from app import create_app, db
|
||
|
||
# 注意：模型导入在函数内部进行，确保在应用上下文中
|
||
|
||
|
||
def init_database():
|
||
"""初始化数据库"""
|
||
print("=" * 50)
|
||
print("KaMiXiTong 数据库初始化开始...")
|
||
print("=" * 50)
|
||
|
||
# 创建应用实例
|
||
app = create_app()
|
||
|
||
# 显示数据库配置信息
|
||
print(f"数据库类型: {app.config['SQLALCHEMY_DATABASE_URI'].split('+')[0].split(':')[0]}")
|
||
print(f"数据库连接: {app.config['SQLALCHEMY_DATABASE_URI']}")
|
||
|
||
with app.app_context():
|
||
try:
|
||
# 0. 测试数据库连接
|
||
print("\n0. 测试数据库连接...")
|
||
with db.engine.connect() as connection:
|
||
result = connection.execute(db.text('SELECT 1'))
|
||
print(" ✓ 数据库连接成功")
|
||
|
||
# 1. 删除所有表（如果存在）
|
||
print("\n1. 清理现有数据库表...")
|
||
db.drop_all()
|
||
print(" ✓ 已删除所有现有表")
|
||
|
||
# 2. 创建所有表
|
||
print("\n2. 创建数据库表结构...")
|
||
db.create_all()
|
||
print(" ✓ 已创建所有数据表")
|
||
|
||
# 2.1 验证表创建
|
||
print("\n2.1 验证表创建...")
|
||
from sqlalchemy import inspect
|
||
inspector = inspect(db.engine)
|
||
tables = inspector.get_table_names()
|
||
print(f" ✓ 成功创建 {len(tables)} 个表: {', '.join(tables)}")
|
||
|
||
if not tables:
|
||
raise Exception("表创建失败，没有找到任何表")
|
||
|
||
# 3. 插入初始数据
|
||
print("\n3. 插入初始数据...")
|
||
insert_initial_data()
|
||
|
||
# 4. 显示数据库信息
|
||
print("\n4. 数据库初始化完成！")
|
||
show_database_info()
|
||
|
||
except Exception as e:
|
||
print(f"\n❌ 数据库初始化失败: {e}")
|
||
db.session.rollback()
|
||
raise
|
||
|
||
|
||
def insert_initial_data():
|
||
"""插入初始数据"""
|
||
# 在应用上下文中导入模型
|
||
from app.models import Admin, Product, Version, License, Device, Ticket
|
||
from app.models.license import License
|
||
|
||
# 1. 创建默认管理员账号
|
||
print(" 创建默认管理员账号...")
|
||
admin = Admin(
|
||
username='admin',
|
||
email='admin@kamaxitong.com',
|
||
role=1, # 超级管理员
|
||
status=1 # 正常
|
||
)
|
||
admin.set_password('admin123')
|
||
db.session.add(admin)
|
||
|
||
# 创建测试管理员账号
|
||
test_admin = Admin(
|
||
username='test_admin',
|
||
email='test@kamaxitong.com',
|
||
role=0, # 普通管理员
|
||
status=1 # 正常
|
||
)
|
||
test_admin.set_password('test123')
|
||
db.session.add(test_admin)
|
||
print(" ✓ 已创建管理员账号 (admin/admin123, test_admin/test123)")
|
||
|
||
# 2. 创建示例产品
|
||
print(" 创建示例产品...")
|
||
products = [
|
||
Product(
|
||
product_id='KMX001',
|
||
product_name='KaMiXiTong 专业版',
|
||
description='专业的软件授权管理系统，支持多种授权模式和完整的设备管理',
|
||
status=1
|
||
),
|
||
Product(
|
||
product_id='KMX002',
|
||
product_name='KaMiXiTong 企业版',
|
||
description='企业级软件授权管理解决方案，支持分布式部署和高并发访问',
|
||
status=1
|
||
),
|
||
Product(
|
||
product_id='KMX003',
|
||
product_name='KaMiXiTong 云端版',
|
||
description='云端SaaS版本的软件授权管理系统，无需本地部署',
|
||
status=0 # 禁用状态
|
||
)
|
||
]
|
||
|
||
for product in products:
|
||
db.session.add(product)
|
||
print(" ✓ 已创建3个示例产品")
|
||
|
||
# 3. 创建版本信息
|
||
print(" 创建版本信息...")
|
||
versions = [
|
||
# KaMiXiTong 专业版版本
|
||
Version(
|
||
product_id='KMX001',
|
||
version_num='1.0.0',
|
||
update_log='初始版本发布\n- 基础授权管理功能\n- 设备绑定验证\n- Web管理界面',
|
||
download_url='https://download.kamaxitong.com/v1.0.0/professional.exe',
|
||
min_license_version='1.0.0',
|
||
force_update=0,
|
||
download_status=1,
|
||
publish_status=1
|
||
),
|
||
Version(
|
||
product_id='KMX001',
|
||
version_num='1.1.0',
|
||
update_log='功能增强版本\n- 新增批量许可证生成\n- 优化设备验证算法\n- 修复已知bug',
|
||
download_url='https://download.kamaxitong.com/v1.1.0/professional.exe',
|
||
min_license_version='1.0.0',
|
||
force_update=0,
|
||
download_status=1,
|
||
publish_status=1
|
||
),
|
||
Version(
|
||
product_id='KMX001',
|
||
version_num='1.2.0',
|
||
update_log='重要更新版本\n- 支持云端同步\n- 增强安全性\n- 新增API接口',
|
||
download_url='https://download.kamaxitong.com/v1.2.0/professional.exe',
|
||
min_license_version='1.0.0',
|
||
force_update=1, # 强制更新
|
||
download_status=1,
|
||
publish_status=1
|
||
),
|
||
|
||
# KaMiXiTong 企业版版本
|
||
Version(
|
||
product_id='KMX002',
|
||
version_num='2.0.0',
|
||
update_log='企业版初始版本\n- 支持分布式部署\n- 高并发处理\n- 企业级安全认证',
|
||
download_url='https://download.kamaxitong.com/v2.0.0/enterprise.exe',
|
||
min_license_version='2.0.0',
|
||
force_update=0,
|
||
download_status=1,
|
||
publish_status=1
|
||
),
|
||
Version(
|
||
product_id='KMX002',
|
||
version_num='2.1.0',
|
||
update_log='企业版功能增强\n- 集群部署支持\n- 负载均衡优化\n- 监控面板',
|
||
download_url='https://download.kamaxitong.com/v2.1.0/enterprise.exe',
|
||
min_license_version='2.0.0',
|
||
force_update=0,
|
||
download_status=1,
|
||
publish_status=0 # 草稿状态
|
||
),
|
||
|
||
# KaMiXiTong 云端版版本
|
||
Version(
|
||
product_id='KMX003',
|
||
version_num='3.0.0',
|
||
update_log='云端版Beta版本\n- 完全云端部署\n- 自动备份\n- 弹性扩容',
|
||
download_url='https://cloud.kamaxitong.com/download/v3.0.0/',
|
||
min_license_version='3.0.0',
|
||
force_update=0,
|
||
download_status=0, # 下载不可用
|
||
publish_status=0 # 草稿状态
|
||
)
|
||
]
|
||
|
||
for version in versions:
|
||
db.session.add(version)
|
||
print(" ✓ 已创建6个版本信息")
|
||
|
||
# 4. 创建示例许可证
|
||
print(" 创建示例许可证...")
|
||
current_time = datetime.utcnow()
|
||
|
||
licenses = [
|
||
# 永久许可证
|
||
License(
|
||
license_key=License.generate_license_key(),
|
||
product_id='KMX001',
|
||
type=1, # 正式版
|
||
status=1, # 已激活
|
||
valid_days=-1, # 永久
|
||
bind_machine_code='DEMO-MACHINE-CODE-001',
|
||
activate_time=current_time - timedelta(days=30),
|
||
expire_time=None, # 永久不过期
|
||
last_verify_time=current_time - timedelta(days=1),
|
||
unbind_count=0
|
||
),
|
||
|
||
# 1年试用期许可证
|
||
License(
|
||
license_key=License.generate_license_key(),
|
||
product_id='KMX001',
|
||
type=0, # 试用版
|
||
status=1, # 已激活
|
||
valid_days=365, # 1年
|
||
bind_machine_code='DEMO-MACHINE-CODE-002',
|
||
activate_time=current_time - timedelta(days=15),
|
||
expire_time=current_time + timedelta(days=350), # 还有350天过期
|
||
last_verify_time=current_time - timedelta(hours=6),
|
||
unbind_count=1
|
||
),
|
||
|
||
# 30天试用许可证（即将过期）
|
||
License(
|
||
license_key=License.generate_license_key(),
|
||
product_id='KMX002',
|
||
type=0, # 试用版
|
||
status=1, # 已激活
|
||
valid_days=30, # 30天
|
||
bind_machine_code='DEMO-MACHINE-CODE-003',
|
||
activate_time=current_time - timedelta(days=25),
|
||
expire_time=current_time + timedelta(days=5), # 还有5天过期
|
||
last_verify_time=current_time - timedelta(hours=12),
|
||
unbind_count=2
|
||
),
|
||
|
||
# 未激活许可证
|
||
License(
|
||
license_key=License.generate_license_key(),
|
||
product_id='KMX002',
|
||
type=1, # 正式版
|
||
status=0, # 未激活
|
||
valid_days=365, # 1年
|
||
bind_machine_code=None,
|
||
activate_time=None,
|
||
expire_time=None,
|
||
last_verify_time=None,
|
||
unbind_count=0
|
||
),
|
||
|
||
# 已过期许可证
|
||
License(
|
||
license_key=License.generate_license_key(),
|
||
product_id='KMX001',
|
||
type=0, # 试用版
|
||
status=2, # 已过期
|
||
valid_days=30, # 30天
|
||
bind_machine_code='DEMO-MACHINE-CODE-004',
|
||
activate_time=current_time - timedelta(days=60),
|
||
expire_time=current_time - timedelta(days=30), # 已过期30天
|
||
last_verify_time=current_time - timedelta(days=35),
|
||
unbind_count=0
|
||
),
|
||
|
||
# 已禁用许可证
|
||
License(
|
||
license_key=License.generate_license_key(),
|
||
product_id='KMX003',
|
||
type=1, # 正式版
|
||
status=3, # 已禁用
|
||
valid_days=-1, # 永久
|
||
bind_machine_code='DEMO-MACHINE-CODE-005',
|
||
activate_time=current_time - timedelta(days=45),
|
||
expire_time=None,
|
||
last_verify_time=current_time - timedelta(days=10),
|
||
unbind_count=1
|
||
)
|
||
]
|
||
|
||
for license_obj in licenses:
|
||
db.session.add(license_obj)
|
||
print(" ✓ 已创建6个示例许可证")
|
||
|
||
# 5. 创建示例设备
|
||
print(" 创建示例设备...")
|
||
devices = [
|
||
Device(
|
||
machine_code='DEMO-MACHINE-CODE-001',
|
||
license_id=1, # 对应第一个许可证
|
||
product_id='KMX001',
|
||
software_version='1.2.0',
|
||
status=1, # 正常
|
||
activate_time=current_time - timedelta(days=30),
|
||
last_verify_time=current_time - timedelta(days=1)
|
||
),
|
||
Device(
|
||
machine_code='DEMO-MACHINE-CODE-002',
|
||
license_id=2, # 对应第二个许可证
|
||
product_id='KMX001',
|
||
software_version='1.1.0',
|
||
status=1, # 正常
|
||
activate_time=current_time - timedelta(days=15),
|
||
last_verify_time=current_time - timedelta(hours=6)
|
||
),
|
||
Device(
|
||
machine_code='DEMO-MACHINE-CODE-003',
|
||
license_id=3, # 对应第三个许可证
|
||
product_id='KMX002',
|
||
software_version='2.0.0',
|
||
status=1, # 正常
|
||
activate_time=current_time - timedelta(days=25),
|
||
last_verify_time=current_time - timedelta(hours=12)
|
||
),
|
||
Device(
|
||
machine_code='DEMO-MACHINE-CODE-004',
|
||
license_id=5, # 对应第五个许可证（已过期）
|
||
product_id='KMX001',
|
||
software_version='1.0.0',
|
||
status=0, # 禁用
|
||
activate_time=current_time - timedelta(days=60),
|
||
last_verify_time=current_time - timedelta(days=35)
|
||
),
|
||
Device(
|
||
machine_code='DEMO-MACHINE-CODE-005',
|
||
license_id=6, # 对应第六个许可证（已禁用）
|
||
product_id='KMX003',
|
||
software_version='3.0.0-beta',
|
||
status=0, # 禁用
|
||
activate_time=current_time - timedelta(days=45),
|
||
last_verify_time=current_time - timedelta(days=10)
|
||
),
|
||
|
||
# 未绑定的设备
|
||
Device(
|
||
machine_code='DEMO-MACHINE-CODE-UNBOUND',
|
||
license_id=None, # 未绑定许可证
|
||
product_id='KMX001',
|
||
software_version='1.2.0',
|
||
status=1, # 正常
|
||
activate_time=None,
|
||
last_verify_time=None
|
||
)
|
||
]
|
||
|
||
for device in devices:
|
||
db.session.add(device)
|
||
print(" ✓ 已创建6个示例设备")
|
||
|
||
# 6. 创建示例工单
|
||
print(" 创建示例工单...")
|
||
tickets = [
|
||
Ticket(
|
||
title='许可证激活失败',
|
||
product_id='KMX001',
|
||
software_version='1.1.0',
|
||
machine_code='CUSTOMER-MACHINE-001',
|
||
license_key='DEMO-LICENSE-KEY-001',
|
||
description='客户反馈在Windows Server 2019系统上无法激活许可证，提示"许可证验证失败"错误。已尝试重新安装软件和清理注册表，问题仍然存在。',
|
||
priority=2, # 高优先级
|
||
status=2, # 已解决
|
||
operator='admin',
|
||
remark='问题已解决，原因是系统时间不正确导致许可证验证失败。指导客户同步系统时间后问题解决。',
|
||
create_time=current_time - timedelta(days=5),
|
||
update_time=current_time - timedelta(days=4),
|
||
resolve_time=current_time - timedelta(days=4)
|
||
),
|
||
Ticket(
|
||
title='设备解绑次数限制问题',
|
||
product_id='KMX002',
|
||
software_version='2.0.0',
|
||
machine_code='CUSTOMER-MACHINE-002',
|
||
license_key='DEMO-LICENSE-KEY-002',
|
||
description='客户需要更换硬件，但许可证解绑次数已达到上限。希望能够重置解绑次数或提供新的许可证。',
|
||
priority=1, # 中优先级
|
||
status=1, # 处理中
|
||
operator='test_admin',
|
||
remark='正在核实客户身份和购买记录，预计今天内处理完成。',
|
||
create_time=current_time - timedelta(days=2),
|
||
update_time=current_time - timedelta(days=1)
|
||
),
|
||
Ticket(
|
||
title='软件更新后功能异常',
|
||
product_id='KMX001',
|
||
software_version='1.2.0',
|
||
machine_code='CUSTOMER-MACHINE-003',
|
||
license_key='DEMO-LICENSE-KEY-003',
|
||
description='更新到1.2.0版本后，许可证验证功能出现异常，经常性提示连接服务器失败。网络连接正常，防火墙已关闭。',
|
||
priority=2, # 高优先级
|
||
status=0, # 待处理
|
||
operator=None,
|
||
remark=None,
|
||
create_time=current_time - timedelta(hours=6),
|
||
update_time=current_time - timedelta(hours=6)
|
||
),
|
||
Ticket(
|
||
title='批量许可证申请',
|
||
product_id='KMX002',
|
||
software_version='2.0.0',
|
||
machine_code=None,
|
||
license_key=None,
|
||
description='我公司需要采购100个企业版许可证，用于公司内部项目管理。希望能够提供批量购买优惠和技术支持。',
|
||
priority=1, # 中优先级
|
||
status=0, # 待处理
|
||
operator=None,
|
||
remark=None,
|
||
create_time=current_time - timedelta(hours=12),
|
||
update_time=current_time - timedelta(hours=12)
|
||
),
|
||
Ticket(
|
||
title='API接口使用咨询',
|
||
product_id='KMX001',
|
||
software_version='1.1.0',
|
||
machine_code='CUSTOMER-MACHINE-004',
|
||
license_key='DEMO-LICENSE-KEY-004',
|
||
description='希望能够了解如何使用API接口进行许可证验证，需要相关的技术文档和示例代码。',
|
||
priority=0, # 低优先级
|
||
status=3, # 已关闭
|
||
operator='admin',
|
||
remark='已发送API文档和示例代码给客户，客户确认问题已解决。',
|
||
create_time=current_time - timedelta(days=10),
|
||
update_time=current_time - timedelta(days=8),
|
||
resolve_time=current_time - timedelta(days=8),
|
||
close_time=current_time - timedelta(days=8)
|
||
)
|
||
]
|
||
|
||
for ticket in tickets:
|
||
db.session.add(ticket)
|
||
print(" ✓ 已创建5个示例工单")
|
||
|
||
# 提交所有更改
|
||
db.session.commit()
|
||
print(" ✓ 所有初始数据插入完成")
|
||
|
||
|
||
def show_database_info():
|
||
"""显示数据库初始化后的信息"""
|
||
# 在应用上下文中导入模型
|
||
from app.models import Admin, Product, Version, License, Device, Ticket
|
||
|
||
print("\n" + "=" * 50)
|
||
print("数据库初始化信息统计")
|
||
print("=" * 50)
|
||
|
||
# 统计各表的记录数
|
||
tables_info = [
|
||
('管理员账号', Admin.query.count()),
|
||
('产品', Product.query.count()),
|
||
('版本', Version.query.count()),
|
||
('许可证', License.query.count()),
|
||
('设备', Device.query.count()),
|
||
('工单', Ticket.query.count())
|
||
]
|
||
|
||
for table_name, count in tables_info:
|
||
print(f" {table_name:8} : {count:3} 条记录")
|
||
|
||
print("\n" + "=" * 50)
|
||
print("默认登录信息")
|
||
print("=" * 50)
|
||
print(" 超级管理员: admin / admin123")
|
||
print(" 普通管理员: test_admin / test123")
|
||
print("\n注意事项:")
|
||
print(" 1. 请在生产环境中修改默认密码")
|
||
print(" 2. 根据实际需求配置数据库连接")
|
||
print(" 3. 定期备份数据库数据")
|
||
print("=" * 50)
|
||
|
||
|
||
def reset_database():
|
||
"""重置数据库（仅删除数据，保留表结构）"""
|
||
print("警告：此操作将删除所有数据，但保留表结构")
|
||
confirm = input("确认重置数据库？(y/N): ")
|
||
|
||
if confirm.lower() != 'y':
|
||
print("操作已取消")
|
||
return
|
||
|
||
app = create_app()
|
||
|
||
with app.app_context():
|
||
# 在应用上下文中导入模型
|
||
from app.models import Ticket, Device, License, Version, Product, Admin
|
||
|
||
try:
|
||
# 删除所有数据
|
||
db.session.query(Ticket).delete()
|
||
db.session.query(Device).delete()
|
||
db.session.query(License).delete()
|
||
db.session.query(Version).delete()
|
||
db.session.query(Product).delete()
|
||
db.session.query(Admin).delete()
|
||
|
||
db.session.commit()
|
||
print("✓ 数据库重置完成")
|
||
|
||
# 重新插入初始数据
|
||
insert_initial_data()
|
||
print("✓ 初始数据重新插入完成")
|
||
|
||
except Exception as e:
|
||
print(f"❌ 数据库重置失败: {e}")
|
||
db.session.rollback()
|
||
raise
|
||
|
||
|
||
def main():
|
||
"""主函数"""
|
||
if len(sys.argv) > 1:
|
||
command = sys.argv[1].lower()
|
||
|
||
if command == 'reset':
|
||
reset_database()
|
||
elif command == 'init':
|
||
init_database()
|
||
else:
|
||
print("用法:")
|
||
print(" python init_db.py init - 初始化数据库（删除并重建）")
|
||
print(" python init_db.py reset - 重置数据库（仅删除数据）")
|
||
else:
|
||
# 默认执行初始化
|
||
init_database()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main() |