schemas.py 607 octets
Newer Older
Maxence Lambard's avatar
Maxence Lambard a validé
from typing import Optional

from pydantic import BaseModel


class ItemBase(BaseModel):
    name: str
    price: int

    class Config:
        orm_mode = True


class ItemSchema(ItemBase):
    id: int

    class Config:
        orm_mode = True


class AddItemSchema(ItemBase):
    pass


class RemoveItemSchema(BaseModel):
    id: int

    class Config:
        orm_mode = True


class PendingItem(BaseModel):
    id: int
    amount: int

    class Config:
        orm_mode = True


class PurchasedItem(BaseModel):
    item: Optional[ItemSchema]
    amount: int

    class Config:
        orm_mode = True