wip

Scripts/sportstime_parser/models/game.py

@@ -3,6 +3,7 @@
 from dataclasses import dataclass, field
 from datetime import datetime
 from typing import Optional
+from zoneinfo import ZoneInfo
 import json
 
 
@@ -64,9 +65,53 @@ class Game:
             "raw_stadium": self.raw_stadium,
         }
 
+    def to_canonical_dict(
+        self,
+        stadium_timezone: str,
+        is_playoff: bool = False,
+        broadcast: Optional[str] = None,
+    ) -> dict:
+        """Convert to canonical dictionary format matching iOS app schema.
+
+        Args:
+            stadium_timezone: IANA timezone of the stadium (e.g., 'America/Chicago')
+            is_playoff: Whether this is a playoff game
+            broadcast: Broadcast network info (e.g., 'ESPN')
+
+        Returns:
+            Dictionary with field names matching JSONCanonicalGame in BootstrapService.swift
+        """
+        # Convert game_date to UTC
+        if self.game_date.tzinfo is None:
+            # Localize naive datetime to stadium timezone first
+            local_tz = ZoneInfo(stadium_timezone)
+            local_dt = self.game_date.replace(tzinfo=local_tz)
+        else:
+            local_dt = self.game_date
+
+        utc_dt = local_dt.astimezone(ZoneInfo("UTC"))
+
+        # Format season as string (e.g., 2025 -> "2025-26" for NBA/NHL, "2025" for MLB)
+        if self.sport in ("nba", "nhl"):
+            season_str = f"{self.season}-{str(self.season + 1)[-2:]}"
+        else:
+            season_str = str(self.season)
+
+        return {
+            "canonical_id": self.id,
+            "sport": self.sport,
+            "season": season_str,
+            "game_datetime_utc": utc_dt.strftime("%Y-%m-%dT%H:%M:%SZ"),
+            "home_team_canonical_id": self.home_team_id,
+            "away_team_canonical_id": self.away_team_id,
+            "stadium_canonical_id": self.stadium_id,
+            "is_playoff": is_playoff,
+            "broadcast": broadcast,
+        }
+
     @classmethod
     def from_dict(cls, data: dict) -> "Game":
-        """Create a Game from a dictionary."""
+        """Create a Game from a dictionary (internal format)."""
         game_date = data["game_date"]
         if isinstance(game_date, str):
             game_date = datetime.fromisoformat(game_date)
@@ -89,6 +134,26 @@ class Game:
             raw_stadium=data.get("raw_stadium"),
         )
 
+    @classmethod
+    def from_canonical_dict(cls, data: dict) -> "Game":
+        """Create a Game from a canonical dictionary (iOS app format)."""
+        game_date = datetime.fromisoformat(data["game_datetime_utc"])
+
+        # Parse season string (e.g., "2025-26" -> 2025, or "2025" -> 2025)
+        season_str = data["season"]
+        season = int(season_str.split("-")[0])
+
+        return cls(
+            id=data["canonical_id"],
+            sport=data["sport"],
+            season=season,
+            home_team_id=data["home_team_canonical_id"],
+            away_team_id=data["away_team_canonical_id"],
+            stadium_id=data["stadium_canonical_id"],
+            game_date=game_date,
+            status="scheduled",
+        )
+
     def to_json(self) -> str:
         """Serialize to JSON string."""
         return json.dumps(self.to_dict(), indent=2)
@@ -106,7 +171,10 @@ def save_games(games: list[Game], filepath: str) -> None:
 
 
 def load_games(filepath: str) -> list[Game]:
-    """Load a list of games from a JSON file."""
+    """Load a list of games 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" and "game_datetime_utc", internal has "id"
+    if data and "canonical_id" in data[0] and "game_datetime_utc" in data[0]:
+        return [Game.from_canonical_dict(d) for d in data]
     return [Game.from_dict(d) for d in data]