107 lines
4.3 KiB
Python
107 lines
4.3 KiB
Python
|
from django.shortcuts import get_object_or_404
|
|||
|
|
from django.http import JsonResponse
|
|||
|
|
import requests
|
|||
|
|
from django.views.decorators.csrf import csrf_exempt
|
|||
|
|
|
|||
|
|
from .models import VideoGeneration
|
|||
|
|
@csrf_exempt
|
|||
|
|
def update_video_status(request, pid, query_type='revid'):
|
|||
|
|
"""
|
|||
|
|
查询视频生成状态接口，支持不同的生成接口查询：revid 或 runwayml
|
|||
|
|
:param pid: 视频生成的 UUID (pid)
|
|||
|
|
:param query_type: 'revid' 表示查询之前接口，'runwayml' 表示新的接口
|
|||
|
|
"""
|
|||
|
|
if query_type == 'revid':
|
|||
|
|
url = f"https://www.revid.ai/api/public/v2/status?pid={pid}"
|
|||
|
|
headers = {
|
|||
|
|
"key": "c4c0f516-33c0-4f5c-9d29-3b1d4e6a93c3",
|
|||
|
|
"Content-Type": "text/plain"
|
|||
|
|
}
|
|||
|
|
elif query_type == 'runwayml':
|
|||
|
|
url = f"https://api.aivideoapi.com/status?uuid={pid}" # 假设 runwayml 的接口类似，实际接口需替换
|
|||
|
|
headers = {
|
|||
|
|
"Authorization": "12e76710fad2047db8c0cc6b25987e2a2", # 替换为你的真实授权密钥
|
|||
|
|
"Content-Type": "application/json"
|
|||
|
|
}
|
|||
|
|
else:
|
|||
|
|
return JsonResponse({'code': 400, 'message': '无效的查询类型'})
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
response = requests.get(url, headers=headers)
|
|||
|
|
response.raise_for_status() # Raises an HTTPError for bad responses
|
|||
|
|
|
|||
|
|
data = response.json()
|
|||
|
|
print(f"查询类型: {query_type}")
|
|||
|
|
print(f"API响应内容: {data}")
|
|||
|
|
|
|||
|
|
# 获取视频生成记录
|
|||
|
|
video_generation = get_object_or_404(VideoGeneration, pid=pid)
|
|||
|
|
|
|||
|
|
if query_type == 'runwayml':
|
|||
|
|
# RunwayML接口返回的数据
|
|||
|
|
status = data.get('status')
|
|||
|
|
progress = data.get('progress', None)
|
|||
|
|
video_url = data.get('url', None)
|
|||
|
|
gif_url = data.get('gif_url', None)
|
|||
|
|
|
|||
|
|
# 根据不同状态更新数据库
|
|||
|
|
if status == 'success':
|
|||
|
|
video_generation.status = 'completed'
|
|||
|
|
video_generation.video_url = video_url # 更新视频 URL
|
|||
|
|
elif status in ['in queue', 'submitted']:
|
|||
|
|
video_generation.status = 'pending'
|
|||
|
|
elif status == 'submitted' and progress is not None:
|
|||
|
|
video_generation.status = 'in_progress'
|
|||
|
|
else:
|
|||
|
|
video_generation.status = 'failed'
|
|||
|
|
video_generation.description = 'Video generation failed due to an error.'
|
|||
|
|
|
|||
|
|
# 如果视频的gif_url存在,可以记录下来
|
|||
|
|
if gif_url:
|
|||
|
|
video_generation.gif_url = gif_url
|
|||
|
|
|
|||
|
|
video_generation.save()
|
|||
|
|
|
|||
|
|
return JsonResponse({
|
|||
|
|
'code': 200,
|
|||
|
|
'message': 'RunwayML 状态更新成功',
|
|||
|
|
'data': {
|
|||
|
|
'status': video_generation.status,
|
|||
|
|
'video_url': video_generation.video_url,
|
|||
|
|
'progress': progress,
|
|||
|
|
'gif_url': gif_url
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
# Revid 接口的处理（保持原来的逻辑）
|
|||
|
|
success = data.get('success')
|
|||
|
|
status = data.get('status')
|
|||
|
|
video_url = data.get('videoUrl', None)
|
|||
|
|
error_message = data.get('error', '')
|
|||
|
|
|
|||
|
|
if video_generation.video_url and video_generation.status == 'completed':
|
|||
|
|
return JsonResponse({'code': 200, 'message': '视频已生成，状态不需更新'})
|
|||
|
|
|
|||
|
|
if success == 1:
|
|||
|
|
if status == 'ready':
|
|||
|
|
video_generation.status = 'completed'
|
|||
|
|
if video_url:
|
|||
|
|
video_generation.video_url = video_url
|
|||
|
|
elif status == 'building':
|
|||
|
|
video_generation.status = 'pending'
|
|||
|
|
elif status == 'error':
|
|||
|
|
video_generation.status = 'failed'
|
|||
|
|
video_generation.description = error_message
|
|||
|
|
video_generation.save()
|
|||
|
|
|
|||
|
|
return JsonResponse({
|
|||
|
|
'code': 200,
|
|||
|
|
'message': '状态更新成功',
|
|||
|
|
'data': {'status': video_generation.status, 'video_url': video_url}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
except requests.exceptions.RequestException as e:
|
|||
|
|
print(f"发生错误: {e}")
|
|||
|
|
return JsonResponse({'code': 500, 'message': str(e)})
|