sportsball 0.3.15__tar.gz → 0.3.16__tar.gz

{sportsball-0.3.15/sportsball.egg-info → sportsball-0.3.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: sportsball
-Version: 0.3.15
+Version: 0.3.16
 Summary: A library for pulling in and normalising sports stats.
 Home-page: https://github.com/8W9aG/sportsball
 Author: Will Sackfield
@@ -164,6 +164,7 @@ A representation of a team within a game.
 * **field_goals**: The sum of the field goals made by the team in the game.
 * **field_goals_attempted**: The sum of the field goals attempted by the team in the game.
 * **offensive_rebounds**: The number of rebounds during offense by the team in the game.
+* **assists**: The number of times the player on the team made a pass that resulted in a field goal in the game.
 
 #### Player
 
@@ -177,6 +178,7 @@ A representation of a player within a team within a game.
 * **field_goals**: The number of field goals the player made in the game.
 * **field_goals_attempted**: The number of field goal attempts the player made in the game.
 * **offensive_rebounds**: The number of rebounds during offense by the player made in the game.
+* **assists**: The number of times the player made a pass that resulted in a field goal in the game.
 
 #### Odds
 

{sportsball-0.3.15 → sportsball-0.3.16}/README.md

@@ -111,6 +111,7 @@ A representation of a team within a game.
 * **field_goals**: The sum of the field goals made by the team in the game.
 * **field_goals_attempted**: The sum of the field goals attempted by the team in the game.
 * **offensive_rebounds**: The number of rebounds during offense by the team in the game.
+* **assists**: The number of times the player on the team made a pass that resulted in a field goal in the game.
 
 #### Player
 
@@ -124,6 +125,7 @@ A representation of a player within a team within a game.
 * **field_goals**: The number of field goals the player made in the game.
 * **field_goals_attempted**: The number of field goal attempts the player made in the game.
 * **offensive_rebounds**: The number of rebounds during offense by the player made in the game.
+* **assists**: The number of times the player made a pass that resulted in a field goal in the game.
 
 #### Odds
 

{sportsball-0.3.15 → sportsball-0.3.16}/setup.py

@@ -23,7 +23,7 @@ def install_requires() -> typing.List[str]:
 
 setup(
     name='sportsball',
-    version='0.3.15',
+    version='0.3.16',
     description='A library for pulling in and normalising sports stats.',
     long_description=long_description,
     long_description_content_type='text/markdown',

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/__init__.py

@@ -1,3 +1,3 @@
 """The main module for sportsball."""
 
-__VERSION__ = "0.3.15"
+__VERSION__ = "0.3.16"

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/afl/afltables/afl_afltables_player_model.py

@@ -28,6 +28,7 @@ def _create_afl_afltables_player_model(
         field_goals=None,
         field_goals_attempted=None,
         offensive_rebounds=None,
+        assists=None,
     )
 
 

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/combined/combined_league_model.py

@@ -3,7 +3,7 @@
 import logging
 import multiprocessing
 from multiprocessing import Pool
-from typing import Iterator
+from typing import Any, Iterator
 
 import requests_cache
 
@@ -14,9 +14,13 @@ from ..league_model import LeagueModel
 from .combined_game_model import create_combined_game_model
 
 
-def _produce_league_games(league_model: LeagueModel) -> list[GameModel]:
+def _produce_league_games(league_model: LeagueModel) -> list[dict[str, Any]]:
     setup_logger()
-    return list(league_model.games)
+    try:
+        return [x.model_dump() for x in league_model.games]
+    except Exception as exc:
+        logging.error("Exception: %s", str(exc))
+        raise ValueError(str(exc))
 
 
 class CombinedLeagueModel(LeagueModel):
@@ -52,7 +56,20 @@ class CombinedLeagueModel(LeagueModel):
         for league_model in self._league_models:
             league_model.clear_session()
         with Pool(min(multiprocessing.cpu_count(), len(self._league_models))) as p:
-            game_lists = p.map(_produce_league_games, self._league_models)
+            # We want to terminate immediately if any of our runners runs into trouble.
+            def _error_callback(exc: BaseException) -> None:
+                p.terminate()
+                raise ValueError(str(exc))
+
+            results = [
+                p.apply_async(
+                    _produce_league_games, args=(x,), error_callback=_error_callback
+                )
+                for x in self._league_models
+            ]
+            game_lists = [
+                [GameModel.model_validate(y) for y in x.get()] for x in results
+            ]
 
         for game_list in game_lists:
             for game_model in game_list:

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/combined/combined_player_model.py

@@ -15,6 +15,7 @@ def create_combined_player_model(
     field_goals = None
     field_goals_attempted = None
     offensive_rebounds = None
+    assists = None
     for player_model in player_models:
         player_model_jersey = player_model.jersey
         if player_model_jersey is not None:
@@ -37,6 +38,9 @@ def create_combined_player_model(
         player_model_offensive_rebounds = player_model.offensive_rebounds
         if player_model_offensive_rebounds is not None:
             offensive_rebounds = player_model_offensive_rebounds
+        player_model_assists = player_model.assists
+        if player_model_assists is not None:
+            assists = player_model_assists
     return PlayerModel(
         identifier=identifier,
         jersey=jersey,
@@ -46,4 +50,5 @@ def create_combined_player_model(
         field_goals=field_goals,
         field_goals_attempted=field_goals_attempted,
         offensive_rebounds=offensive_rebounds,
+        assists=assists,
     )

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/espn/espn_player_model.py

@@ -37,6 +37,7 @@ def _create_espn_player_model(
         field_goals=None,
         field_goals_attempted=None,
         offensive_rebounds=None,
+        assists=None,
     )
 
 

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/google/google_address_model.py

@@ -973,6 +973,60 @@ GOLDEN_1_CENTER = SportsballGeocodeTuple(
     housenumber="500",
     country="USA",
 )
+DELTA_CENTER = SportsballGeocodeTuple(
+    city="Salt Lake City",
+    state="UT",
+    postal="",
+    lat=40.768333,
+    lng=-111.901111,
+    housenumber="301",
+    country="USA",
+)
+ATT_CENTER = SportsballGeocodeTuple(
+    city="San Antonio",
+    state="TX",
+    postal="",
+    lat=29.426944,
+    lng=-98.4375,
+    housenumber="1",
+    country="USA",
+)
+FTX_ARENA = SportsballGeocodeTuple(
+    city="Miami",
+    state="FL",
+    postal="",
+    lat=25.781389,
+    lng=-80.188056,
+    housenumber="601",
+    country="USA",
+)
+VIVINT_ARENA = SportsballGeocodeTuple(
+    city="Salt Lake City",
+    state="UT",
+    postal="",
+    lat=40.768333,
+    lng=-111.901111,
+    housenumber="301",
+    country="USA",
+)
+CHESAPEAKE_ENERGY_ARENA = SportsballGeocodeTuple(
+    city="Oklahoma City",
+    state="OK",
+    postal="",
+    lat=35.463333,
+    lng=-97.515,
+    housenumber="100",
+    country="USA",
+)
+ESPN_WIDE_WORLD_OF_SPORTS_COMPLEX = SportsballGeocodeTuple(
+    city="Lake Buena Vista",
+    state="FL",
+    postal="32830",
+    lat=28.337083,
+    lng=-81.556,
+    housenumber="1375",
+    country="USA",
+)
 _CACHED_GEOCODES: dict[str, Any] = {
     "S.C.G. - Australia": SCG,
     "Victoria Park - Australia": SportsballGeocodeTuple(
@@ -6288,15 +6342,7 @@ _CACHED_GEOCODES: dict[str, Any] = {
         housenumber="200",
         country="USA",
     ),
-    "Vivint Arena, Salt Lake City, Utah": SportsballGeocodeTuple(
-        city="Salt Lake City",
-        state="UT",
-        postal="",
-        lat=40.768333,
-        lng=-111.901111,
-        housenumber="301",
-        country="USA",
-    ),
+    "Vivint Arena, Salt Lake City, Utah": VIVINT_ARENA,
     "Rock Hill Sports & Event Center, Rock Hill, South Carolina": SportsballGeocodeTuple(
         city="Rock Hill",
         state="SC",
@@ -6487,24 +6533,8 @@ _CACHED_GEOCODES: dict[str, Any] = {
     "Pepsi Center - Denver, Colorado, U.S. - United States": PEPSI_CENTRE,
     "Staples Center": STAPLES_CENTRE,
     "American Airlines Arena": AMERICAN_AIRLINES_ARENA,
-    "Chesapeake Energy Arena - Oklahoma City, Oklahoma - United States": SportsballGeocodeTuple(
-        city="Oklahoma City",
-        state="OK",
-        postal="",
-        lat=35.463333,
-        lng=-97.515,
-        housenumber="100",
-        country="USA",
-    ),
-    "AT&T Center - San Antonio, Texas - United States": SportsballGeocodeTuple(
-        city="San Antonio",
-        state="TX",
-        postal="",
-        lat=29.426944,
-        lng=-98.4375,
-        housenumber="1",
-        country="USA",
-    ),
+    "Chesapeake Energy Arena - Oklahoma City, Oklahoma - United States": CHESAPEAKE_ENERGY_ARENA,
+    "AT&T Center - San Antonio, Texas - United States": ATT_CENTER,
     "Oakland Arena - Oakland, California - United States": SportsballGeocodeTuple(
         city="Oakland",
         state="CA",
@@ -6514,15 +6544,7 @@ _CACHED_GEOCODES: dict[str, Any] = {
         housenumber="7000",
         country="USA",
     ),
-    "ESPN Wide World of Sports Complex": SportsballGeocodeTuple(
-        city="Lake Buena Vista",
-        state="FL",
-        postal="32830",
-        lat=28.337083,
-        lng=-81.556,
-        housenumber="1375",
-        country="USA",
-    ),
+    "ESPN Wide World of Sports Complex": ESPN_WIDE_WORLD_OF_SPORTS_COMPLEX,
     "International Amphitheatre": SportsballGeocodeTuple(
         city="Chicago",
         state="IL",
@@ -6798,15 +6820,7 @@ _CACHED_GEOCODES: dict[str, Any] = {
     ),
     "Target Center - Minneapolis, Minnesota - United States": TARGET_CENTER,
     "Thomas & Mack Center - Paradise, NV - United States": THOMAS_AND_MACK_CENTER,
-    "Delta Center - Salt Lake City, Utah - United States": SportsballGeocodeTuple(
-        city="Salt Lake City",
-        state="UT",
-        postal="",
-        lat=40.768333,
-        lng=-111.901111,
-        housenumber="301",
-        country="USA",
-    ),
+    "Delta Center - Salt Lake City, Utah - United States": DELTA_CENTER,
     "Yokohama Arena": SportsballGeocodeTuple(
         city="Kanagawa",
         state="",
@@ -6829,15 +6843,7 @@ _CACHED_GEOCODES: dict[str, Any] = {
     "Amway Arena": ORLANDO_ARENA,
     "TD Garden - Boston, Massachusetts - United States": TD_GARDEN,
     "American Airlines Center - Dallas, Texas, USA - United States": AMERICAN_AIRLINES_ARENA,
-    "FTX Arena": SportsballGeocodeTuple(
-        city="Miami",
-        state="FL",
-        postal="",
-        lat=25.781389,
-        lng=-80.188056,
-        housenumber="601",
-        country="USA",
-    ),
+    "FTX Arena": FTX_ARENA,
     "Rose Garden": SportsballGeocodeTuple(
         city="Portland",
         state="OR",
@@ -6981,6 +6987,13 @@ _CACHED_GEOCODES: dict[str, Any] = {
         housenumber="3930",
         country="USA",
     ),
+    "Delta Center, Salt Lake City, Utah": DELTA_CENTER,
+    "AT&T Center, San Antonio, Texas": ATT_CENTER,
+    "FTX Arena, Miami, Florida": FTX_ARENA,
+    "Vivint Smart Home Arena, Salt Lake City, Utah": VIVINT_ARENA,
+    "Chesapeake Energy Arena, Oklahoma City, Oklahoma": CHESAPEAKE_ENERGY_ARENA,
+    "The Arena, Bay Lake, Florida": ESPN_WIDE_WORLD_OF_SPORTS_COMPLEX,
+    "HP Field House, Bay Lake, Florida": ESPN_WIDE_WORLD_OF_SPORTS_COMPLEX,
 }
 
 

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/nba/nba/nba_nba_player_model.py

@@ -37,6 +37,7 @@ def _create_nba_nba_player_model(
         field_goals=None,
         field_goals_attempted=None,
         offensive_rebounds=None,
+        assists=None,
     )
 
 

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/nba/nba/nba_nba_team_model.py

@@ -48,6 +48,7 @@ def _create_nba_nba_team_model(
         field_goals=row["FGM" + suffix],
         field_goals_attempted=row["FGA" + suffix],
         offensive_rebounds=row["OREB" + suffix],
+        assists=row["AST" + suffix],
     )
 
 

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/player_model.py

@@ -13,6 +13,7 @@ PLAYER_FUMBLES_LOST_COLUMN: Literal["fumbles_lost"] = "fumbles_lost"
 FIELD_GOALS_COLUMN: Literal["field_goals"] = "field_goals"
 FIELD_GOALS_ATTEMPTED_COLUMN: Literal["field_goals_attempted"] = "field_goals_attempted"
 OFFENSIVE_REBOUNDS_COLUMN: Literal["offensive_rebounds"] = "offensive_rebounds"
+ASSISTS_COLUMN: Literal["assists"] = "assists"
 
 
 class PlayerModel(BaseModel):
@@ -54,3 +55,8 @@ class PlayerModel(BaseModel):
         json_schema_extra={TYPE_KEY: FieldType.LOOKAHEAD},
         alias=OFFENSIVE_REBOUNDS_COLUMN,
     )
+    assists: int | None = Field(
+        ...,
+        json_schema_extra={TYPE_KEY: FieldType.LOOKAHEAD},
+        alias=ASSISTS_COLUMN,
+    )

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/sportsdb/sportsdb_league_model.py

@@ -103,18 +103,19 @@ class SportsDBLeagueModel(LeagueModel):
                             except ValueError:
                                 pass
                         case SeasonType.REGULAR:
-                            try:
-                                for count, round_str in enumerate(range(1, 125)):
-                                    yield from self._produce_games(
-                                        str(round_str),
-                                        count,
-                                        self._league_id,
-                                        season_year,
-                                        season_type,
-                                        pbar,
-                                    )
-                            except ValueError:
-                                pass
+                            if self.league != League.NBA:
+                                try:
+                                    for count, round_str in enumerate(range(1, 125)):
+                                        yield from self._produce_games(
+                                            str(round_str),
+                                            count,
+                                            self._league_id,
+                                            season_year,
+                                            season_type,
+                                            pbar,
+                                        )
+                                except ValueError:
+                                    pass
                         case SeasonType.POSTSEASON:
                             for count, round_str in enumerate(
                                 [125, 150, 160, 170, 180, 200]

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/sportsreference/sportsreference_game_model.py

@@ -1,12 +1,12 @@
 """Sports Reference game model."""
 
-# pylint: disable=too-many-locals,too-many-statements,unused-argument
+# pylint: disable=too-many-locals,too-many-statements,unused-argument,protected-access
 import datetime
 import io
 import logging
-import time
 import urllib.parse
 
+import dateutil
 import pandas as pd
 import pytest_is_running
 import requests_cache
@@ -36,10 +36,19 @@ def _create_sportsreference_game_model(
         raise ValueError("scorebox_meta_div is not a Tag.")
 
     in_divs = scorebox_meta_div.find_all("div")
-    in_div = in_divs[0]
+    current_in_div_idx = 0
+    in_div = in_divs[current_in_div_idx]
     in_div_text = in_div.get_text().strip()
-    dt = parse(in_div_text)
-    venue_div = in_divs[1]
+    current_in_div_idx += 1
+    if "Tournament" in in_div_text:
+        in_div_text = in_divs[1].get_text().strip()
+        current_in_div_idx += 1
+    try:
+        dt = parse(in_div_text)
+    except dateutil.parser._parser.ParserError as exc:  # type: ignore
+        logging.error("Failed to parse date for URL: %s", url)
+        raise exc
+    venue_div = in_divs[current_in_div_idx]
     venue_name = venue_div.get_text().strip()
     scorebox_div = soup.find("div", class_="scorebox")
     if not isinstance(scorebox_div, Tag):
@@ -62,6 +71,7 @@ def _create_sportsreference_game_model(
     fg = {}
     fga = {}
     offensive_rebounds = {}
+    assists = {}
     for df in dfs:
         if df.index.nlevels > 1:
             df.columns = df.columns.get_level_values(1)
@@ -79,6 +89,10 @@ def _create_sportsreference_game_model(
                 orebs = df["OREB"].tolist()
                 for idx, player in enumerate(players):
                     offensive_rebounds[player] = orebs[idx]
+            if "AST" in df.columns.values:
+                asts = df["AST"].tolist()
+                for idx, player in enumerate(players):
+                    assists[player] = asts[idx]
 
     teams: list[TeamModel] = []
     for a in scorebox_div.find_all("a"):
@@ -95,6 +109,7 @@ def _create_sportsreference_game_model(
                     fg,
                     fga,
                     offensive_rebounds,
+                    assists,
                 )
             )
 

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/sportsreference/sportsreference_player_model.py

@@ -3,6 +3,7 @@
 # pylint: disable=too-many-arguments,unused-argument
 import datetime
 import http
+import logging
 from urllib.parse import unquote
 
 import pytest_is_running
@@ -26,18 +27,21 @@ def _create_sportsreference_player_model(
     fg: dict[str, int],
     fga: dict[str, int],
     offensive_rebounds: dict[str, int],
+    assists: dict[str, int],
 ) -> PlayerModel | None:
     """Create a player model from NCAAB sports reference."""
     player_url = _fix_url(player_url)
     response = session.get(player_url, timeout=DEFAULT_TIMEOUT)
     # Some players can't be accessed on sports reference
     if response.status_code == http.HTTPStatus.FORBIDDEN:
+        logging.warning("Cannot access player at URL %s", player_url)
         return None
     response.raise_for_status()
     soup = BeautifulSoup(response.text, "html.parser")
     h1 = soup.find("h1")
     if h1 is None:
-        raise ValueError("h1 is null.")
+        logging.warning("h1 is null for %s", player_url)
+        return None
     name = h1.get_text().strip()
     return PlayerModel(
         identifier=name,
@@ -48,6 +52,7 @@ def _create_sportsreference_player_model(
         field_goals=fg.get(name),
         field_goals_attempted=fga.get(name),
         offensive_rebounds=offensive_rebounds.get(name),
+        assists=assists.get(name),
     )
 
 
@@ -58,9 +63,10 @@ def _cached_create_sportsreference_player_model(
     fg: dict[str, int],
     fga: dict[str, int],
     offensive_rebounds: dict[str, int],
+    assists: dict[str, int],
 ) -> PlayerModel | None:
     return _create_sportsreference_player_model(
-        session, player_url, fg, fga, offensive_rebounds
+        session, player_url, fg, fga, offensive_rebounds, assists
     )
 
 
@@ -71,13 +77,14 @@ def create_sportsreference_player_model(
     fg: dict[str, int],
     fga: dict[str, int],
     offensive_rebounds: dict[str, int],
+    assists: dict[str, int],
 ) -> PlayerModel | None:
     """Create a player model from sports reference."""
     if not pytest_is_running.is_running():
         return _cached_create_sportsreference_player_model(
-            session, player_url, fg, fga, offensive_rebounds
+            session, player_url, fg, fga, offensive_rebounds, assists
         )
     with session.cache_disabled():
         return _create_sportsreference_player_model(
-            session, player_url, fg, fga, offensive_rebounds
+            session, player_url, fg, fga, offensive_rebounds, assists
         )

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/sportsreference/sportsreference_team_model.py

@@ -29,6 +29,7 @@ def _create_sportsreference_team_model(
     fg: dict[str, int],
     fga: dict[str, int],
     offensive_rebounds: dict[str, int],
+    assists: dict[str, int],
 ) -> TeamModel:
     response = session.get(url)
     response.raise_for_status()
@@ -40,7 +41,7 @@ def _create_sportsreference_team_model(
     except (json.decoder.JSONDecodeError, IndexError) as exc:
         h1 = soup.find("h1")
         if not isinstance(h1, Tag):
-            raise ValueError("h1 is null.") from exc
+            raise ValueError(f"h1 is null for {url}.") from exc
         span = h1.find_all("span")
         name = span[1].get_text().strip()
 
@@ -57,7 +58,7 @@ def _create_sportsreference_team_model(
             y
             for y in [  # pyright: ignore
                 create_sportsreference_player_model(
-                    session, x, dt, fg, fga, offensive_rebounds
+                    session, x, dt, fg, fga, offensive_rebounds, assists
                 )
                 for x in valid_player_urls
             ]
@@ -83,9 +84,19 @@ def _cached_create_sportsreference_team_model(
     fg: dict[str, int],
     fga: dict[str, int],
     offensive_rebounds: dict[str, int],
+    assists: dict[str, int],
 ) -> TeamModel:
     return _create_sportsreference_team_model(
-        session, url, dt, league, player_urls, points, fg, fga, offensive_rebounds
+        session,
+        url,
+        dt,
+        league,
+        player_urls,
+        points,
+        fg,
+        fga,
+        offensive_rebounds,
+        assists,
     )
 
 
@@ -99,13 +110,32 @@ def create_sportsreference_team_model(
     fg: dict[str, int],
     fga: dict[str, int],
     offensive_rebounds: dict[str, int],
+    assists: dict[str, int],
 ) -> TeamModel:
     """Create a team model from Sports Reference."""
     if not pytest_is_running.is_running():
         return _cached_create_sportsreference_team_model(
-            session, url, dt, league, player_urls, points, fg, fga, offensive_rebounds
+            session,
+            url,
+            dt,
+            league,
+            player_urls,
+            points,
+            fg,
+            fga,
+            offensive_rebounds,
+            assists,
         )
     with session.cache_disabled():
         return _create_sportsreference_team_model(
-            session, url, dt, league, player_urls, points, fg, fga, offensive_rebounds
+            session,
+            url,
+            dt,
+            league,
+            player_urls,
+            points,
+            fg,
+            fga,
+            offensive_rebounds,
+            assists,
         )

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/team_model.py

@@ -17,6 +17,7 @@ NAME_COLUMN: Literal["name"] = "name"
 FIELD_GOALS_COLUMN: Literal["field_goals"] = "field_goals"
 FIELD_GOALS_ATTEMPTED_COLUMN: Literal["field_goals_attempted"] = "field_goals_attempted"
 OFFENSIVE_REBOUNDS_COLUMN: Literal["offensive_rebounds"] = "offensive_rebounds"
+ASSISTS_COLUMN: Literal["assists"] = "assists"
 
 
 def _calculate_kicks(data: dict[str, Any]) -> int | None:
@@ -75,6 +76,20 @@ def _calculate_offensive_rebounds(data: dict[str, Any]) -> int | None:
     return offensive_rebounds
 
 
+def _calculate_assists(data: dict[str, Any]) -> int | None:
+    assists = 0
+    found_assists = False
+    for player in data[PLAYER_COLUMN_PREFIX]:
+        player_assists = player.assists
+        if player_assists is None:
+            continue
+        found_assists = True
+        assists += player_assists
+    if not found_assists:
+        return None
+    return assists
+
+
 class TeamModel(BaseModel):
     """The serialisable team class."""
 
@@ -116,3 +131,8 @@ class TeamModel(BaseModel):
         json_schema_extra={TYPE_KEY: FieldType.LOOKAHEAD},
         alias=OFFENSIVE_REBOUNDS_COLUMN,
     )
+    assists: int | None = Field(
+        default_factory=_calculate_assists,
+        json_schema_extra={TYPE_KEY: FieldType.LOOKAHEAD},
+        alias=ASSISTS_COLUMN,
+    )

{sportsball-0.3.15 → sportsball-0.3.16}/sportsball/data/weather/openmeteo/openmeteo_weather_model.py

@@ -26,17 +26,23 @@ def _parse_openmeteo(
         return None
     if hourly is None:
         raise ValueError("hourly is null.")
+    dt_index = pd.date_range(
+        start=pd.to_datetime(hourly.Time(), unit="s"),
+        end=pd.to_datetime(hourly.TimeEnd(), unit="s"),
+        freq=pd.Timedelta(seconds=hourly.Interval()),
+        inclusive="left",
+        tz=tz,
+    )
+    row_count = min(
+        len(hourly.Variables(0).ValuesAsNumpy()),  # type: ignore
+        len(hourly.Variables(1).ValuesAsNumpy()),  # type: ignore
+        len(dt_index),
+    )
     hourly_df = pd.DataFrame(
-        index=pd.date_range(
-            start=pd.to_datetime(hourly.Time(), unit="s"),
-            end=pd.to_datetime(hourly.TimeEnd(), unit="s"),
-            freq=pd.Timedelta(seconds=hourly.Interval()),
-            inclusive="left",
-            tz=tz,
-        )[: len(hourly.Variables(0).ValuesAsNumpy())],  # type: ignore
+        index=dt_index[:row_count],  # type: ignore
         data={
-            "temperature_2m": hourly.Variables(0).ValuesAsNumpy(),  # type: ignore
-            "relative_humidity_2m": hourly.Variables(1).ValuesAsNumpy(),  # type: ignore
+            "temperature_2m": hourly.Variables(0).ValuesAsNumpy()[:row_count],  # type: ignore
+            "relative_humidity_2m": hourly.Variables(1).ValuesAsNumpy()[:row_count],  # type: ignore
         },
     )
     dt = dt.replace(tzinfo=None)