mirror of
https://gitlab.com/megazordpobeda/DataRush.git
synced 2026-05-22 23:17:09 +00:00
Андрей Сумин
67 lines
1.5 KiB
Python
67 lines
1.5 KiB
Python
from typing import Literal
|
|
from uuid import UUID
|
|
|
|
from ninja import ModelSchema, Schema
|
|
|
|
from apps.competition.models import Competition, State
|
|
|
|
|
|
class CompetitionOut(ModelSchema):
|
|
id: UUID
|
|
state: Literal["not_started", "started", "finished"]
|
|
|
|
@staticmethod
|
|
def resolve_state(
|
|
self, context
|
|
) -> Literal["not_started", "started", "finished"]:
|
|
if not (
|
|
state := State.objects.filter(
|
|
user=context.get("request").auth, competition=self
|
|
).first()
|
|
):
|
|
return "not_started"
|
|
return state.state
|
|
|
|
class Meta:
|
|
model = Competition
|
|
exclude = ("participants",)
|
|
|
|
|
|
class StateOut(ModelSchema):
|
|
class Meta:
|
|
model = State
|
|
fields = ("state",)
|
|
|
|
|
|
class StateIn(Schema):
|
|
state: Literal["started", "not_started", "finished"]
|
|
|
|
|
|
class CompetitionListInstanceOut(ModelSchema):
|
|
id: UUID
|
|
is_participating: bool
|
|
completed: bool
|
|
|
|
@staticmethod
|
|
def resolve_is_participating(self, context):
|
|
user = context["request"].auth
|
|
return self.participants.filter(id=user.id).exists()
|
|
|
|
@staticmethod
|
|
def resolve_completed(self, context):
|
|
user = context["request"].auth
|
|
return State.objects.filter(
|
|
competition=self, user=user, state="finished"
|
|
).exists()
|
|
|
|
class Meta:
|
|
model = Competition
|
|
fields = (
|
|
"id",
|
|
"title",
|
|
"description",
|
|
"start_date",
|
|
"end_date",
|
|
"image_url",
|
|
)
|