github-heatmap 1.3.9__tar.gz → 1.4.0__tar.gz

{github_heatmap-1.3.9 → github_heatmap-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: github_heatmap
-Version: 1.3.9
+Version: 1.4.0
 Summary: Make everything a GitHub svg poster and Skyline!
 Home-page: https://github.com/malinkang/GitHubPoster
 Author: malinkang

{github_heatmap-1.3.9 → github_heatmap-1.4.0}/github_heatmap/loader/notion_loader.py

@@ -6,7 +6,7 @@ import pendulum
 import requests
 import json
 
-from github_heatmap.loader.base_loader import BaseLoader, LoadError
+from github_heatmap.loader.base_loader import BaseLoader
 from github_heatmap.loader.config import NOTION_API_URL, NOTION_API_VERSION
 
 
@@ -72,7 +72,9 @@ class NotionLoader(BaseLoader):
             help="The Notion property name used for tooltip text.",
         )
 
-    def get_api_data(self, start_cursor="", page_size=100, data_list=[]):
+    def get_api_data(self, start_cursor="", page_size=100, data_list=None):
+        if data_list is None:
+            data_list = []
         payload = {
             "page_size": page_size,
             "filter": {
@@ -100,14 +102,23 @@ class NotionLoader(BaseLoader):
             "Authorization": "Bearer " + self.notion_token,
         }
 
-        resp = requests.post(
-            NOTION_API_URL.format(database_id=self.database_id),
-            json=payload,
-            headers=headers,
-        )
+        try:
+            resp = requests.post(
+                NOTION_API_URL.format(database_id=self.database_id),
+                json=payload,
+                headers=headers,
+            )
+        except requests.RequestException:
+            return data_list
+
         if not resp.ok:
-            raise LoadError("Can not get Notion data, please check your config")
-        data = resp.json()
+            # Treat non-OK responses as an empty result set so we can still draw
+            # a heatmap even when the Notion API yields no rows for the period.
+            return data_list
+        try:
+            data = resp.json()
+        except ValueError:
+            return data_list
         results = data["results"]
         next_cursor = data["next_cursor"]
         data_list.extend(results)

{github_heatmap-1.3.9 → github_heatmap-1.4.0}/github_heatmap.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: github_heatmap
-Version: 1.3.9
+Version: 1.4.0
 Summary: Make everything a GitHub svg poster and Skyline!
 Home-page: https://github.com/malinkang/GitHubPoster
 Author: malinkang

{github_heatmap-1.3.9 → github_heatmap-1.4.0}/setup.py

@@ -6,7 +6,7 @@ setup(
     author_email="linkang.ma@gmail.com",
     url="https://github.com/malinkang/GitHubPoster",
     license="MIT",
-    version="1.3.9",
+    version="1.4.0",
     description="Make everything a GitHub svg poster and Skyline!",
     packages=find_packages(),
     include_package_data=True,

{github_heatmap-1.3.9 → github_heatmap-1.4.0}/tests/test_notion_loader.py

@@ -1,3 +1,6 @@
+import requests
+
+from github_heatmap.loader import notion_loader
 from github_heatmap.loader.notion_loader import NotionLoader
 
 
@@ -91,3 +94,42 @@ def test_notion_loader_extract_property_text_variants():
         },
     }
     assert NotionLoader._extract_property_text(rollup_array) == "Note, Alice, Bob"
+
+
+def test_notion_loader_handles_http_error(monkeypatch):
+    loader = build_loader()
+
+    class DummyResp:
+        ok = False
+        status_code = 404
+
+        @staticmethod
+        def json():
+            return {"object": "error"}
+
+    monkeypatch.setattr(
+        notion_loader.requests, "post", lambda *args, **kwargs: DummyResp()
+    )
+
+    tracks, years = loader.get_all_track_data()
+
+    assert years == [2024]
+    assert tracks
+    assert all(value == 0 for value in tracks.values())
+    assert loader.number_list
+
+
+def test_notion_loader_handles_request_exception(monkeypatch):
+    loader = build_loader()
+
+    def fake_post(*args, **kwargs):
+        raise requests.RequestException("boom")
+
+    monkeypatch.setattr(notion_loader.requests, "post", fake_post)
+
+    tracks, years = loader.get_all_track_data()
+
+    assert years == [2024]
+    assert tracks
+    assert all(value == 0 for value in tracks.values())
+    assert loader.number_list