338 lines
11 KiB
Python
338 lines
11 KiB
Python
from django.http import JsonResponse
|
||
from django.views.decorators.http import require_GET
|
||
from django.utils import timezone # 使用 Django 的时区支持
|
||
import json
|
||
# 确保导入你的 AsyncTaskDataQuery 类
|
||
from chuanglinagBaidu.migrations.BaidutaskQuery import AsyncTaskDataQuery # 替换 your_module 为实际模块名
|
||
#
|
||
#
|
||
# # @require_GET
|
||
# 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")
|
||
#
|
||
# if not task_id:
|
||
# return JsonResponse({
|
||
# "code": 400,
|
||
# "message": "缺少参数: task_id",
|
||
# "time": current_time
|
||
# }, status=400)
|
||
#
|
||
# try:
|
||
# # 转换为整数
|
||
# task_id = int(task_id)
|
||
# except ValueError:
|
||
# return JsonResponse({
|
||
# "code": 400,
|
||
# "message": "task_id 必须是整数",
|
||
# "time": current_time
|
||
# }, status=400)
|
||
#
|
||
# # 创建查询实例并获取数据
|
||
# task_query = AsyncTaskDataQuery(task_id)
|
||
# result = task_query.get_task_details()
|
||
# # 获取任务日志并处理JSON嵌套
|
||
# task_logs = result.get("task_logs", [])
|
||
#
|
||
# # 处理可能的嵌套JSON字符串
|
||
# 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:
|
||
# # 可选的日志记录
|
||
# # logger.debug(f"无法解析 task_param: {log['task_param']}")
|
||
# 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": timezone.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||
# "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)
|
||
# }
|
||
#
|
||
# # 返回时禁用ASCII转义
|
||
# return JsonResponse(
|
||
# response_data,
|
||
# json_dumps_params={'ensure_ascii': False}
|
||
# )
|
||
# from django.http import JsonResponse, HttpResponse
|
||
# from django.views.decorators.http import require_GET
|
||
# from django.utils import timezone
|
||
# from django.shortcuts import render
|
||
# import json
|
||
# import re
|
||
# # 确保导入你的 AsyncTaskDataQuery 类
|
||
# from chuanglinagBaidu.migrations.BaidutaskQuery import AsyncTaskDataQuery # 替换为实际模块名
|
||
#
|
||
#
|
||
# def render_json_pretty(data, status=200):
|
||
# """渲染美观的JSON HTML视图"""
|
||
# # 准备响应数据 - 格式化JSON用于展示
|
||
# formatted_json = json.dumps(data, indent=4, ensure_ascii=False)
|
||
#
|
||
# # 生成HTML内容
|
||
# html_content = f"""
|
||
#
|
||
# """
|
||
# return HttpResponse(html_content, status=status, content_type='text/html')
|
||
#
|
||
#
|
||
# @require_GET
|
||
# 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")
|
||
# host = request.get_host()
|
||
#
|
||
# # 参数验证
|
||
# if not task_id:
|
||
# error_data = {
|
||
# "code": 400,
|
||
# "message": "缺少参数: task_id",
|
||
# "time": current_time,
|
||
# "path": request.path,
|
||
# "method": request.method,
|
||
# "host": host
|
||
# }
|
||
#
|
||
# # 检查是否为浏览器请求
|
||
# if 'text/html' in request.META.get('HTTP_ACCEPT', ''):
|
||
# return render_json_pretty(error_data, 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,
|
||
# "path": request.path,
|
||
# "host": host
|
||
# }
|
||
#
|
||
# if 'text/html' in request.META.get('HTTP_ACCEPT', ''):
|
||
# return render_json_pretty(error_data, 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()
|
||
#
|
||
# # 处理任务日志并解析嵌套JSON
|
||
# 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:
|
||
# # 保留原始字符串并添加解析错误标记
|
||
# log['task_param'] = {
|
||
# "__parse_error": True,
|
||
# "raw_value": log['task_param']
|
||
# }
|
||
#
|
||
# # 处理 result_data
|
||
# if isinstance(log.get('result_data'), str):
|
||
# try:
|
||
# log['result_data'] = json.loads(log['result_data'])
|
||
# except json.JSONDecodeError:
|
||
# log['result_data'] = {
|
||
# "__parse_error": True,
|
||
# "raw_value": log['result_data']
|
||
# }
|
||
#
|
||
# # 准备响应数据
|
||
# 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),
|
||
# "api_details": {
|
||
# "method": request.method,
|
||
# "path": request.path,
|
||
# "host": host
|
||
# }
|
||
# }
|
||
#
|
||
# # 检查是否为浏览器请求
|
||
# if 'text/html' in request.META.get('HTTP_ACCEPT', ''):
|
||
# return render_json_pretty(response_data)
|
||
#
|
||
# # 为JSON响应添加元数据
|
||
# response_data['api_details']['rendered_format'] = 'json'
|
||
# return JsonResponse(
|
||
# response_data,
|
||
# json_dumps_params={
|
||
# 'ensure_ascii': False,
|
||
# 'indent': 2 if 'pretty' in request.GET else None
|
||
# }
|
||
# )
|
||
#
|
||
# except Exception as e:
|
||
# # 记录异常信息
|
||
# error_data = {
|
||
# "code": 500,
|
||
# "message": "服务器内部错误",
|
||
# "error_detail": str(e),
|
||
# "time": current_time,
|
||
# "task_id": task_id,
|
||
# "host": host,
|
||
# "path": request.path
|
||
# }
|
||
#
|
||
# if 'text/html' in request.META.get('HTTP_ACCEPT', ''):
|
||
# return render_json_pretty(error_data, 500)
|
||
#
|
||
# return JsonResponse(
|
||
# error_data,
|
||
# status=500,
|
||
# json_dumps_params={'ensure_ascii': False}
|
||
# )
|
||
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_formatter.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_formatter.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_formatter.html', {
|
||
"json_data": json.dumps(error_data, ensure_ascii=False),
|
||
"status": 500
|
||
})
|
||
return JsonResponse(error_data, status=500, json_dumps_params={'ensure_ascii': False})
|