Add Methods To Check If Event Starts And Ends This Year

python · JacobCoffee · Sep 26, 2024 · Dec 25, 2020 · Dec 25, 2020 · Dec 25, 2020
commit 1502c108dbb36a1fc94c9fac7eae4185f9f21da5
@@ -181,6 +181,24 @@ def next_time(self):
        except IndexError:
            return None

+    def is_scheduled_to_start_this_year(self) -> bool:
+        current_year: int = datetime.datetime.now().year
+        try:
+            if self.next_time.dt_start.year == current_year:
+                return True
+        except Exception:
+            pass
+        return False
+
+    def is_scheduled_to_end_this_year(self) -> bool:
+        current_year: int = datetime.datetime.now().year
+        try:
+            if self.next_time.dt_end.year == current_year:
+                return True
+        except Exception:
+            pass
+        return False
+
    @property
    def previous_time(self):
        now = timezone.now()

@@ -62,7 +62,6 @@ def test_recurring_event(self):
        self.assertEqual(self.event.next_time.dt_start, recurring_time_dtstart)
        self.assertTrue(rt.valid_dt_end())

-
        rt.begin = now - datetime.timedelta(days=5)
        rt.finish = now - datetime.timedelta(days=3)
        rt.save()
@@ -186,3 +185,25 @@ def test_event_previous_event(self):
        # 'Event.previous_event' can return None if there is no
        # OccurringRule or RecurringRule found.
        self.assertIsNone(self.event.previous_event)
+
+    def test_scheduled_to_start_this_year_method(self):
+        now = seconds_resolution(timezone.now())
+
+        occurring_time_dtstart = now + datetime.timedelta(days=3)
+        OccurringRule.objects.create(
+            event=self.event,
+            dt_start=occurring_time_dtstart,
+            dt_end=occurring_time_dtstart
+        )
+        self.assertTrue(self.event.is_scheduled_to_start_this_year())
+
+    def test_scheduled_to_end_this_year_method(self):
+        now = seconds_resolution(timezone.now())
+
+        occurring_time_dtstart = now + datetime.timedelta(days=3)
+        OccurringRule.objects.create(
+            event=self.event,
+            dt_start=occurring_time_dtstart,
+            dt_end=occurring_time_dtstart
+        )
+        self.assertTrue(self.event.is_scheduled_to_end_this_year())