Update legislative session automatically

keyrings/live/blackbox-admins.txt

@@ -1,3 +1,4 @@
 datamade@scrapers-us-municipal
 [email protected]
 [email protected]
+[email protected]

lametro/__init__.py

@@ -3,6 +3,7 @@
 from .bills import LametroBillScraper
 from .people import LametroPersonScraper
 from .events import LametroEventScraper
+from datetime import datetime
 
 
 class Lametro(Jurisdiction):
@@ -16,14 +17,23 @@ class Lametro(Jurisdiction):
         "events": LametroEventScraper,
     }
 
-    legislative_sessions = []
-    for year in range(2014, 2025):
-        session = {
-            "identifier": "{}".format(year),
-            "start_date": "{}-07-01".format(year),
-            "end_date": "{}-06-30".format(year + 1),
-        }
-        legislative_sessions.append(session)
+    @staticmethod
+    def get_legislative_sessions():
+        today = datetime.now()
+        this_year = today.year
+        allowed_years = list(range(2014, this_year))
+
+        if (today.month == 6 and today.day >= 23) or today.month >= 7:
+            allowed_years.append(this_year)
+
+        for year in allowed_years:
+            session = {
+                "identifier": "{}".format(year),
+                "start_date": "{}-07-01".format(year),
+                "end_date": "{}-06-30".format(year + 1),
+            }
+            yield session
+
 
     def get_organizations(self):
         org = Organization(name="Board of Directors", classification="legislature")

lametro/bills.py

@@ -76,7 +76,7 @@ def session(self, action_date) :
         localize = pytz.timezone(self.TIMEZONE).localize
         fmt = '%Y-%m-%d'
 
-        for session in Lametro.legislative_sessions:
+        for session in Lametro.get_legislative_sessions():
             start_datetime = datetime.datetime.strptime(session['start_date'], fmt)
             end_datetime = datetime.datetime.strptime(session['end_date'], fmt)
 

requirements.txt

@@ -8,5 +8,6 @@ pytest-mock==1.10.1
 requests-mock==1.5.2
 requests[security]
 sentry-sdk
+freezegun==1.5.1
 pdfplumber==0.11.3
 pytesseract==0.3.10

tests/test_jurisdiction.py

@@ -0,0 +1,35 @@
+from datetime import datetime
+
+import pytest
+from freezegun import freeze_time
+from lametro import Lametro
+
+
+@pytest.mark.parametrize("test_date", [
+    '2024-06-10',
+    '2024-06-23',
+    '2024-10-01',
+    '2025-05-01',
+    '2025-06-30'
+])
+def test_legislative_session(test_date):
+    '''
+    Test that next fiscal year's legislative sessions are included
+    only when it's the last week of the current one.
+    '''
+    date_format = "%Y-%m-%d"
+    test_year = test_date.split('-')[0]
+    last_week_of_year = datetime.strptime(f"{test_year}-06-23", date_format).date()
+
+    with freeze_time(test_date):
+        fake_now = datetime.now()
+        fake_date = fake_now.date()
+        next_year = str(fake_now.year + 1)  
+
+        sessions = list(Lametro.get_legislative_sessions())
+        latest_session_date = sessions[-1]["end_date"]
+
+        if fake_date < last_week_of_year:
+            assert next_year not in latest_session_date
+        elif fake_date >= last_week_of_year:
+            assert next_year in latest_session_date