From e873325c34f18a23240bacf47f5bbbb64d0b459a Mon Sep 17 00:00:00 2001 From: AlexSserb Date: Wed, 20 Nov 2024 13:29:45 +0400 Subject: [PATCH] fix: forbidden starting and finishing shifts for deleted users --- services/work_shifts.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/services/work_shifts.py b/services/work_shifts.py index 3355021..6d23206 100644 --- a/services/work_shifts.py +++ b/services/work_shifts.py @@ -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)