first commit
This commit is contained in:
ytc1012
289
MeetSpot/test_minimal_api.py
Normal file
289
MeetSpot/test_minimal_api.py
Normal file
@@ -0,0 +1,289 @@
|
||||
#!/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)
|
||||
Reference in New Issue
Block a user