(hopefully) fixed bug with overlapping time periods

1 files changed, 6 insertions, 1 deletions

diff --git a/calendar_manager.py b/calendar_manager.py
index 9ef6596..f6ec0db 100644
--- a/calendar_manager.py
+++ b/calendar_manager.py
@@ -4,7 +4,7 @@ import uuid
 from datetime import datetime
 
 class CalendarEntry:
-    def __init__(self, start_date, end_date, keyword, entry_id=None, corrected_start_date=None, corrected_end_date=None, commentary: str = ""):
+    def __init__(self, start_date, end_date, keyword, entry_id=None, corrected_start_date=None, corrected_end_date=None, time_period=None, commentary: str = ""):
         self.id = entry_id if entry_id else str(uuid.uuid4())
         # Convert string dates to datetime if necessary
         self.start_date = (
@@ -20,6 +20,7 @@ class CalendarEntry:
         self.keyword = keyword
         self.corrected_start_date = corrected_start_date
         self.corrected_end_date = corrected_end_date
+        self.time_period = time_period
         self.commentary = commentary                                                        
     def to_dict(self):
         return {
@@ -135,9 +136,13 @@ class CalendarManager:
         for entry in self.entries:
             entry.corrected_start_date = None
             entry.corrected_end_date = None
+            entry.time_period = None
             
         for start_date, end_date, original_id in list_of_events:
             entry = self.get_entry_by_id(original_id)
+            if not entry:
+                continue
+
             entry.corrected_start_date = start_date
             entry.corrected_end_date = end_date