hyh
125 lines
4.4 KiB
Python
125 lines
4.4 KiB
Python
import logging
|
|
|
|
from django.db import transaction
|
|
from rest_framework import status
|
|
from rest_framework.decorators import api_view, permission_classes
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
|
|
from .llm import analyze_requirements
|
|
from .models import Requirement, Submission
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@api_view(["POST"])
|
|
@permission_classes([IsAuthenticated])
|
|
@transaction.atomic
|
|
def create_requirement(request):
|
|
title = (request.data.get("title") or "").strip()
|
|
content = (request.data.get("content") or "").strip()
|
|
if not title or not content:
|
|
return Response({"detail": "title 和 content 不能为空"}, status=status.HTTP_400_BAD_REQUEST)
|
|
try:
|
|
spec = analyze_requirements(content)
|
|
except Exception:
|
|
logger.exception("AI 分析作业要求失败, 使用默认检测计划")
|
|
from .llm import _default_spec
|
|
spec = _default_spec()
|
|
requirement = Requirement.objects.create(title=title, content=content, spec=spec)
|
|
from .serializers import RequirementSerializer
|
|
return Response(RequirementSerializer(requirement).data, status=status.HTTP_201_CREATED)
|
|
|
|
|
|
@api_view(["GET"])
|
|
@permission_classes([IsAuthenticated])
|
|
def list_requirements(request):
|
|
qs = Requirement.objects.all()
|
|
q = request.query_params.get("q")
|
|
if q:
|
|
qs = qs.filter(title__icontains=q) | qs.filter(content__icontains=q)
|
|
from .serializers import RequirementSerializer
|
|
page = int(request.query_params.get("page", 1))
|
|
size = int(request.query_params.get("size", 20))
|
|
items = qs[(page - 1) * size: page * size]
|
|
return Response({
|
|
"count": qs.count(),
|
|
"page": page,
|
|
"items": RequirementSerializer(items, many=True).data,
|
|
})
|
|
|
|
|
|
@api_view(["GET"])
|
|
@permission_classes([IsAuthenticated])
|
|
def get_requirement(request, pk):
|
|
try:
|
|
requirement = Requirement.objects.get(pk=pk)
|
|
except Requirement.DoesNotExist:
|
|
return Response({"detail": "不存在"}, status=status.HTTP_404_NOT_FOUND)
|
|
from .serializers import RequirementSerializer
|
|
return Response(RequirementSerializer(requirement).data)
|
|
|
|
|
|
@api_view(["POST"])
|
|
@permission_classes([IsAuthenticated])
|
|
@transaction.atomic
|
|
def create_submission(request, pk):
|
|
try:
|
|
requirement = Requirement.objects.get(pk=pk)
|
|
except Requirement.DoesNotExist:
|
|
return Response({"detail": "作业要求不存在"}, status=status.HTTP_404_NOT_FOUND)
|
|
name = (request.data.get("name") or "").strip()
|
|
file = request.FILES.get("file")
|
|
if not file:
|
|
return Response({"detail": "缺少上传文件 file"}, status=status.HTTP_400_BAD_REQUEST)
|
|
submission = Submission.objects.create(
|
|
requirement=requirement,
|
|
name=name or file.name,
|
|
file=file,
|
|
submitted_by=request.user,
|
|
)
|
|
from .tasks import run_submission
|
|
try:
|
|
run_submission(submission.id)
|
|
except Exception:
|
|
logger.exception("检测任务启动失败")
|
|
submission.status = "failed"
|
|
submission.save(update_fields=["status"])
|
|
from .serializers import SubmissionSerializer
|
|
return Response(SubmissionSerializer(submission).data, status=status.HTTP_201_CREATED)
|
|
|
|
|
|
@api_view(["GET"])
|
|
@permission_classes([IsAuthenticated])
|
|
def list_submissions(request):
|
|
qs = Submission.objects.all()
|
|
req_id = request.query_params.get("requirement")
|
|
if req_id:
|
|
qs = qs.filter(requirement_id=req_id)
|
|
from .serializers import SubmissionSerializer
|
|
return Response({"items": SubmissionSerializer(qs, many=True).data})
|
|
|
|
|
|
@api_view(["GET"])
|
|
@permission_classes([IsAuthenticated])
|
|
def get_submission(request, pk):
|
|
try:
|
|
submission = Submission.objects.get(pk=pk)
|
|
except Submission.DoesNotExist:
|
|
return Response({"detail": "不存在"}, status=status.HTTP_404_NOT_FOUND)
|
|
from .serializers import SubmissionSerializer
|
|
return Response(SubmissionSerializer(submission).data)
|
|
|
|
|
|
@api_view(["GET"])
|
|
@permission_classes([IsAuthenticated])
|
|
def get_report(request, pk):
|
|
try:
|
|
submission = Submission.objects.get(pk=pk)
|
|
except Submission.DoesNotExist:
|
|
return Response({"detail": "不存在"}, status=status.HTTP_404_NOT_FOUND)
|
|
if not submission.report:
|
|
return Response({"detail": "报告尚未生成", "status": submission.status},
|
|
status=status.HTTP_202_ACCEPTED)
|
|
return Response(submission.report)
|