349 lines
11 KiB
Python
349 lines
11 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
SQLite 数据库初始化脚本
|
||
KaMiXiTong 软件授权管理系统
|
||
|
||
使用 SQLite 数据库的简化版本，避免 MySQL 配置问题
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
from datetime import datetime, timedelta
|
||
from werkzeug.security import generate_password_hash
|
||
|
||
# 添加项目根目录到Python路径
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
def init_sqlite_database():
|
||
"""初始化 SQLite 数据库"""
|
||
print("=" * 50)
|
||
print("KaMiXiTong SQLite 数据库初始化开始...")
|
||
print("=" * 50)
|
||
|
||
# 临时设置环境变量为 SQLite
|
||
os.environ['DATABASE_URL'] = 'sqlite:///kamaxitong.db'
|
||
|
||
try:
|
||
# 导入应用（在设置环境变量之后）
|
||
from app import create_app, db
|
||
from app.models import Admin, Product, Version, License, Device, Ticket
|
||
from app.models.license import License
|
||
|
||
# 创建应用实例
|
||
app = create_app()
|
||
|
||
print(f"数据库类型: SQLite")
|
||
print(f"数据库文件: kamaxitong.db")
|
||
|
||
with app.app_context():
|
||
# 1. 删除所有表（如果存在）
|
||
print("\n1. 清理现有数据库表...")
|
||
db.drop_all()
|
||
print(" ✓ 已删除所有现有表")
|
||
|
||
# 2. 创建所有表
|
||
print("\n2. 创建数据库表结构...")
|
||
db.create_all()
|
||
print(" ✓ 已创建所有数据表")
|
||
|
||
# 显式提交事务以确保表被创建
|
||
db.session.commit()
|
||
|
||
# 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_sqlite_data()
|
||
|
||
# 显式提交并关闭会话
|
||
db.session.commit()
|
||
db.session.close()
|
||
|
||
# 4. 显示数据库信息
|
||
print("\n4. SQLite 数据库初始化完成！")
|
||
show_sqlite_database_info()
|
||
|
||
# 5. 显示数据库文件位置
|
||
db_path = os.path.abspath('kamaxitong.db')
|
||
print(f"\n5. 数据库文件位置: {db_path}")
|
||
# 只有在文件存在时才显示大小
|
||
if os.path.exists(db_path):
|
||
print(f" 文件大小: {os.path.getsize(db_path)} 字节")
|
||
else:
|
||
print(" 文件不存在")
|
||
|
||
except Exception as e:
|
||
print(f"\n❌ SQLite 数据库初始化失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
raise
|
||
|
||
|
||
def insert_sqlite_data():
|
||
"""插入 SQLite 初始数据"""
|
||
from app import db
|
||
from app.models import Admin, Product, Version, License, Device, Ticket
|
||
|
||
# 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
|
||
)
|
||
]
|
||
|
||
for product in products:
|
||
db.session.add(product)
|
||
print(" ✓ 已创建2个示例产品")
|
||
|
||
# 3. 创建版本信息
|
||
print(" 创建版本信息...")
|
||
versions = [
|
||
Version(
|
||
product_id='KMX001',
|
||
version_num='1.0.0',
|
||
update_log='初始版本发布',
|
||
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='功能增强版本',
|
||
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='KMX002',
|
||
version_num='2.0.0',
|
||
update_log='企业版初始版本',
|
||
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
|
||
)
|
||
]
|
||
|
||
for version in versions:
|
||
db.session.add(version)
|
||
print(" ✓ 已创建3个版本信息")
|
||
|
||
# 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
|
||
),
|
||
# 未激活许可证
|
||
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
|
||
)
|
||
]
|
||
|
||
for license_obj in licenses:
|
||
db.session.add(license_obj)
|
||
print(" ✓ 已创建3个示例许可证")
|
||
|
||
# 5. 创建示例设备
|
||
print(" 创建示例设备...")
|
||
devices = [
|
||
Device(
|
||
machine_code='DEMO-MACHINE-CODE-001',
|
||
license_id=1, # 对应第一个许可证
|
||
product_id='KMX001',
|
||
software_version='1.1.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.0.0',
|
||
status=1, # 正常
|
||
activate_time=current_time - timedelta(days=15),
|
||
last_verify_time=current_time - timedelta(hours=6)
|
||
)
|
||
]
|
||
|
||
for device in devices:
|
||
db.session.add(device)
|
||
print(" ✓ 已创建2个示例设备")
|
||
|
||
# 6. 创建示例工单
|
||
print(" 创建示例工单...")
|
||
tickets = [
|
||
Ticket(
|
||
title='许可证激活失败',
|
||
product_id='KMX001',
|
||
software_version='1.0.0',
|
||
machine_code='CUSTOMER-MACHINE-001',
|
||
license_key='DEMO-LICENSE-KEY-001',
|
||
description='客户反馈无法激活许可证',
|
||
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=None,
|
||
license_key=None,
|
||
description='客户咨询批量购买事宜',
|
||
priority=1, # 中优先级
|
||
status=0, # 待处理
|
||
operator=None,
|
||
remark=None,
|
||
create_time=current_time - timedelta(days=2),
|
||
update_time=current_time - timedelta(days=2)
|
||
)
|
||
]
|
||
|
||
for ticket in tickets:
|
||
db.session.add(ticket)
|
||
print(" ✓ 已创建2个示例工单")
|
||
|
||
# 提交所有更改
|
||
db.session.commit()
|
||
print(" ✓ 所有初始数据插入完成")
|
||
|
||
|
||
def show_sqlite_database_info():
|
||
"""显示 SQLite 数据库信息"""
|
||
from app import db
|
||
from app.models import Admin, Product, Version, License, Device, Ticket
|
||
|
||
print("\n" + "=" * 50)
|
||
print("SQLite 数据库初始化信息统计")
|
||
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. SQLite 数据库文件为 kamaxitong.db")
|
||
print(" 3. 如需使用 MySQL，请修改 .env 文件中的 DATABASE_URL")
|
||
print(" 4. 定期备份数据库文件")
|
||
print("=" * 50)
|
||
|
||
|
||
def main():
|
||
"""主函数"""
|
||
try:
|
||
init_sqlite_database()
|
||
print("\n🎉 SQLite 数据库初始化成功！")
|
||
print("\n启动服务器:")
|
||
print(" python run.py")
|
||
print(" 或")
|
||
print(" python start.py")
|
||
print("\n访问: http://localhost:5000")
|
||
except Exception as e:
|
||
print(f"\n❌ 初始化失败: {e}")
|
||
sys.exit(1)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main() |