feat: add support for European Championship

probberechts · Jun 16, 2024 · beb8f5b · beb8f5b
1 parent 1da000f
commit beb8f5b
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 5 deletions.
diff --git a/soccerdata/_config.py b/soccerdata/_config.py
@@ -171,6 +171,15 @@
         "WhoScored": "International - FIFA World Cup",
         "season_code": "single-year",
     },
+    "INT-European Championship": {
+        "FBref": "UEFA European Football Championship",
+        "FotMob": "INT-EURO",
+        "Sofascore": "EURO",
+        "WhoScored": "International - European Championship",
+        "season_start": "Jun",
+        "season_end": "Jul",
+        "season_code": "single-year",
+    },
     "INT-Women's World Cup": {
         "FBref": "FIFA Women's World Cup",
         "FotMob": "INT-Women's World Cup",

diff --git a/soccerdata/fbref.py b/soccerdata/fbref.py
@@ -441,10 +441,14 @@ def read_team_match_stats(  # noqa: C901
                     + f"/{stat_type}"
                 )
             else:  # special case: latest season
-                season_format = "{}-{}".format(
-                    datetime.strptime(skey[:2], "%y").year,  # noqa: DTZ007
-                    datetime.strptime(skey[2:], "%y").year,  # noqa: DTZ007
-                )
+                if SeasonCode.from_league(lkey) == SeasonCode.MULTI_YEAR:
+                    _skey = SeasonCode.MULTI_YEAR.parse(skey)
+                    season_format = "{}-{}".format(
+                        datetime.strptime(_skey[:2], "%y").year,  # noqa: DTZ007
+                        datetime.strptime(_skey[2:], "%y").year,  # noqa: DTZ007
+                    )
+                else:
+                    season_format = SeasonCode.SINGLE_YEAR.parse(skey)
                 url = (
                     FBREF_API
                     + team_url.rsplit("/", 1)[0]
@@ -522,7 +526,13 @@ def read_team_match_stats(  # noqa: C901
             lambda x: x["team"] if x["venue"] == "Away" else x["opponent"], axis=1
         )
         df["game"] = df_tmp.apply(make_game_id, axis=1)
-        return df.set_index(["league", "season", "team", "game"]).sort_index().loc[self.leagues]
+        return (
+            df
+            # .dropna(subset="league")
+            .set_index(["league", "season", "team", "game"])
+            .sort_index()
+            .loc[self.leagues]
+        )
 
     def read_player_season_stats(self, stat_type: str = "standard") -> pd.DataFrame:
         """Retrieve players from the datasource for the selected leagues and seasons.