fix: forbidden starting and finishing shifts for deleted users

This commit is contained in:
2024-11-20 13:29:45 +04:00
parent 9707faf60c
commit e873325c34

View File

@@ -31,7 +31,7 @@ class WorkShiftsService(BaseService):
async def start_shift(self, user_id: int) -> StartShiftResponse:
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} не найден")
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:
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} не найден")
work_shift = await self._get_last_work_shift_for_today(user_id)