fakz9/Fulfillment-Backend
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: forbidden starting and finishing shifts for deleted users

Browse Source
This commit is contained in:
AlexSserb
2024-11-20 13:29:45 +04:00
parent 9707faf60c
commit e873325c34
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
services/work_shifts.py
View File

@@ -31,7 +31,7 @@ class WorkShiftsService(BaseService):
async def start_shift(self, user_id: int) -> StartShiftResponse: async def start_shift(self, user_id: int) -> StartShiftResponse:
employee = await self.session.get(User, user_id) employee = await self.session.get(User, user_id)
if not employee: if not employee or employee.is_deleted:
return StartShiftResponse(ok=False, message=f"Пользователь с ID {user_id} не найден") return StartShiftResponse(ok=False, message=f"Пользователь с ID {user_id} не найден")
work_shift = await self._get_last_work_shift_for_today(user_id) work_shift = await self._get_last_work_shift_for_today(user_id)
@@ -47,7 +47,7 @@ class WorkShiftsService(BaseService):
async def finish_shift(self, user: User, user_id: int) -> FinishShiftResponse: async def finish_shift(self, user: User, user_id: int) -> FinishShiftResponse:
employee = await self.session.get(User, user_id) employee = await self.session.get(User, user_id)
if not employee: if not employee or employee.is_deleted:
return FinishShiftResponse(ok=False, message=f"Пользователь с ID {user_id} не найден") return FinishShiftResponse(ok=False, message=f"Пользователь с ID {user_id} не найден")
work_shift = await self._get_last_work_shift_for_today(user_id) work_shift = await self._get_last_work_shift_for_today(user_id)