Allow missing matchDetails in MA3 feeds

Co-authored-by: Raf Hermans <[email protected]>
ML-KULeuven · Sep 10, 2023 · d5aa385 · d5aa385
1 parent 72aca85
commit d5aa385
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 36 deletions.
diff --git a/socceraction/data/opta/parsers/ma3_json.py b/socceraction/data/opta/parsers/ma3_json.py
@@ -68,45 +68,35 @@ def extract_games(self) -> Dict[str, Dict[str, Any]]:
             each game in the data stream.
         """
         match_info = self._get_match_info()
-        live_data = self._get_live_data()
+        game_id = assertget(match_info, "id")
         season = assertget(match_info, "tournamentCalendar")
         competition = assertget(match_info, "competition")
         contestant = assertget(match_info, "contestant")
-        venue = assertget(match_info, "venue")
-        game_id = assertget(match_info, "id")
-        match_details = assertget(live_data, "matchDetails")
-        scores = assertget(match_details, "scores")
-        score_total = assertget(scores, "total")
-        home_score = None
-        away_score = None
-        if isinstance(score_total, dict):
-            home_score = assertget(score_total, "home")
-            away_score = assertget(score_total, "away")
-
         game_date = assertget(match_info, "date")[0:10]
         game_time = assertget(match_info, "time")[0:8]
         game_datetime = f"{game_date}T{game_time}"
-        return {
-            game_id: dict(
-                # Fields required by the base schema
-                game_id=game_id,
-                season_id=assertget(season, "id"),
-                competition_id=assertget(competition, "id"),
-                game_day=int(match_info["week"]) if "week" in match_info else None,
-                game_date=datetime.strptime(game_datetime, "%Y-%m-%dT%H:%M:%S"),
-                home_team_id=self._extract_team_id(contestant, "home"),
-                away_team_id=self._extract_team_id(contestant, "away"),
-                # Optional fields
-                home_score=home_score,
-                away_score=away_score,
-                duration=assertget(match_details, "matchLengthMin"),
-                # referee=?
-                venue=assertget(venue, "shortName"),
-                # attendance=?
-                # home_manager=?
-                # away_manager=?
-            )
-        }
+        venue = assertget(match_info, "venue")
+        game_obj = dict(
+            game_id=game_id,
+            competition_id=assertget(competition, "id"),
+            season_id=assertget(season, "id"),
+            game_day=int(match_info["week"]) if "week" in match_info else None,
+            game_date=datetime.strptime(game_datetime, "%Y-%m-%dT%H:%M:%S"),
+            home_team_id=self._extract_team_id(contestant, "home"),
+            away_team_id=self._extract_team_id(contestant, "away"),
+            venue=assertget(venue, "shortName"),
+        )
+        live_data = self._get_live_data()
+        if "matchDetails" in live_data:
+            match_details = assertget(live_data, "matchDetails")
+            if "matchLengthMin" in match_details:
+                game_obj["duration"] = assertget(match_details, "matchLengthMin")
+            if "scores" in match_details:
+                scores = assertget(match_details, "scores")
+                game_obj["home_score"] = assertget(scores, "total")["home"]
+                game_obj["away_score"] = assertget(scores, "total")["away"]
+
+        return {game_id: game_obj}
 
     def extract_teams(self) -> Dict[str, Dict[str, Any]]:
         """Return a dictionary with all available teams.

diff --git a/socceraction/data/opta/schema.py b/socceraction/data/opta/schema.py
@@ -21,11 +21,11 @@ class OptaCompetitionSchema(CompetitionSchema):
 class OptaGameSchema(GameSchema):
     """Definition of a dataframe containing a list of games."""
 
-    home_score: Optional[Series[int]]
+    home_score: Optional[Series[int]] = pa.Field(nullable=True)
     """The final score of the home team."""
-    away_score: Optional[Series[int]]
+    away_score: Optional[Series[int]] = pa.Field(nullable=True)
     """The final score of the away team."""
-    duration: Optional[Series[int]]
+    duration: Optional[Series[int]] = pa.Field(nullable=True)
     """The total duration of the game in minutes."""
     referee: Optional[Series[str]] = pa.Field(nullable=True)
     """The name of the referee."""