Proper return type for pusher_throttle.

matrix-org · Oct 24, 2023 · 5153b9a · 5153b9a
1 parent 36559c8
commit 5153b9a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/synapse/storage/databases/main/pusher.py b/synapse/storage/databases/main/pusher.py
@@ -372,7 +372,7 @@ async def get_throttle_params_by_room(
         self, pusher_id: int
     ) -> Dict[str, ThrottleParams]:
         res = cast(
-            List[Tuple[str, int, int]],
+            List[Tuple[str, Optional[int], Optional[int]]],
             await self.db_pool.simple_select_list(
                 "pusher_throttle",
                 {"pusher": pusher_id},
@@ -383,7 +383,9 @@ async def get_throttle_params_by_room(
 
         params_by_room = {}
         for room_id, last_sent_ts, throttle_ms in res:
-            params_by_room[room_id] = ThrottleParams(last_sent_ts, throttle_ms)
+            params_by_room[room_id] = ThrottleParams(
+                last_sent_ts or 0, throttle_ms or 0
+            )
 
         return params_by_room