mirror of
https://gitlab.com/megazordpobeda/DataRush.git
synced 2026-05-22 23:17:09 +00:00
Андрей Сумин
119 lines
3.6 KiB
Python
119 lines
3.6 KiB
Python
from http import HTTPStatus as status
|
|
from uuid import UUID
|
|
|
|
from django.shortcuts import get_object_or_404
|
|
from ninja import Router
|
|
|
|
from api.v1.ping.schemas import PingOut
|
|
from api.v1.schemas import ForbiddenError, NotFoundError, UnauthorizedError
|
|
from api.v1.task.schemas import (
|
|
TaskOutSchema,
|
|
TaskSubmissionIn,
|
|
TaskSubmissionOut,
|
|
)
|
|
from apps.competition.models import State
|
|
from apps.task.models import (
|
|
CompetetionTaskSumbission,
|
|
Competition,
|
|
CompetitionTask,
|
|
)
|
|
|
|
router = Router(tags=["competition"])
|
|
|
|
|
|
@router.post(
|
|
"competitions/{competition_id}/start",
|
|
description="Start a competition completing (open access to tasks)",
|
|
response={
|
|
status.OK: PingOut,
|
|
status.UNAUTHORIZED: UnauthorizedError,
|
|
status.NOT_FOUND: NotFoundError,
|
|
},
|
|
)
|
|
def start_competition(request, competition_id: UUID) -> PingOut:
|
|
competition = get_object_or_404(Competition, pk=competition_id)
|
|
state_obj, _ = State.objects.update_or_create(
|
|
user=request.auth, competition=competition, state="started"
|
|
)
|
|
return status.OK, PingOut()
|
|
|
|
|
|
@router.get(
|
|
"competitions/{competition_id}/tasks",
|
|
description="Get all tasks of competition (works only if user started competition)",
|
|
response={
|
|
status.OK: list[TaskOutSchema],
|
|
status.UNAUTHORIZED: UnauthorizedError,
|
|
status.FORBIDDEN: ForbiddenError,
|
|
status.NOT_FOUND: NotFoundError,
|
|
},
|
|
)
|
|
def get_competition_tasks(
|
|
request, competition_id: UUID
|
|
) -> list[TaskOutSchema]:
|
|
competition = get_object_or_404(Competition, pk=competition_id)
|
|
state = State.objects.filter(
|
|
user=request.auth, competition=competition, state="started"
|
|
).first()
|
|
if not state:
|
|
return 403, ForbiddenError()
|
|
|
|
return status.OK, CompetitionTask.objects.filter(
|
|
competition=competition
|
|
).all()
|
|
|
|
|
|
@router.get(
|
|
"competitions/{competition_id}/tasks/{task_id}",
|
|
description="Get a task of competition task",
|
|
response={
|
|
status.OK: TaskOutSchema,
|
|
status.UNAUTHORIZED: UnauthorizedError,
|
|
status.FORBIDDEN: ForbiddenError,
|
|
status.NOT_FOUND: NotFoundError,
|
|
},
|
|
)
|
|
def get_task(request, competition_id: str, task_id: str) -> TaskOutSchema: ...
|
|
|
|
|
|
@router.post(
|
|
"competitions/{competition_id}/tasks/{task_id}/submit",
|
|
description="Submit task solution",
|
|
response={
|
|
status.OK: TaskSubmissionOut,
|
|
status.UNAUTHORIZED: UnauthorizedError,
|
|
status.FORBIDDEN: ForbiddenError,
|
|
status.NOT_FOUND: NotFoundError,
|
|
},
|
|
)
|
|
def submit_task(
|
|
request, competition_id: str, task_id: str, submission: TaskSubmissionIn
|
|
) -> PingOut:
|
|
user = request.auth
|
|
competetion = get_object_or_404(Competition, id=competition_id)
|
|
task = get_object_or_404(
|
|
CompetitionTask, competetion=competetion, id=task_id
|
|
)
|
|
|
|
if task.type == CompetitionTask.CompetitionTaskType.INPUT:
|
|
CompetetionTaskSumbission.objects.create(
|
|
user=user,
|
|
task=task,
|
|
status=CompetetionTaskSumbission.StatusChoices.CHECKED,
|
|
result={"correct": submission.content == task.answer_file_path},
|
|
)
|
|
if task.type == CompetitionTask.CompetitionTaskType.REVIEW:
|
|
CompetetionTaskSumbission.objects.create(
|
|
user=user,
|
|
task=task,
|
|
status=CompetetionTaskSumbission.StatusChoices.SENT,
|
|
)
|
|
if task.type == CompetitionTask.CompetitionTaskType.CHECKER:
|
|
CompetetionTaskSumbission.objects.create(
|
|
user=user,
|
|
task=task,
|
|
status=CompetetionTaskSumbission.StatusChoices.CHECKING,
|
|
)
|
|
|
|
return TaskSubmissionOut(id=CompetetionTaskSumbission.id)
|