diff --git a/api/src/repository/user_repository.py b/api/src/repository/user_repository.py index 07ddd94..733ef56 100644 --- a/api/src/repository/user_repository.py +++ b/api/src/repository/user_repository.py @@ -105,7 +105,7 @@ class UserRepository: query = select(UserOrg.org_unit_id).where(UserOrg.user_id == user_id) return (await self.db.execute(query)).scalars().all() - async def set_many_ssp(self, user_id: int, ssp_ids: list[int]) -> None: + async def set_many_ssp(self, user_id: int, ssp_ids: list[int]) -> bool: for ssp_id in ssp_ids: await self.db.execute( text( @@ -114,8 +114,9 @@ class UserRepository: {"user_id": user_id, "org_unit_id": ssp_id}, ) await self.db.commit() + return True - async def unset_many_ssp(self, user_id: int, ssp_ids: list[int]) -> None: + async def unset_many_ssp(self, user_id: int, ssp_ids: list[int]) -> bool: for ssp_id in ssp_ids: await self.db.execute( text( @@ -124,6 +125,7 @@ class UserRepository: {"user_id": user_id, "org_unit_id": ssp_id}, ) await self.db.commit() + return True async def clear_many_ssp(self, user_id: int) -> None: org_unit_ids = await self.get_many_ssp_ids(user_id)