from datetime import date, datetime from typing import List from django.shortcuts import get_object_or_404 from django.utils.timezone import now as django_timezone_now from ninja import Field, ModelSchema, NinjaAPI, Router, Schema from games.models import PlayEvent api = NinjaAPI() playevent_router = Router() NOW_FACTORY = django_timezone_now class PlayEventIn(Schema): game_id: int started: date | None = None ended: date | None = None note: str = "" days_to_finish: int | None = None class AutoPlayEventIn(ModelSchema): class Meta: model = PlayEvent fields = ["game", "started", "ended", "note"] class UpdatePlayEventIn(Schema): started: date | None = None ended: date | None = None note: str = "" class PlayEventOut(Schema): id: int game: str = Field(..., alias="game.name") started: date | None = None ended: date | None = None days_to_finish: int | None = None note: str = "" updated_at: datetime created_at: datetime @playevent_router.get("/", response=List[PlayEventOut]) def list_playevents(request): return PlayEvent.objects.all() @playevent_router.post("/", response={201: PlayEventOut}) def create_playevent(request, payload: PlayEventIn): playevent = PlayEvent.objects.create(**payload.dict()) return playevent @playevent_router.get("/{playevent_id}", response=PlayEventOut) def get_playevent(request, playevent_id: int): playevent = get_object_or_404(PlayEvent, id=playevent_id) return playevent @playevent_router.patch("/{playevent_id}", response=PlayEventOut) def partial_update_playevent(request, playevent_id: int, payload: UpdatePlayEventIn): playevent = get_object_or_404(PlayEvent, id=playevent_id) for attr, value in payload.dict(exclude_unset=True).items(): setattr(playevent, attr, value) playevent.save() return playevent @playevent_router.delete("/{playevent_id}", response={204: None}) def delete_playevent(request, playevent_id: int): playevent = get_object_or_404(PlayEvent, id=playevent_id) playevent.delete() return 204, None api.add_router("/playevent", playevent_router)