wip

Scripts/sportstime_parser/models/stadium.py

@@ -60,9 +60,32 @@ class Stadium:
             "timezone": self.timezone,
         }
 
+    def to_canonical_dict(self, primary_team_abbrevs: list[str] | None = None) -> dict:
+        """Convert to canonical dictionary format matching iOS app schema.
+
+        Args:
+            primary_team_abbrevs: List of team abbreviations that play at this stadium.
+                                  If None, defaults to empty list.
+
+        Returns:
+            Dictionary with field names matching JSONCanonicalStadium in BootstrapService.swift
+        """
+        return {
+            "canonical_id": self.id,
+            "name": self.name,
+            "city": self.city,
+            "state": self.state,
+            "latitude": self.latitude,
+            "longitude": self.longitude,
+            "capacity": self.capacity if self.capacity is not None else 0,
+            "sport": self.sport,
+            "primary_team_abbrevs": primary_team_abbrevs or [],
+            "year_opened": self.opened_year,
+        }
+
     @classmethod
     def from_dict(cls, data: dict) -> "Stadium":
-        """Create a Stadium from a dictionary."""
+        """Create a Stadium from a dictionary (internal format)."""
         return cls(
             id=data["id"],
             sport=data["sport"],
@@ -80,6 +103,22 @@ class Stadium:
             timezone=data.get("timezone"),
         )
 
+    @classmethod
+    def from_canonical_dict(cls, data: dict) -> "Stadium":
+        """Create a Stadium from a canonical dictionary (iOS app format)."""
+        return cls(
+            id=data["canonical_id"],
+            sport=data["sport"],
+            name=data["name"],
+            city=data["city"],
+            state=data["state"],
+            country="USA",  # Canonical format doesn't include country
+            latitude=data["latitude"],
+            longitude=data["longitude"],
+            capacity=data.get("capacity"),
+            opened_year=data.get("year_opened"),
+        )
+
     def to_json(self) -> str:
         """Serialize to JSON string."""
         return json.dumps(self.to_dict(), indent=2)
@@ -102,7 +141,10 @@ def save_stadiums(stadiums: list[Stadium], filepath: str) -> None:
 
 
 def load_stadiums(filepath: str) -> list[Stadium]:
-    """Load a list of stadiums from a JSON file."""
+    """Load a list of stadiums from a JSON file (auto-detects format)."""
     with open(filepath, "r", encoding="utf-8") as f:
         data = json.load(f)
+    # Detect format: canonical has "canonical_id", internal has "id"
+    if data and "canonical_id" in data[0]:
+        return [Stadium.from_canonical_dict(d) for d in data]
     return [Stadium.from_dict(d) for d in data]