flibusta-apps/services_manager_server
1
0
Fork 0
mirror of https://github.com/flibusta-apps/services_manager_server.git synced 2025-12-06 12:35:39 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add service count limit

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-05-14 23:12:15 +02:00
parent 492e1087d3
commit 43e9424920
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/app/views.py
View File

@@ -45,6 +45,11 @@ async def delete_service(id: int):
@router.post("/", response_model=ServiceDetail) @router.post("/", response_model=ServiceDetail)
async def register_service(data: ServiceCreate): 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()) return await Service.objects.create(**data.dict())