#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
生产环境诊断脚本
检查Python版本、可用模块和推荐的脚本版本
"""

import sys
import subprocess

def check_python_version():
    """检查Python版本"""
    version = sys.version_info
    version_str = "{}.{}".format(version.major, version.minor)
    print("Python版本: {}".format(version_str))

    if version.major >= 3 and version.minor >= 7:
        return "full", version_str
    elif version.major >= 3 and version.minor >= 6:
        return "py36", version_str
    else:
        return "py34", version_str

def check_module(module_name):
    """检查模块是否可用"""
    try:
        __import__(module_name)
        return True, "可用"
    except ImportError:
        return False, "不可用"

def check_dependencies():
    """检查关键依赖"""
    deps = ['pymysql', 'pandas', 'sqlalchemy']
    results = {}
    for dep in deps:
        available, status = check_module(dep)
        results[dep] = (available, status)
        print("依赖 {}: {}".format(dep, status))
    return results

def recommend_script(python_level, has_typing):
    """推荐使用的脚本"""
    if python_level == "full" and has_typing:
        return "02_parse_score_data_fixed.py", "完整功能版本"
    elif python_level in ["full", "py36"] and has_typing:
        return "02_parse_score_data_py36_compatible.py", "Python 3.6兼容版本"
    else:
        return "02_parse_score_data_py34_compatible.py", "超兼容版本（无typing）"

def test_script_syntax(script_name):
    """测试脚本语法"""
    try:
        subprocess.check_call([sys.executable, "-m", "py_compile", script_name],
                            stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
        return True, "语法正确"
    except subprocess.CalledProcessError:
        return False, "语法错误"

def main():
    print("=== 生产环境诊断工具 ===")
    print()

    # 检查Python版本
    python_level, version_str = check_python_version()

    # 检查typing模块
    has_typing, typing_status = check_module('typing')
    print("typing模块: {}".format(typing_status))

    print()

    # 检查依赖
    print("检查依赖包:")
    deps_status = check_dependencies()

    print()

    # 推荐脚本
    script_name, script_desc = recommend_script(python_level, has_typing)
    print("推荐脚本: {} ({})".format(script_name, script_desc))

    # 测试推荐脚本语法
    syntax_ok, syntax_msg = test_script_syntax(script_name)
    print("脚本语法: {}".format(syntax_msg))

    print()

    # 给出建议
    print("=== 执行建议 ===")

    if syntax_ok:
        print("✅ 环境检查通过")
        print("执行命令:")
        print("  nohup python3 {} > 2.log 2>&1 &".format(script_name))
        print("  tail -f 2.log")
    else:
        print("❌ 发现问题，请检查环境配置")

    # 检查是否有缺失的依赖
    missing_deps = [dep for dep, (available, _) in deps_status.items() if not available]
    if missing_deps:
        print()
        print("缺失依赖，请安装:")
        print("  pip3 install --user -r requirements-prod.txt")

    print()
    print("=== 版本信息 ===")
    print("Python: {}".format(version_str))
    print("Typing: {}".format(typing_status))
    print("推荐脚本: {}".format(script_name))

if __name__ == "__main__":
    main()