Process current month

mysociety · Feb 24, 2024 · ff7bf8c · ff7bf8c
1 parent ca6c754
commit ff7bf8c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/voting_data/commons_votes.py b/src/voting_data/commons_votes.py
@@ -49,6 +49,9 @@ def next(self) -> Month:
     def is_current(self) -> bool:
         return self == self.current_month()
 
+    def is_future(self) -> bool:
+        return self.months_to_current() < 0
+
     def months_to_current(self) -> int:
         current = self.current_month()
         return (current.year - self.year) * 12 + current.month - self.month
@@ -57,7 +60,7 @@ def months_to_current(self) -> int:
     @to_list
     def all_months(cls) -> Iterator[Month]:
         month = Month(2016, 3)
-        while not month.is_current():
+        while not month.is_future():
             yield month
             month = month.next()