From: Dustin Walde <redacted>
Date: Mon, 15 May 2023 21:01:30 +0000 (-0700)
Subject: Fix first/no entry for category bugs
X-Git-Url: https://git.walde.dev/?a=commitdiff_plain;h=0da85ba5ad64e7dc971565e0e6f14a0ea8e10138;p=punch

Fix first/no entry for category bugs
---

diff --git a/src/time_sheet.py b/src/time_sheet.py
index d3d638a..a95de22 100644
--- a/src/time_sheet.py
+++ b/src/time_sheet.py
@@ -82,7 +82,7 @@ class TimeSheet(Puncher):
                 if category.is_group:
                     print(category.name)
                 else:
-                    entries = self.cat_entries[category.abbr]
+                    entries = self.cat_entries.get(category.abbr, [])
                     if len(entries) == 0:
                         print(category.abbr + ":\t" + category.name)
                     else:
@@ -149,6 +149,9 @@ class TimeSheet(Puncher):
         if time is None:
             time = datetime.now().astimezone()
 
+        if category not in self.cat_entries:
+            self.cat_entries[category] = []
+
         cat_entries = self.cat_entries[category]
         if len(cat_entries) > 0:
             last_entry = self.entries[cat_entries[-1]]