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)})