Fix first/no entry for category bugs

diff --git a/src/time_sheet.py b/src/time_sheet.py
index d3d638a44c92827f0638b33d249ccd40317ccf97..a95de22d63a150c7307d220dc436ab3e4365fb1a 100644 (file)
--- 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]]