mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-22 14:13:21 +00:00
functionnal api
This commit is contained in:
parent
51b3ce03a1
commit
27e183c7ab
@ -19,7 +19,6 @@ from ninja_extra.pagination import PageNumberPaginationExtra
|
||||
from ninja_extra.schemas import PaginatedResponseSchema
|
||||
|
||||
from core.api_permissions import CanView, IsOldSubscriber, IsRoot
|
||||
from core.models import User
|
||||
from counter.models import Counter, Permanency
|
||||
from counter.schemas import CounterSchema, PermanencyFilterSchema, PermanencySchema
|
||||
|
||||
@ -28,32 +27,28 @@ from counter.schemas import CounterSchema, PermanencyFilterSchema, PermanencySch
|
||||
class CounterController(ControllerBase):
|
||||
@route.get("", response=list[CounterSchema], permissions=[IsRoot])
|
||||
def fetch_all(self):
|
||||
return Counter.objects.annotate_is_open()
|
||||
return Counter.objects.all()
|
||||
|
||||
@route.get("{counter_id}/", response=CounterSchema, permissions=[CanView])
|
||||
def fetch_one(self, counter_id: int):
|
||||
return self.get_object_or_exception(
|
||||
Counter.objects.annotate_is_open(), pk=counter_id
|
||||
)
|
||||
return self.get_object_or_exception(Counter.objects.all(), pk=counter_id)
|
||||
|
||||
@route.get("bar/", response=list[CounterSchema], permissions=[CanView])
|
||||
def fetch_bars(self):
|
||||
counters = list(Counter.objects.annotate_is_open().filter(type="BAR"))
|
||||
counters = list(Counter.objects.all().filter(type="BAR"))
|
||||
for c in counters:
|
||||
self.check_object_permissions(c)
|
||||
return counters
|
||||
|
||||
|
||||
|
||||
@api_controller("/permanency")
|
||||
class PermanencyController(ControllerBase):
|
||||
@route.get("", response=PaginatedResponseSchema[PermanencySchema], permissions=[IsOldSubscriber])
|
||||
@route.get(
|
||||
"",
|
||||
response=PaginatedResponseSchema[PermanencySchema],
|
||||
permissions=[IsOldSubscriber],
|
||||
exclude_none=True,
|
||||
)
|
||||
@paginate(PageNumberPaginationExtra, page_size=100)
|
||||
def fetch_permanancies(self, filters:Query[PermanencyFilterSchema]):
|
||||
user: User = self.context.request.user
|
||||
if not user.is_root:
|
||||
filters.barmen=None
|
||||
resp = filters.filter(Permanency.objects.values()).distinct()
|
||||
breakpoint()
|
||||
if not user.is_root:
|
||||
for perm in resp:
|
||||
perm.user = None
|
||||
return resp
|
||||
def fetch_permanancies(self, filters: Query[PermanencyFilterSchema]):
|
||||
return filters.filter(Permanency.objects.all()).distinct().order_by("-start")
|
||||
|
@ -3,26 +3,24 @@ from datetime import datetime
|
||||
from ninja import FilterSchema, ModelSchema
|
||||
from pydantic import Field
|
||||
|
||||
from core.models import User
|
||||
from core.schemas import SimpleUserSchema
|
||||
from counter.models import Counter, Permanency
|
||||
|
||||
|
||||
class CounterSchema(ModelSchema):
|
||||
barmen_list: list[SimpleUserSchema]
|
||||
is_open: bool
|
||||
|
||||
class Meta:
|
||||
model = Counter
|
||||
fields = ["id", "name", "type", "club", "products"]
|
||||
|
||||
fields = ["id", "name", "type"]
|
||||
|
||||
|
||||
class PermanencySchema(ModelSchema):
|
||||
counter: CounterSchema
|
||||
|
||||
class Meta:
|
||||
model = Permanency
|
||||
fields = ["user", "counter", "start", "end", "activity"]
|
||||
|
||||
fields = ["start", "end"]
|
||||
|
||||
|
||||
class PermanencyFilterSchema(FilterSchema):
|
||||
start_date: datetime | None = Field(None, q="start__gte")
|
||||
end_date: datetime | None = Field(None, q="end__lte")
|
||||
barmen: set[int] | None = Field(None, q="user_id__in")
|
||||
counter: set[int] | None = Field(None, q="counter_id__in")
|
||||
|
Loading…
Reference in New Issue
Block a user