#!/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()