from django.http import JsonResponse
from django.views.decorators.http import require_GET
from django.utils import timezone  # 使用 Django 的时区支持
import json
from chuangliangBaidu.migrations.BaidutaskQuery import AsyncTaskDataQuery
from django.http import JsonResponse
from django.views.decorators.http import require_GET
from django.utils import timezone
from django.shortcuts import render
import json


def get_task_details(request):
    """获取任务详情的API接口（优化版）"""
    # 获取任务ID参数
    task_id = request.GET.get('task_id')
    current_time = timezone.now().strftime("%Y-%m-%d %H:%M:%S")

    # 处理浏览器请求的Accept头
    is_browser_request = 'text/html' in request.META.get('HTTP_ACCEPT', '')

    # 参数验证
    if not task_id:
        error_data = {
            "code": 400,
            "message": "缺少参数: task_id",
            "time": current_time,
            "endpoint": request.path,
            "method": request.method
        }
        if is_browser_request:
            return render(request, 'json_formatter3.html', {
                "json_data": json.dumps(error_data, ensure_ascii=False),
                "status": 400
            })
        return JsonResponse(error_data, status=400, json_dumps_params={'ensure_ascii': False})

    try:
        # 转换为整数
        task_id = int(task_id)
    except ValueError:
        error_data = {
            "code": 400,
            "message": "task_id 必须是整数",
            "time": current_time,
            "provided_value": task_id
        }
        if is_browser_request:
            return render(request, 'json_formatter3.html', {
                "json_data": json.dumps(error_data, ensure_ascii=False),
                "status": 400
            })
        return JsonResponse(error_data, status=400, json_dumps_params={'ensure_ascii': False})

    # 创建查询实例并获取数据
    try:
        task_query = AsyncTaskDataQuery(task_id)
        result = task_query.get_task_details()
        # 处理任务日志并解码中文
        task_logs = result.get("task_logs", [])
        for log in task_logs:
            # 处理 task_param
            if isinstance(log.get('task_param'), str):
                try:
                    log['task_param'] = json.loads(log['task_param'])
                except json.JSONDecodeError:
                    pass
            # 处理 result_data
            if isinstance(log.get('result_data'), str):
                try:
                    log['result_data'] = json.loads(log['result_data'])
                except json.JSONDecodeError:
                    pass

        # 准备响应数据
        response_data = {
            "code": 200,
            "message": "成功",
            "time": current_time,
            "task_id": task_id,
            "task_logs": task_logs,
            "batch_id": result.get("batch_id", ""),
            "create_status": result.get("create_status", ""),
            "total_logs": len(task_logs)
        }

        # 如果是浏览器请求，返回HTML视图
        if is_browser_request:
            return render(request, 'json_formatter3.html', {
                "json_data": response_data,
                # "json_data": json.dumps(response_data, ensure_ascii=False, indent=2),
                "status": 200,
                "task_id": task_id
            })

        # 返回JSON响应
        return JsonResponse(
            response_data,
            json_dumps_params={'ensure_ascii': False}
        )

    except Exception as e:
        error_data = {
            "code": 500,
            "message": "服务器内部错误",
            "error": str(e),
            "time": current_time,
            "task_id": task_id
        }
        print(e)
        if is_browser_request:
            return render(request, 'json_formatter3.html', {
                "json_data": json.dumps(error_data, ensure_ascii=False),
                "status": 500
            })
        return JsonResponse(error_data, status=500, json_dumps_params={'ensure_ascii': False})