103 lines
2.7 KiB
Python
103 lines
2.7 KiB
Python
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,
|
|
}
|