Ушел от try except и вынес rollback на уровень сессии

api/src/db/session.py

@@ -15,6 +15,9 @@ async def get_db() -> AsyncGenerator:
     try:
         yield db
         await db.commit()
+    except Exception:
+        await db.rollback()
+        raise
     finally:
         await db.close()
 

api/src/repository/user_repository.py

@@ -105,8 +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]) -> bool:
+    async def set_many_ssp(self, user_id: int, ssp_ids: list[int]) -> None:
-        try:
         for ssp_id in ssp_ids:
             await self.db.execute(
                 text(
@@ -115,13 +114,8 @@ class UserRepository:
                 {"user_id": user_id, "org_unit_id": ssp_id},
             )
         await self.db.commit()
-            return True
-        except Exception:
-            await self.db.rollback()
-            return False
 
-    async def unset_many_ssp(self, user_id: int, ssp_ids: list[int]) -> bool:
+    async def unset_many_ssp(self, user_id: int, ssp_ids: list[int]) -> None:
-        try:
         for ssp_id in ssp_ids:
             await self.db.execute(
                 text(
@@ -130,13 +124,8 @@ class UserRepository:
                 {"user_id": user_id, "org_unit_id": ssp_id},
             )
         await self.db.commit()
-            return True
-        except Exception:
-            await self.db.rollback()
-            return False
 
-    async def clear_many_ssp(self, user_id: int) -> bool:
+    async def clear_many_ssp(self, user_id: int) -> None:
-        try:
         org_unit_ids = await self.get_many_ssp_ids(user_id)
         await self.db.execute(
             text(
@@ -145,10 +134,6 @@ class UserRepository:
             {"user_id": user_id, "org_unit_ids": org_unit_ids},
         )
         await self.db.commit()
-            return True
-        except Exception:
-            await self.db.rollback()
-            return False
 
     async def create(self, user_data: dict) -> AppUser:
         hashed_password = (