117 lines
4.0 KiB
Python
117 lines
4.0 KiB
Python
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})
|