Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 5 additions & 6 deletions app/api/decks.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@

import litestar
from litestar.params import FromPath # noqa: TC002
from litestar.plugins.pydantic import PydanticDTO

from app import models, schemas
from app.repositories import CardsRepository, DecksRepository # noqa: TC001
Expand All @@ -11,7 +10,7 @@
@litestar.get("/decks/")
async def list_decks(decks_repository: DecksRepository) -> schemas.Decks:
objects = await decks_repository.get_many()
return schemas.Decks(items=objects) # ty: ignore[invalid-argument-type]
return schemas.Decks.from_models(objects)


@litestar.get("/decks/{deck_id:int}/")
Expand Down Expand Up @@ -39,10 +38,10 @@ async def create_deck(data: schemas.DeckCreate, decks_repository: DecksRepositor
@litestar.get("/decks/{deck_id:int}/cards/")
async def list_cards(deck_id: FromPath[int], cards_repository: CardsRepository) -> schemas.Cards:
objects = await cards_repository.list_for_deck(deck_id)
return schemas.Cards(items=objects) # ty: ignore[invalid-argument-type]
return schemas.Cards.from_models(objects)


@litestar.get("/cards/{card_id:int}/", return_dto=PydanticDTO[schemas.Card])
@litestar.get("/cards/{card_id:int}/")
async def get_card(card_id: FromPath[int], cards_repository: CardsRepository) -> schemas.Card:
instance = await cards_repository.get_one(models.Card.id == card_id)
return schemas.Card.model_validate(instance)
Expand All @@ -53,15 +52,15 @@ async def create_cards(
deck_id: FromPath[int], data: list[schemas.CardCreate], cards_repository: CardsRepository
) -> schemas.Cards:
objects = await cards_repository.add_cards(deck_id, data)
return schemas.Cards(items=objects) # ty: ignore[invalid-argument-type]
return schemas.Cards.from_models(objects)


@litestar.put("/decks/{deck_id:int}/cards/")
async def update_cards(
deck_id: FromPath[int], data: list[schemas.Card], cards_repository: CardsRepository
) -> schemas.Cards:
objects = await cards_repository.upsert_cards(deck_id, data)
return schemas.Cards(items=objects) # ty: ignore[invalid-argument-type]
return schemas.Cards.from_models(objects)


ROUTER: typing.Final = litestar.Router(
Expand Down
22 changes: 18 additions & 4 deletions app/schemas.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,25 @@
from typing import TYPE_CHECKING, Self

import pydantic
from pydantic import BaseModel, PositiveInt


if TYPE_CHECKING:
from collections.abc import Iterable


class Base(BaseModel):
model_config = pydantic.ConfigDict(from_attributes=True)


class Collection[T: Base](Base):
items: list[T]

@classmethod
def from_models(cls, objects: Iterable[object]) -> Self:
return cls.model_validate({"items": list(objects)})


class CardBase(Base):
front: str
back: str | None = None
Expand All @@ -21,8 +35,8 @@ class Card(CardBase):
deck_id: PositiveInt | None = None


class Cards(Base):
items: list[Card]
class Cards(Collection[Card]):
pass


class DeckBase(Base):
Expand All @@ -39,5 +53,5 @@ class Deck(DeckBase):
cards: list[Card] | None


class Decks(Base):
items: list[Deck]
class Decks(Collection[Deck]):
pass