Kamixitong/start.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
KaMiXiTong 启动脚本
快速启动开发服务器
"""

import os
import sys
import subprocess
import threading
import time
from pathlib import Path

def check_python_version():
    """检查Python版本"""
    if sys.version_info < (3, 6):
        print("❌ 错误: 需要Python 3.6或更高版本")
        print(f"当前版本: {sys.version}")
        sys.exit(1)
    print(f"✅ Python版本检查通过: {sys.version}")

def install_dependencies():
    """安装依赖"""
    print("📦 正在安装依赖包...")
    try:
        subprocess.check_call([sys.executable, '-m', 'pip', 'install', '-r', 'requirements.txt'])
        # 检查并安装FastAPI依赖
        if os.path.exists('requirements-fastapi.txt'):
            try:
                subprocess.check_call([sys.executable, '-m', 'pip', 'install', '-r', 'requirements-fastapi.txt'])
                print("✅ FastAPI依赖安装完成")
            except subprocess.CalledProcessError:
                print("⚠️  FastAPI依赖安装失败，将跳过FastAPI服务")
        print("✅ 依赖安装完成")
    except subprocess.CalledProcessError:
        print("❌ 依赖安装失败")
        sys.exit(1)

def check_env_file():
    """检查环境配置文件"""
    env_file = Path('.env')
    if not env_file.exists():
        print("⚠️  未找到 .env 文件，正在创建默认配置...")
        try:
            with open('.env.example', 'r', encoding='utf-8') as src:
                with open('.env', 'w', encoding='utf-8') as dst:
                    dst.write(src.read())
            print("✅ 已创建默认 .env 文件")
        except FileNotFoundError:
            print("❌ 未找到 .env.example 文件")
    else:
        print("✅ 环境配置文件已存在")

def init_database():
    """初始化数据库"""
    print("🗄️  正在初始化数据库...")
    try:
        from app import create_app, db
        from app.models.admin import Admin

        app = create_app('development')
        with app.app_context():
            # 创建数据表
            db.create_all()

            # 创建默认管理员
            admin = Admin.query.filter_by(username='admin').first()
            if not admin:
                admin = Admin(
                    username='admin',
                    email='admin@example.com',
                    role=1  # 超级管理员
                )
                admin.set_password('admin123')
                db.session.add(admin)
                db.session.commit()
                print("✅ 默认管理员账号创建成功: admin/admin123")
            else:
                print("✅ 数据库已初始化")

    except Exception as e:
        print(f"❌ 数据库初始化失败: {e}")
        sys.exit(1)

def create_directories():
    """创建必要的目录"""
    directories = [
        'logs',
        'static/uploads',
        'static/css',
        'static/js'
    ]

    for directory in directories:
        Path(directory).mkdir(parents=True, exist_ok=True)
    print("✅ 目录结构创建完成")

def start_flask_app():
    """启动Flask应用"""
    try:
        from run import app
        print("🚀 正在启动Flask应用...")
        app.run(
            host=os.environ.get('HOST', '0.0.0.0'),
            port=int(os.environ.get('PORT', 5088)),
            debug=True,
            use_reloader=False  # 禁用重载器以避免子进程问题
        )
    except Exception as e:
        print(f"❌ Flask应用启动失败: {e}")

def start_fastapi_app():
    """启动FastAPI应用"""
    try:
        # 检查FastAPI应用文件是否存在
        if not os.path.exists('fastapi_app.py'):
            print("⚠️  FastAPI应用文件不存在，跳过FastAPI服务启动")
            return
            
        print("🚀 正在启动FastAPI应用...")
        # 使用uvicorn启动FastAPI应用
        fastapi_host = os.environ.get('FASTAPI_HOST', '127.0.0.1')
        fastapi_port = os.environ.get('FASTAPI_PORT', '9002')
        
        subprocess.run([
            sys.executable, '-m', 'uvicorn', 
            'fastapi_app:app', 
            '--host', fastapi_host, 
            '--port', fastapi_port
        ])
    except Exception as e:
        print(f"❌ FastAPI应用启动失败: {e}")

def main():
    """主函数"""
    print("=" * 50)
    print("    KaMiXiTong 软件授权管理系统")
    print("           启动脚本 v1.0")
    print("=" * 50)

    # 检查Python版本
    check_python_version()

    # 创建目录
    create_directories()

    # 检查环境配置
    check_env_file()

    # 安装依赖
    if not os.environ.get('SKIP_INSTALL'):
        install_dependencies()
    else:
        print("⏭️  跳过依赖安装")

    # 初始化数据库
    init_database()

    print("\n" + "=" * 50)
    print("🚀 启动开发服务器...")
    print("=" * 50)
    host = os.environ.get('HOST', '127.0.0.1')
    port = os.environ.get('PORT', '5088')
    fastapi_port = os.environ.get('FASTAPI_PORT', '9002')
    
    print(f"📍 Flask访问地址: http://{host}:{port}")
    print(f"📍 FastAPI访问地址: http://{host}:{fastapi_port}")
    print(f"📍 FastAPI文档: http://{host}:{fastapi_port}/docs")
    print("👤 管理员账号: admin")
    print("🔑 管理员密码: admin123")
    print("⏹️  按 Ctrl+C 停止服务器")
    print("=" * 50 + "\n")

    # 启动Flask和FastAPI应用
    try:
        # 在单独的线程中启动Flask应用
        flask_thread = threading.Thread(target=start_flask_app)
        flask_thread.daemon = True
        flask_thread.start()
        
        # 等待Flask启动
        time.sleep(2)
        
        # 在主线程中启动FastAPI应用
        start_fastapi_app()
        
    except KeyboardInterrupt:
        print("\n👋 服务器已停止")
    except Exception as e:
        print(f"❌ 启动失败: {e}")
        sys.exit(1)

if __name__ == '__main__':
    main()