fix: fixed typechecking issue

ErikBjare · ErikBjare · commit 5d06b0868153 · 2023-10-25T07:22:20.000+02:00
diff --git a/aw_client/client.py b/aw_client/client.py
@@ -21,6 +21,7 @@
 import requests as req
 from aw_core.dirs import get_data_dir
 from aw_core.models import Event
+from aw_transform.heartbeats import heartbeat_merge
 
 from .config import load_config
 from .singleinstance import SingleInstance
@@ -31,10 +32,9 @@
 
 
 def _log_request_exception(e: req.RequestException):
-    r = e.response
     logger.warning(str(e))
     try:
-        d = r.json()
+        d = e.response.json() if e.response else None
         logger.warning(f"Error message received: {d}")
     except json.JSONDecodeError:
         pass
@@ -152,7 +152,7 @@ def get_event(
             event = self._get(endpoint).json()
             return Event(**event)
         except req.exceptions.HTTPError as e:
-            if e.response.status_code == 404:
+            if e.response and e.response.status_code == 404:
                 return None
             else:
                 raise
@@ -230,8 +230,6 @@ def heartbeat(
               the function will in that case always returns None.
         """
 
-        from aw_transform.heartbeats import heartbeat_merge
-
         endpoint = f"buckets/{bucket_id}/heartbeat?pulsetime={pulsetime}"
         _commit_interval = commit_interval or self.commit_interval
 
