"""
测试服务是否能正常启动和运行
"""
import sys
import os

# 添加当前目录到路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

def test_imports():
    """测试所有模块是否能正常导入"""
    print("="*60)
    print("测试模块导入...")
    print("="*60)
    
    try:
        from flask import Flask
        print("✓ Flask 导入成功")
    except ImportError as e:
        print(f"✗ Flask 导入失败: {e}")
        return False
    
    try:
        from services.ai_service import AIService
        print("✓ AIService 导入成功")
    except ImportError as e:
        print(f"✗ AIService 导入失败: {e}")
        return False
    
    try:
        from services.field_service import FieldService
        print("✓ FieldService 导入成功")
    except ImportError as e:
        print(f"✗ FieldService 导入失败: {e}")
        return False
    
    try:
        from utils.response import success_response, error_response
        print("✓ response 工具导入成功")
    except ImportError as e:
        print(f"✗ response 工具导入失败: {e}")
        return False
    
    return True

def test_field_service():
    """测试字段服务"""
    print("\n" + "="*60)
    print("测试字段服务...")
    print("="*60)
    
    try:
        from services.field_service import FieldService
        field_service = FieldService()
        
        # 测试获取字段
        fields = field_service.get_output_fields_by_business_type('INVESTIGATION')
        print(f"✓ 成功获取 {len(fields)} 个输出字段")
        
        if fields:
            print(f"  示例字段: {fields[0].get('name', 'N/A')} ({fields[0].get('field_code', 'N/A')})")
        
        return True
    except Exception as e:
        print(f"✗ 字段服务测试失败: {e}")
        import traceback
        traceback.print_exc()
        return False

def test_ai_service():
    """测试AI服务"""
    print("\n" + "="*60)
    print("测试AI服务...")
    print("="*60)
    
    try:
        from services.ai_service import AIService
        ai_service = AIService()
        
        print(f"✓ AI服务初始化成功")
        print(f"  AI提供商: {ai_service.ai_provider}")
        
        if ai_service.ai_provider == 'none':
            print("  ⚠ 警告: 未配置AI服务，请设置SILICONFLOW_API_KEY")
        elif ai_service.ai_provider == 'siliconflow':
            print("  ✓ 已配置硅基流动API")
        elif ai_service.ai_provider == 'huawei':
            print("  ✓ 已配置华为大模型API")
        
        return True
    except Exception as e:
        print(f"✗ AI服务测试失败: {e}")
        import traceback
        traceback.print_exc()
        return False

def main():
    """主测试函数"""
    print("\n" + "="*60)
    print("智慧监督AI文书写作服务 - 启动测试")
    print("="*60)
    
    # 测试导入
    if not test_imports():
        print("\n✗ 模块导入测试失败，请检查依赖是否安装")
        return False
    
    # 测试字段服务
    if not test_field_service():
        print("\n✗ 字段服务测试失败")
        return False
    
    # 测试AI服务
    if not test_ai_service():
        print("\n✗ AI服务测试失败")
        return False
    
    print("\n" + "="*60)
    print("✓ 所有测试通过！")
    print("="*60)
    print("\n可以使用以下命令启动服务：")
    print("  python app.py")
    print("\n然后在浏览器访问: http://localhost:5000/")
    
    return True

if __name__ == '__main__':
    success = main()
    sys.exit(0 if success else 1)