Ticket: remove or 0 when counting tickets

app/core/tickets/cruds_tickets.py

@@ -10,6 +10,7 @@
 
 from app.core.tickets import models_tickets, schemas_tickets
 from app.core.users import schemas_users
+from app.types.exceptions import NoneResultWhenCountingDbError
 
 
 async def get_all_events(db: AsyncSession) -> Sequence[schemas_tickets.EventSimple]:
@@ -711,7 +712,11 @@ async def count_tickets_by_event_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def count_tickets_by_category_id(
@@ -725,7 +730,11 @@ async def count_tickets_by_category_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def count_tickets_by_session_id(
@@ -739,7 +748,11 @@ async def count_tickets_by_session_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def count_valid_checkouts_by_event_id(
@@ -757,7 +770,11 @@ async def count_valid_checkouts_by_event_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def count_valid_checkouts_and_tickets_by_event_id(
@@ -777,7 +794,11 @@ async def count_valid_checkouts_and_tickets_by_event_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def count_valid_checkouts_and_tickets_by_session_id(
@@ -797,7 +818,11 @@ async def count_valid_checkouts_and_tickets_by_session_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def count_valid_checkouts_and_tickets_by_category_id(
@@ -817,7 +842,11 @@ async def count_valid_checkouts_and_tickets_by_category_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def count_valid_checkouts_by_category_id(
@@ -835,7 +864,11 @@ async def count_valid_checkouts_by_category_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def count_valid_checkouts_by_session_id(
@@ -853,7 +886,11 @@ async def count_valid_checkouts_by_session_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def count_answers_by_question_id(
@@ -866,7 +903,11 @@ async def count_answers_by_question_id(
         ),
     )
 
-    return result.scalar() or 0
+    nb = result.scalar()
+    if nb is None:
+        raise NoneResultWhenCountingDbError()
+
+    return nb
 
 
 async def update_event(

app/types/exceptions.py

@@ -237,3 +237,8 @@ def __init__(self, object_name: str, object_id: str | UUID):
         super().__init__(
             f"Object {object_name} with id {object_id} was expected in database but not found",
         )
+
+
+class NoneResultWhenCountingDbError(Exception):
+    def __init__(self):
+        super().__init__("Counting rows in db should never return None")