"""
API接口测试脚本
用于测试解析接口和文档生成接口
"""
import requests
import json

BASE_URL = "http://localhost:7500"

def test_extract_api():
    """测试解析接口"""
    print("="*60)
    print("测试解析接口 (/ai/extract)")
    print("="*60)
    
    url = f"{BASE_URL}/ai/extract"
    
    # 测试数据
    test_data = {
        "inputData": [
            {
                "fieldCode": "clue_info",
                "fieldValue": "被举报用户名称是张三，年龄30岁，某公司总经理，男性，1980年5月出生，中共党员"
            }
        ],
        "outputData": [
            {"fieldCode": "target_name"},
            {"fieldCode": "target_gender"},
            {"fieldCode": "target_organization_and_position"}
        ]
    }
    
    print(f"\n请求URL: {url}")
    print(f"请求数据:\n{json.dumps(test_data, ensure_ascii=False, indent=2)}")
    
    try:
        response = requests.post(url, json=test_data, timeout=30)
        print(f"\n响应状态码: {response.status_code}")
        
        result = response.json()
        print(f"响应数据:\n{json.dumps(result, ensure_ascii=False, indent=2)}")
        
        if result.get('isSuccess'):
            print("\n✓ 解析接口测试成功！")
            if result.get('data', {}).get('outData'):
                print("\n提取的字段：")
                for item in result['data']['outData']:
                    print(f"  - {item.get('fieldCode')}: {item.get('fieldValue', '(空)')}")
        else:
            print(f"\n✗ 解析接口测试失败: {result.get('errorMsg', '未知错误')}")
            
    except requests.exceptions.ConnectionError:
        print("\n✗ 连接失败：请确保服务已启动")
    except requests.exceptions.Timeout:
        print("\n✗ 请求超时")
    except Exception as e:
        print(f"\n✗ 测试失败: {str(e)}")
    
    print()


def test_generate_document_api():
    """测试文档生成接口"""
    print("="*60)
    print("测试文档生成接口 (/ai/generate-document)")
    print("="*60)
    
    url = f"{BASE_URL}/ai/generate-document"
    
    # 测试数据
    test_data = {
        "inputData": [
            {
                "fieldCode": "target_name",
                "fieldValue": "张三"
            },
            {
                "fieldCode": "target_gender",
                "fieldValue": "男"
            },
            {
                "fieldCode": "target_organization_and_position",
                "fieldValue": "某公司总经理"
            }
        ],
        "fpolicFieldParamFileList": [
            {
                "fileId": 1,
                "fileName": "测试文档.doc",
                "templateCode": "PRELIMINARY_VERIFICATION_APPROVAL"
            }
        ]
    }
    
    print(f"\n请求URL: {url}")
    print(f"请求数据:\n{json.dumps(test_data, ensure_ascii=False, indent=2)}")
    
    try:
        response = requests.post(url, json=test_data, timeout=60)
        print(f"\n响应状态码: {response.status_code}")
        
        result = response.json()
        print(f"响应数据:\n{json.dumps(result, ensure_ascii=False, indent=2)}")
        
        if result.get('isSuccess'):
            print("\n✓ 文档生成接口测试成功！")
            if result.get('data', {}).get('fpolicFieldParamFileList'):
                print("\n生成的文件：")
                for file_info in result['data']['fpolicFieldParamFileList']:
                    print(f"  - {file_info.get('fileName')}: {file_info.get('filePath', '(无路径)')}")
        else:
            print(f"\n✗ 文档生成接口测试失败: {result.get('errorMsg', '未知错误')}")
            print(f"错误码: {result.get('code', 'N/A')}")
            
    except requests.exceptions.ConnectionError:
        print("\n✗ 连接失败：请确保服务已启动")
    except requests.exceptions.Timeout:
        print("\n✗ 请求超时（文档生成可能需要较长时间）")
    except Exception as e:
        print(f"\n✗ 测试失败: {str(e)}")
        import traceback
        traceback.print_exc()
    
    print()


def main():
    """主函数"""
    print("\n" + "="*60)
    print("API接口测试")
    print("="*60)
    print(f"\n服务地址: {BASE_URL}")
    print("请确保服务已启动（运行 python app.py）\n")
    
    # 测试解析接口
    test_extract_api()
    
    # 测试文档生成接口（注释掉，因为需要MinIO配置和模板文件）
    # test_generate_document_api()
    
    print("="*60)
    print("测试完成")
    print("="*60)


if __name__ == '__main__':
    main()