Merge pull request #1 from wren/cal-heatmap-1759

alichtman · Jun 27, 2024 · a824977 · a824977
2 parents 81f0f6f + 4e90cae
commit a824977
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 30 deletions.
diff --git a/.vscode/launch.json b/.vscode/launch.json
diff --git a/.vscode/settings.json b/.vscode/settings.json
diff --git a/jrnl/datatypes/NestedDict.py b/jrnl/datatypes/NestedDict.py
diff --git a/jrnl/datatypes/__init__.py b/jrnl/datatypes/__init__.py
diff --git a/jrnl/plugins/calendar_heatmap_exporter.py b/jrnl/plugins/calendar_heatmap_exporter.py
@@ -16,7 +16,7 @@
 from jrnl.plugins.util import get_journal_frequency_nested
 
 if TYPE_CHECKING:
-    from jrnl.datatypes import NestedDict
+    from jrnl.plugins.util import NestedDict
     from jrnl.journals import Entry
     from jrnl.journals import Journal
 

diff --git a/jrnl/plugins/util.py b/jrnl/plugins/util.py
@@ -4,12 +4,18 @@
 from collections import Counter
 from typing import TYPE_CHECKING
 
-from jrnl.datatypes import NestedDict
-
 if TYPE_CHECKING:
     from jrnl.journals import Journal
 
 
+class NestedDict(dict):
+    """https://stackoverflow.com/a/74873621/8740440"""
+
+    def __missing__(self, x):
+        self[x] = NestedDict()
+        return self[x]
+
+
 def get_tags_count(journal: "Journal") -> set[tuple[int, str]]:
     """Returns a set of tuples (count, tag) for all tags present in the journal."""
     # Astute reader: should the following line leave you as puzzled as me the first time