From 43e9424920ca2f00196f6a5e9c9d65e2d634d2f2 Mon Sep 17 00:00:00 2001 From: Bulat Kurbanov Date: Sun, 14 May 2023 23:12:15 +0200 Subject: [PATCH] Add service count limit --- src/app/views.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/app/views.py b/src/app/views.py index 3019669..764006f 100644 --- a/src/app/views.py +++ b/src/app/views.py @@ -45,6 +45,11 @@ async def delete_service(id: int): @router.post("/", response_model=ServiceDetail) async def register_service(data: ServiceCreate): + user_services_count = await Service.objects.filter(user=data.user).count() + + if user_services_count >= 3: + raise HTTPException(status.HTTP_402_PAYMENT_REQUIRED) + return await Service.objects.create(**data.dict())