添加提交

submission/api.py

@@ -0,0 +1,102 @@
+from typing import List
+from uuid import UUID
+from ninja import Router, Query
+from ninja.errors import HttpError
+from ninja.pagination import paginate
+from django.shortcuts import get_object_or_404
+from django.contrib.auth.decorators import login_required
+
+from account.decorators import admin_required
+from .schemas import (
+    SubmissionFilter,
+    SubmissionIn,
+    SubmissionOut,
+    SubmissionScoreIn,
+    SubmissionScoreOut,
+)
+
+
+from .models import Submission
+from task.models import Task
+from account.models import RoleChoices
+
+router = Router()
+
+
+@router.post("/")
+@login_required
+def create_submission(request, payload: SubmissionIn):
+    """
+    创建一个新的提交
+    """
+    task = get_object_or_404(Task, id=payload.task_id)
+
+    Submission.objects.create(
+        user=request.user,
+        task=task,
+        html=payload.html,
+        css=payload.css,
+        js=payload.js,
+    )
+
+
+@router.get("/", response=List[SubmissionOut])
+@paginate
+def list_submissions(request, filters: SubmissionFilter = Query(...)):
+    """
+    获取提交列表，支持按任务和用户过滤
+    """
+    queryset = Submission.objects.all()
+
+    if filters.task_id:
+        queryset = queryset.filter(task_id=filters.task_id)
+    if filters.task_id:
+        queryset = queryset.filter(task_task_type=filters.task_type)
+    if filters.username:
+        queryset = queryset.filter(user_username=filters.username)
+
+    return [SubmissionOut.list(submission) for submission in queryset]
+
+
+@router.get("/{submission_id}", response=SubmissionOut)
+@login_required
+def get_submission(request, submission_id: UUID):
+    """
+    获取单个提交的详细信息
+    """
+    # 如果是普通用户，只能查看自己的提交
+    if request.user.role == RoleChoices.NORMAL:
+        submission = get_object_or_404(Submission, id=submission_id, user=request.user)
+    else:
+        submission = get_object_or_404(Submission, id=submission_id)
+
+    return SubmissionOut.get(submission)
+
+
+@router.put("/{submission_id}/score", response=SubmissionScoreOut)
+@admin_required
+def update_score(request, submission_id: UUID, payload: SubmissionScoreIn):
+    """
+    给提交打分
+    """
+    if payload.score <= 0:
+        raise HttpError(400, "分数不能为零")
+
+    submission = get_object_or_404(Submission, id=submission_id)
+
+    if submission.score > 0:
+        raise HttpError(400, "该提交已经有分数了")
+    if (
+        request.user.role == RoleChoices.NORMAL
+        and submission.user.id == request.user.id
+    ):
+        raise HttpError(400, "不能自己给自己打分")
+
+    submission.score = payload.score
+    submission.referee = request.user
+    submission.save()
+
+    return {
+        "id": submission.id,
+        "score": submission.score,
+    }