139 lines
4.8 KiB
Python
139 lines
4.8 KiB
Python
from utils.api import APIView
|
|
from account.decorators import login_required
|
|
from flowchart.models import FlowchartSubmission, FlowchartSubmissionStatus
|
|
from flowchart.serializers import (
|
|
CreateFlowchartSubmissionSerializer,
|
|
FlowchartSubmissionSerializer,
|
|
FlowchartSubmissionListSerializer
|
|
)
|
|
from flowchart.tasks import evaluate_flowchart_task
|
|
|
|
class FlowchartSubmissionAPI(APIView):
|
|
@login_required
|
|
def post(self, request):
|
|
"""创建流程图提交"""
|
|
serializer = CreateFlowchartSubmissionSerializer(data=request.data)
|
|
if not serializer.is_valid():
|
|
return self.error(serializer.errors)
|
|
|
|
data = serializer.validated_data
|
|
|
|
# 验证题目存在
|
|
try:
|
|
from problem.models import Problem
|
|
problem = Problem.objects.get(_id=data['problem_id'])
|
|
except Problem.DoesNotExist:
|
|
return self.error("Problem doesn't exist")
|
|
|
|
# 验证题目是否允许流程图提交
|
|
if not problem.allow_flowchart:
|
|
return self.error("This problem does not allow flowchart submission")
|
|
|
|
# 创建提交记录
|
|
submission = FlowchartSubmission.objects.create(
|
|
user=request.user,
|
|
problem=problem,
|
|
mermaid_code=data['mermaid_code'],
|
|
flowchart_data=data.get('flowchart_data', {})
|
|
)
|
|
|
|
# 启动AI评分任务
|
|
evaluate_flowchart_task.send(submission.id)
|
|
|
|
return self.success({
|
|
'submission_id': submission.id,
|
|
'status': 'pending'
|
|
})
|
|
|
|
@login_required
|
|
def get(self, request):
|
|
"""获取流程图提交详情"""
|
|
submission_id = request.GET.get('id')
|
|
if not submission_id:
|
|
return self.error("submission_id is required")
|
|
|
|
try:
|
|
submission = FlowchartSubmission.objects.get(id=submission_id)
|
|
except FlowchartSubmission.DoesNotExist:
|
|
return self.error("Submission doesn't exist")
|
|
|
|
if not submission.check_user_permission(request.user):
|
|
return self.error("No permission for this submission")
|
|
|
|
serializer = FlowchartSubmissionSerializer(submission)
|
|
return self.success(serializer.data)
|
|
|
|
class FlowchartSubmissionListAPI(APIView):
|
|
@login_required
|
|
def get(self, request):
|
|
"""获取流程图提交列表"""
|
|
user_id = request.GET.get('user_id')
|
|
problem_id = request.GET.get('problem_id')
|
|
offset = int(request.GET.get('offset', 0))
|
|
limit = int(request.GET.get('limit', 20))
|
|
|
|
queryset = FlowchartSubmission.objects.select_related('user', 'problem')
|
|
|
|
# 权限过滤
|
|
if not request.user.is_admin_role():
|
|
queryset = queryset.filter(user=request.user)
|
|
|
|
# 其他过滤条件
|
|
if user_id:
|
|
queryset = queryset.filter(user_id=user_id)
|
|
if problem_id:
|
|
queryset = queryset.filter(problem_id=problem_id)
|
|
|
|
total = queryset.count()
|
|
submissions = queryset[offset:offset + limit]
|
|
|
|
serializer = FlowchartSubmissionListSerializer(submissions, many=True)
|
|
|
|
return self.success({
|
|
'results': serializer.data,
|
|
'total': total
|
|
})
|
|
|
|
|
|
class FlowchartSubmissionRetryAPI(APIView):
|
|
@login_required
|
|
def post(self, request):
|
|
"""重新触发AI评分"""
|
|
submission_id = request.data.get('submission_id')
|
|
if not submission_id:
|
|
return self.error("submission_id is required")
|
|
|
|
try:
|
|
submission = FlowchartSubmission.objects.get(id=submission_id)
|
|
except FlowchartSubmission.DoesNotExist:
|
|
return self.error("Submission doesn't exist")
|
|
|
|
# 检查权限
|
|
if not submission.check_user_permission(request.user):
|
|
return self.error("No permission for this submission")
|
|
|
|
# 检查是否可以重新评分
|
|
if submission.status not in [FlowchartSubmissionStatus.FAILED, FlowchartSubmissionStatus.COMPLETED]:
|
|
return self.error("Submission is not in a state that allows retry")
|
|
|
|
# 重置状态并重新启动AI评分
|
|
submission.status = FlowchartSubmissionStatus.PENDING
|
|
submission.ai_score = None
|
|
submission.ai_grade = None
|
|
submission.ai_feedback = None
|
|
submission.ai_suggestions = None
|
|
submission.ai_criteria_details = {}
|
|
submission.processing_time = None
|
|
submission.evaluation_time = None
|
|
submission.save()
|
|
|
|
# 重新启动AI评分任务
|
|
evaluate_flowchart_task.send(submission.id)
|
|
|
|
return self.success({
|
|
'submission_id': submission.id,
|
|
'status': 'pending',
|
|
'message': 'AI evaluation restarted'
|
|
})
|
|
|