Merge pull request #70 from kaitlynmcgoldrick/fix-compare-offset-date…

…times-error Fix can't compare offset-naive and offset-aware datetimes error
jazzband · Feb 11, 2021 · fee7f42 · fee7f42
2 parents 89f3cf7 + bd4a63f
commit fee7f42
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/icalevents/icalparser.py b/icalevents/icalparser.py
@@ -73,6 +73,11 @@ def __lt__(self, other):
     def __str__(self):
         n = now()
 
+        if not self.start.tzinfo:
+            self.start = normalize(self.start)
+        if not self.end.tzinfo:
+            self.end = normalize(self.end)
+
         # compute time delta description
         if not self.all_day:
             if self.end > n > self.start:

diff --git a/test/test_icalparser.py b/test/test_icalparser.py
@@ -100,3 +100,8 @@ def test_attendee(self):
     def test_organizer(self):
         self.assertIsInstance(self.eventA.organizer, str)
         self.assertIsInstance(self.eventB.organizer, str)
+
+    def test_str(self):
+        self.eventA.start = datetime(year=2017, month=2, day=3, hour=12, minute=5)
+        self.eventA.end = datetime(year=2017, month=2, day=3, hour=15, minute=5)
+        self.assertEqual('2017-02-03 12:05:00+00:00: Event A (ended)', str(self.eventA))