Add Methods To Check If Event Starts And Ends This Year

Author: alvindera97

events/models.py

@@ -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()

events/tests/test_models.py

@@ -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())