mirror of
https://github.com/ae-utbm/sith.git
synced 2024-12-22 07:41:14 +00:00
39 lines
907 B
Python
39 lines
907 B
Python
from typing import Annotated
|
|
|
|
from annotated_types import MinLen
|
|
from ninja import Field, FilterSchema, ModelSchema
|
|
|
|
from core.schemas import SimpleUserSchema
|
|
from counter.models import Counter, Customer, Product
|
|
|
|
|
|
class CounterSchema(ModelSchema):
|
|
barmen_list: list[SimpleUserSchema]
|
|
is_open: bool
|
|
|
|
class Meta:
|
|
model = Counter
|
|
fields = ["id", "name", "type", "club", "products"]
|
|
|
|
|
|
class CustomerSchema(ModelSchema):
|
|
class Meta:
|
|
model = Customer
|
|
fields = ["user", "account_id", "amount", "recorded_products"]
|
|
|
|
|
|
class CounterFilterSchema(FilterSchema):
|
|
search: Annotated[str, MinLen(1)] = Field(None, q="name__icontains")
|
|
|
|
|
|
class SimplifiedCounterSchema(ModelSchema):
|
|
class Meta:
|
|
model = Counter
|
|
fields = ["id", "name"]
|
|
|
|
|
|
class ProductSchema(ModelSchema):
|
|
class Meta:
|
|
model = Product
|
|
fields = ["id", "name", "code"]
|