From: Dustin Walde <redacted>
Date: Sat, 23 Sep 2023 04:20:06 +0000 (-0700)
Subject: Fix bug from accessing last entry of new category
X-Git-Url: https://git.walde.dev/?a=commitdiff_plain;h=5819855afb8cf1c45246662ecc80299ad2c489b1;p=punch

Fix bug from accessing last entry of new category
---

diff --git a/src/time_sheet.py b/src/time_sheet.py
index cbdccb1..34686eb 100644
--- a/src/time_sheet.py
+++ b/src/time_sheet.py
@@ -76,8 +76,8 @@ class TimeSheet(Puncher):
 
 
     def get_last_entry(self, cat_id: str) -> Optional[Entry]:
-        entries = self.cat_entries[cat_id]
-        if len(entries) == 0:
+        entries = self.cat_entries.get(cat_id)
+        if entries is None or len(entries) == 0:
             return None
 
         return self.entries[entries[-1]]