#!/usr/bin/env python
"""
环境配置检查脚本
"""
import os
import sys


def check_env():
    """检查环境配置"""
    print("=== Tolan 环境检查 ===\n")
    
    # 检查Python版本
    print(f"Python版本: {sys.version}")
    if sys.version_info < (3, 9):
        print("❌ 需要Python 3.9或更高版本")
        return False
    else:
        print("✅ Python版本满足要求\n")
    
    # 检查.env文件
    if not os.path.exists('.env'):
        print("⚠️  .env文件不存在")
        print("   请复制 .env.example 为 .env 并配置API Key\n")
    else:
        print("✅ .env文件存在\n")
    
    # 检查依赖包
    print("检查依赖包...")
    required_packages = [
        'fastapi',
        'uvicorn',
        'websockets',
        'dashscope',
        'openai',
        'aiohttp',
        'pydantic',
    ]
    
    missing_packages = []
    for package in required_packages:
        try:
            __import__(package)
            print(f"  ✅ {package}")
        except ImportError:
            print(f"  ❌ {package}")
            missing_packages.append(package)
    
    if missing_packages:
        print(f"\n⚠️  缺少依赖包: {', '.join(missing_packages)}")
        print("   请运行: python -m pip install -r requirements.txt\n")
        return False
    else:
        print("\n✅ 所有依赖包已安装\n")
    
    print("=== 环境检查完成 ===")
    return True


if __name__ == "__main__":
    if check_env():
        print("\n🎉 环境配置正常，可以启动服务")
        sys.exit(0)
    else:
        print("\n❌ 环境配置存在问题，请先修复")
        sys.exit(1)