mirror of
https://github.com/zhouxiaoka/autoclip.git
synced 2026-09-03 06:24:14 +08:00
97 lines
3.6 KiB
Python
97 lines
3.6 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
测试切片API的数据转换
|
||
"""
|
||
|
||
import requests
|
||
import json
|
||
|
||
def test_clips_api():
|
||
"""测试切片API的数据转换"""
|
||
base_url = "http://localhost:8000/api/v1"
|
||
project_id = "1fdb0bf1-7f3c-44f7-a69d-90c5a1d26fbe"
|
||
|
||
print("🧪 开始测试切片API...")
|
||
|
||
# 1. 测试后端API
|
||
print("\n1. 测试后端API...")
|
||
try:
|
||
response = requests.get(f"{base_url}/clips/?project_id={project_id}")
|
||
if response.status_code == 200:
|
||
data = response.json()
|
||
print(f"✅ 后端API返回 {len(data['items'])} 个切片")
|
||
|
||
if data['items']:
|
||
clip = data['items'][0]
|
||
print(f" 第一个切片:")
|
||
print(f" - ID: {clip['id']}")
|
||
print(f" - 标题: {clip['title']}")
|
||
print(f" - 开始时间: {clip['start_time']}秒")
|
||
print(f" - 结束时间: {clip['end_time']}秒")
|
||
print(f" - 分数: {clip['score']}")
|
||
|
||
metadata = clip.get('clip_metadata', {})
|
||
print(f" - 推荐理由: {metadata.get('recommend_reason', '无')}")
|
||
print(f" - 内容要点: {len(metadata.get('content', []))} 个")
|
||
else:
|
||
print(f"❌ 后端API失败: {response.status_code}")
|
||
return
|
||
except Exception as e:
|
||
print(f"❌ 后端API异常: {e}")
|
||
return
|
||
|
||
# 2. 测试前端API（模拟）
|
||
print("\n2. 测试前端API数据转换...")
|
||
try:
|
||
# 模拟前端的数据转换逻辑
|
||
clips = data['items']
|
||
converted_clips = []
|
||
|
||
for clip in clips:
|
||
# 转换秒数为时间字符串格式
|
||
def format_seconds_to_time(seconds):
|
||
hours = int(seconds // 3600)
|
||
minutes = int((seconds % 3600) // 60)
|
||
secs = int(seconds % 60)
|
||
return f"{hours:02d}:{minutes:02d}:{secs:02d}"
|
||
|
||
# 获取metadata中的内容
|
||
metadata = clip.get('clip_metadata', {})
|
||
|
||
converted_clip = {
|
||
'id': clip['id'],
|
||
'title': clip['title'],
|
||
'generated_title': clip['title'],
|
||
'start_time': format_seconds_to_time(clip['start_time']),
|
||
'end_time': format_seconds_to_time(clip['end_time']),
|
||
'final_score': clip['score'] or 0,
|
||
'recommend_reason': metadata.get('recommend_reason') or clip['description'] or '',
|
||
'outline': metadata.get('outline') or clip['description'] or '',
|
||
'content': metadata.get('content') or [clip['description'] or ''],
|
||
'chunk_index': metadata.get('chunk_index') or 0
|
||
}
|
||
converted_clips.append(converted_clip)
|
||
|
||
print(f"✅ 前端转换完成: {len(converted_clips)} 个切片")
|
||
|
||
if converted_clips:
|
||
clip = converted_clips[0]
|
||
print(f" 第一个转换后的切片:")
|
||
print(f" - ID: {clip['id']}")
|
||
print(f" - 标题: {clip['title']}")
|
||
print(f" - 开始时间: {clip['start_time']}")
|
||
print(f" - 结束时间: {clip['end_time']}")
|
||
print(f" - 分数: {clip['final_score']}")
|
||
print(f" - 推荐理由: {clip['recommend_reason'][:50]}...")
|
||
print(f" - 内容要点: {len(clip['content'])} 个")
|
||
|
||
except Exception as e:
|
||
print(f"❌ 前端转换异常: {e}")
|
||
return
|
||
|
||
print("\n🎉 切片API测试完成!")
|
||
|
||
if __name__ == "__main__":
|
||
test_clips_api()
|
||
|