🐛 (api.py): Raise correct error when timeout

lucasheld · Aug 10, 2023 · 3c9dece · 3c9dece
1 parent be97a4f
commit 3c9dece
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/uptime_kuma_api/api.py b/uptime_kuma_api/api.py
@@ -500,11 +500,14 @@ def _get_event_data(self, event) -> Any:
         return deepcopy(self._event_data[event].copy())
 
     def _call(self, event, data=None) -> Any:
-        r = self.sio.call(event, data, timeout=self.timeout)
-        if isinstance(r, dict) and "ok" in r:
-            if not r["ok"]:
-                raise UptimeKumaException(r.get("msg"))
+        try:
+            r = self.sio.call(event, data, timeout=self.timeout)
             r.pop("ok")
+        except socketio.exceptions.TimeoutError:
+            raise Timeout(f"Timed out while waiting for event {event}")
+        except Exception as e:
+            raise UptimeKumaException(r.get("msg"))
+
         return r
 
     # event handlers