290 lines
8.8 KiB
Python
290 lines
8.8 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
MeetSpot 精简版 API 测试脚本
|
||
=============================
|
||
|
||
测试精简后的 API 接口是否正常工作，并输出性能数据。
|
||
|
||
使用方法:
|
||
python test_minimal_api.py
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
import time
|
||
import json
|
||
from pathlib import Path
|
||
|
||
# 添加项目根目录到路径
|
||
project_root = Path(__file__).resolve().parent
|
||
sys.path.insert(0, str(project_root))
|
||
|
||
def test_import():
|
||
"""测试模块导入"""
|
||
print("📦 测试模块导入...")
|
||
try:
|
||
from app.config_simple import config
|
||
from app.tool.meetspot_recommender import CafeRecommender
|
||
print(" ✅ 核心模块导入成功")
|
||
return True
|
||
except Exception as e:
|
||
print(f" ❌ 导入失败: {e}")
|
||
return False
|
||
|
||
def test_tmap_config():
|
||
"""测试腾讯地图配置"""
|
||
print("\n🗺️ 测试腾讯地图配置...")
|
||
try:
|
||
from app.config_simple import config
|
||
|
||
if not config or not config.tmap:
|
||
print(" ⚠️ 配置未加载（使用环境变量）")
|
||
return True
|
||
|
||
api_key = config.tmap.api_key
|
||
if api_key and len(api_key) > 10:
|
||
print(f" ✅ API Key 已配置: {api_key[:10]}...")
|
||
else:
|
||
print(" ⚠️ API Key 未配置（可从环境变量 TMAP_API_KEY 获取）")
|
||
|
||
return True
|
||
except Exception as e:
|
||
print(f" ❌ 配置测试失败: {e}")
|
||
return False
|
||
|
||
def test_geocoding():
|
||
"""测试地理编码"""
|
||
print("\n📍 测试地理编码...")
|
||
try:
|
||
import asyncio
|
||
from app.tool.meetspot_recommender import CafeRecommender
|
||
|
||
async def test():
|
||
recommender = CafeRecommender()
|
||
result = await recommender._geocode("北京大学")
|
||
|
||
if result and result.get("location"):
|
||
print(f" ✅ 地理编码成功: {result['location']}")
|
||
print(f" 📌 地址: {result.get('formatted_address', 'N/A')}")
|
||
return True
|
||
else:
|
||
print(" ⚠️ 地理编码失败（可能 API Key 未配置）")
|
||
return False
|
||
|
||
return asyncio.run(test())
|
||
except Exception as e:
|
||
print(f" ❌ 测试失败: {e}")
|
||
return False
|
||
|
||
def test_poi_search():
|
||
"""测试 POI 搜索"""
|
||
print("\n🏪 测试 POI 搜索...")
|
||
try:
|
||
import asyncio
|
||
from app.tool.meetspot_recommender import CafeRecommender
|
||
|
||
async def test():
|
||
recommender = CafeRecommender()
|
||
pois = await recommender._search_pois("116.4074,39.9042", "咖啡馆", radius=1000)
|
||
|
||
if pois and len(pois) > 0:
|
||
print(f" ✅ POI 搜索成功: 找到 {len(pois)} 个结果")
|
||
for i, poi in enumerate(pois[:3]):
|
||
print(f" {i+1}. {poi.get('name', 'N/A')}")
|
||
return True
|
||
else:
|
||
print(" ⚠️ POI 搜索失败（可能 API Key 未配置）")
|
||
return False
|
||
|
||
return asyncio.run(test())
|
||
except Exception as e:
|
||
print(f" ❌ 测试失败: {e}")
|
||
return False
|
||
|
||
def test_complete_recommendation():
|
||
"""测试完整推荐流程"""
|
||
print("\n🎯 测试完整推荐流程...")
|
||
try:
|
||
import asyncio
|
||
from app.tool.meetspot_recommender import CafeRecommender
|
||
|
||
async def test():
|
||
start_time = time.time()
|
||
|
||
recommender = CafeRecommender()
|
||
result = await recommender.run(
|
||
locations=["北京大学", "清华大学", "中关村"],
|
||
keywords="咖啡馆",
|
||
user_requirements="环境安静"
|
||
)
|
||
|
||
elapsed = time.time() - start_time
|
||
|
||
if result and "HTML页面:" in result.output:
|
||
print(f" ✅ 推荐成功（耗时: {elapsed:.2f}秒）")
|
||
print(f" 📄 输出: {result.output[:80]}...")
|
||
return True
|
||
else:
|
||
print(" ⚠️ 推荐失败（可能 API Key 未配置）")
|
||
print(f" 错误信息: {result.output if result else 'None'}")
|
||
return False
|
||
|
||
return asyncio.run(test())
|
||
except Exception as e:
|
||
print(f" ❌ 测试失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
def test_minimal_api():
|
||
"""测试精简版 API"""
|
||
print("\n🚀 测试精简版 API...")
|
||
try:
|
||
sys.path.insert(0, str(project_root))
|
||
os.environ.setdefault("TMAP_API_KEY", "test_key_for_import")
|
||
|
||
# 测试是否可以导入
|
||
import importlib.util
|
||
spec = importlib.util.spec_from_file_location("api_minimal", project_root / "api_minimal.py")
|
||
if spec and spec.loader:
|
||
module = importlib.util.module_from_spec(spec)
|
||
spec.loader.exec_module(module)
|
||
print(" ✅ 精简版 API 导入成功")
|
||
return True
|
||
else:
|
||
print(" ⚠️ 精简版 API 文件不存在")
|
||
return False
|
||
except Exception as e:
|
||
print(f" ❌ 导入失败: {e}")
|
||
return False
|
||
|
||
def test_file_structure():
|
||
"""测试文件结构"""
|
||
print("\n📁 测试文件结构...")
|
||
|
||
essential_files = [
|
||
"api_minimal.py",
|
||
"api/tool/meetspot_recommender.py",
|
||
"app/config_simple.py",
|
||
"requirements.txt"
|
||
]
|
||
|
||
all_exist = True
|
||
for file_path in essential_files:
|
||
full_path = project_root / file_path
|
||
if full_path.exists():
|
||
size = full_path.stat().st_size
|
||
print(f" ✅ {file_path:<40} {size:>6} bytes")
|
||
else:
|
||
print(f" ❌ {file_path:<40} 不存在")
|
||
all_exist = False
|
||
|
||
return all_exist
|
||
|
||
def calculate_performance_score(results: dict) -> int:
|
||
"""计算性能分数"""
|
||
score = 0
|
||
max_score = 100
|
||
|
||
# 模块导入（20分）
|
||
if results.get('import', False):
|
||
score += 20
|
||
|
||
# 配置测试（10分）
|
||
if results.get('config', False):
|
||
score += 10
|
||
|
||
# 文件结构（20分）
|
||
if results.get('structure', False):
|
||
score += 20
|
||
|
||
# 精简版 API（10分）
|
||
if results.get('minimal_api', False):
|
||
score += 10
|
||
|
||
# 地理编码（20分）- 可选（需要 API Key）
|
||
if results.get('geocoding', False):
|
||
score += 20
|
||
|
||
# POI 搜索（20分）- 可选（需要 API Key）
|
||
if results.get('poi', False):
|
||
score += 20
|
||
|
||
return min(score, max_score)
|
||
|
||
def main():
|
||
"""主测试函数"""
|
||
print("=" * 70)
|
||
print("MeetSpot 精简版 API 测试")
|
||
print("=" * 70)
|
||
|
||
# 检查文件是否存在
|
||
if not (project_root / "api_minimal.py").exists():
|
||
print("❌ 错误: api_minimal.py 不存在")
|
||
print("请确保已运行精简脚本或手动创建")
|
||
sys.exit(1)
|
||
|
||
results = {}
|
||
|
||
# 运行测试
|
||
results['import'] = test_import()
|
||
results['config'] = test_tmap_config()
|
||
results['structure'] = test_file_structure()
|
||
results['minimal_api'] = test_minimal_api()
|
||
results['geocoding'] = test_geocoding()
|
||
results['poi'] = test_poi_search()
|
||
results['complete'] = test_complete_recommendation()
|
||
|
||
# 计算性能分数
|
||
score = calculate_performance_score(results)
|
||
|
||
# 输出测试报告
|
||
print("\n" + "=" * 70)
|
||
print("📊 测试报告")
|
||
print("=" * 70)
|
||
|
||
passed = sum(1 for v in results.values() if v)
|
||
total = len(results)
|
||
|
||
print(f"\n✅ 通过: {passed}/{total}")
|
||
print(f"📈 性能分数: {score}/100")
|
||
|
||
if score >= 80:
|
||
print(f"\n🎉 优秀! 精简版 API 已准备就绪")
|
||
elif score >= 60:
|
||
print(f"\n✅ 良好! 基本功能正常，部分功能需要 API Key")
|
||
else:
|
||
print(f"\n⚠️ 需要改进! 请检查错误信息")
|
||
|
||
# 输出建议
|
||
print(f"\n💡 建议:")
|
||
if not results.get('geocoding') or not results.get('poi'):
|
||
print(f" - 配置 TMAP_API_KEY 环境变量以启用完整功能")
|
||
print(f" - export TMAP_API_KEY=your_key")
|
||
|
||
if score >= 60:
|
||
print(f" - 可以开始部署到云托管")
|
||
print(f" - 使用 Dockerfile.minimal 构建镜像")
|
||
print(f" - 配置 VPC 访问类型")
|
||
|
||
print(f"\n🚀 下一步:")
|
||
print(f" 1. 配置真实 API Key")
|
||
print(f" 2. 部署到云托管")
|
||
print(f" 3. 测试小程序调用")
|
||
print(f" 4. 开发小程序前端")
|
||
|
||
return score
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
score = main()
|
||
sys.exit(0 if score >= 60 else 1)
|
||
except KeyboardInterrupt:
|
||
print("\n\n❌ 测试被中断")
|
||
sys.exit(1)
|
||
except Exception as e:
|
||
print(f"\n\n❌ 测试失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
sys.exit(1)
|