Fix access denied error

chelnak · Dec 2, 2022 · 75a96bf · 75a96bf
1 parent 7f34d8d
commit 75a96bf
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/status_cake_exporter/_status_cake.py b/status_cake_exporter/_status_cake.py
@@ -117,15 +117,19 @@ def list_maintenance_windows(self) -> list[dict[str, Any]]:
             return response
 
         except ForbiddenException as e:
-            message = json.loads(e.body)["message"]
-            if "Your current plan has no access to this feature" in message:
-                logger.warn(
-                    (
-                        "Your current plan has no access to this feature or your account is "
-                        "using legacy maintenance windows. Skipping maintenance window check."
+            if e.body:
+                message = json.loads(e.body)["message"]
+                if (
+                    message
+                    and "Your current plan has no access to this feature" in message
+                ):
+                    logger.warn(
+                        (
+                            "Your current plan has no access to this feature or your account is "
+                            "using legacy maintenance windows. Skipping maintenance window check."
+                        )
                     )
-                )
-                return []
+                    return []
 
             # re-raise here because it might be genuine.
             raise e

diff --git a/status_cake_exporter/_test_collector.py b/status_cake_exporter/_test_collector.py
@@ -168,8 +168,11 @@ def collect(self):
             yield uptime_gauge
 
         except Exception as e:
+            import traceback
+
             # This should stop the expoter from crashing if there is an error.
             logger.fatal(f"A fatal error occurred: {e}")
+            logger.debug(traceback.format_exc())
 
         finally:
             logger.info("Collector finished.")