correct types and move large const into utils

Signed-off-by: John Seekins <[email protected]>

Author: johnseekins

enricher.py

@@ -8,6 +8,7 @@
 import time
 from urllib.parse import quote
 from utils import (
+    facility_sheet_header,
     logger,
     session,
 )
@@ -19,38 +20,6 @@
 WIKIPEDIA_DELAY = 0.5  # Be respectful to Wikipedia
 WIKIDATA_DELAY = 0.5  # Be respectful to Wikidata
 
-# extracted ADP sheet header list 2025-09-07
-facility_sheet_header = [
-    "Name",
-    "Address",
-    "City",
-    "State",
-    "Zip",
-    "AOR",
-    "Type Detailed",
-    "Male/Female",
-    "FY25 ALOS",
-    "Level A",
-    "Level B",
-    "Level C",
-    "Level D",
-    "Male Crim",
-    "Male Non-Crim",
-    "Female Crim",
-    "Female Non-Crim",
-    "ICE Threat Level 1",
-    "ICE Threat Level 2",
-    "ICE Threat Level 3",
-    "No ICE Threat Level",
-    "Mandatory",
-    "Guaranteed Minimum",
-    "Last Inspection Type",
-    "Last Inspection End Date",
-    "Pending FY25 Inspection",
-    "Last Inspection Standard",
-    "Last Final Rating",
-]
-
 
 class ExternalDataEnricher(object):
     def __init__(self):
@@ -67,7 +36,7 @@ def _download_sheet(self) -> None:
                     if chunk:
                         f.write(chunk)
 
-    def _load_sheet(self) -> polars.DataFrame:
+    def _load_sheet(self) -> dict:
         """Convert the detentionstats sheet data into something we can update our facilities with"""
         self._download_sheet()
         df = polars.read_excel(

schemas.py

@@ -8,7 +8,7 @@
 }
 
 # default keys to "false"-y values so we can merge easier
-facility_schema = {
+facility_schema: dict = {
     "address": {
         "administrative_area": "",
         "country": "",

utils.py

@@ -21,6 +21,38 @@
 default_timestamp = "1970-01-01T00:00:00-+0000"
 timestamp_format = "%Y-%m-%dT%H:%M:%S-%z"
 
+# extracted ADP sheet header list 2025-09-07
+facility_sheet_header = [
+    "Name",
+    "Address",
+    "City",
+    "State",
+    "Zip",
+    "AOR",
+    "Type Detailed",
+    "Male/Female",
+    "FY25 ALOS",
+    "Level A",
+    "Level B",
+    "Level C",
+    "Level D",
+    "Male Crim",
+    "Male Non-Crim",
+    "Female Crim",
+    "Female Non-Crim",
+    "ICE Threat Level 1",
+    "ICE Threat Level 2",
+    "ICE Threat Level 3",
+    "No ICE Threat Level",
+    "Mandatory",
+    "Guaranteed Minimum",
+    "Last Inspection Type",
+    "Last Inspection End Date",
+    "Pending FY25 Inspection",
+    "Last Inspection Standard",
+    "Last Final Rating",
+]
+
 
 def _flatdict(d: dict, parent_key: str = "", sep: str = ".") -> dict:
     """flatten a nested dictionary for nicer printing in CSV"""