"""
快速测试深度分析功能
"""
import requests

API_BASE = "http://129.211.173.87:10001"

def login():
    resp = requests.post(f"{API_BASE}/auth/login", json={
        "username": "admin",
        "password": "abcd1234"
    })
    return resp.json()["access_token"]

token = login()
headers = {"Authorization": f"Bearer {token}"}

print("=" * 60)
print("🧪 深度分析功能快速测试")
print("=" * 60)

# 1. 检查已有分析
print("\n📋 当前分析列表:")
resp = requests.get(f"{API_BASE}/deep-analysis/list?page=1&page_size=5", headers=headers)
data = resp.json()
print(f"   总数: {data.get('total', 0)}")
for item in data.get('items', [])[:3]:
    icon = {'pending': '⏳', 'processing': '🔄', 'completed': '✅', 'failed': '❌'}.get(item['status'], '❓')
    print(f"   {icon} [{item['id']}] {item['company_name']}: {item['status']}")
    if item.get('error_message'):
        print(f"      ⚠️ {item['error_message'][:60]}...")

# 2. 检查队列状态
print("\n🎛️ 队列状态:")
resp = requests.get(f"{API_BASE}/deep-analysis/queue/status", headers=headers)
qs = resp.json()
print(f"   队列中: {qs.get('queued_count', 0)} 个")
print(f"   处理中: {'是' if qs.get('is_processing') else '否'}")

# 3. 查看第一个分析的详情
if data.get('items'):
    first = data['items'][0]
    print(f"\n📄 查看详情 (ID: {first['id']}):")
    resp = requests.get(f"{API_BASE}/deep-analysis/{first['id']}", headers=headers)
    if resp.status_code == 200:
        detail = resp.json()
        print(f"   公司: {detail.get('company_name')}")
        print(f"   状态: {detail.get('status')}")
        print(f"   模型: {detail.get('model_used')}")
        if detail.get('canvas_summary'):
            summary = detail['canvas_summary']
            print(f"   评分: {summary.get('overall_score', 'N/A')}")
        if detail.get('error_message'):
            print(f"   错误: {detail['error_message'][:80]}...")

print("\n" + "=" * 60)
print("✅ 测试完成!")
print("=" * 60)