Source code for pbpstats.data_loader.data_nba.boxscore_loader

"""
``DataNbaBoxscoreLoader`` loads boxscore data for a game and creates :obj:`~pbpstats.resources.boxscore.data_nba_boxscore_item.DataNbaBoxscoreItem` objects for each player and team

The following code will load boxscore data for game id "0021900001" from a file located in a subdirectory of the /data directory

.. code-block:: python

    from pbpstats.data_loader import DataNbaBoxscoreLoader

    boxscore_loader = DataNbaBoxscoreLoader("0021900001", "file", "/data")
    print(boxscore_loader.items[0].data) # prints dict with a player's boxscore data for game
"""

import json
import os

from pbpstats import NBA_STRING, G_LEAGUE_STRING
from pbpstats.data_loader.abs_data_loader import check_file_directory
from pbpstats.data_loader.data_nba.file_loader import DataNbaFileLoader
from pbpstats.data_loader.data_nba.web_loader import DataNbaWebLoader
from pbpstats.resources.boxscore.data_nba_boxscore_item import DataNbaBoxscoreItem


[docs]class DataNbaBoxscoreLoader(DataNbaFileLoader, DataNbaWebLoader): """ Loads data.nba.com source boxscore data for game. Team/Player data is stored in items attribute as :obj:`~pbpstats.resources.boxscore.data_nba_boxscore_item.DataNbaBoxscoreItem` objects :param str game_id: NBA Stats Game Id :param str source: Where should data be loaded from. Options are 'web' or 'file' :param str file_directory: (optional if source is 'web') Directory in which data should be either stored (if source is web) or loaded from (if source is file). The specific file location will be `data_<game_id>.json` in the `/game_details` subdirectory. If not provided response data will not be saved on disk. """ data_provider = "data_nba" resource = "Boxscore" parent_object = "Game" def __init__(self, game_id, source, file_directory=None): self.game_id = game_id self.file_directory = file_directory self.source = source self._load_data() self._make_boxscore_items() def _load_data(self): source_method = getattr(self, f"_from_{self.source}") source_method() @check_file_directory def _from_file(self): self.file_path = f"{self.file_directory}/game_details/data_{self.game_id}.json" self._load_data_from_file() def _from_web(self): league_url_part = NBA_STRING if self.league == G_LEAGUE_STRING else self.league self.url = f"http://data.{league_url_part}.com/data/v2015/json/mobile_teams/{self.league}/{self.season}/scores/gamedetail/{self.game_id}_gamedetail.json" self._load_request_data() def _save_data_to_file(self): if self.file_directory is not None and os.path.isdir(self.file_directory): file_path = f"{self.file_directory}/game_details/data_{self.game_id}.json" with open(file_path, "w") as outfile: json.dump(self.source_data, outfile) def _make_boxscore_items(self): """ makes :obj:`~pbpstats.resources.boxscore.DataNbaBoxscoreItem` items for each player/team """ home = self.data["hls"] away = self.data["vls"] self.items = [ DataNbaBoxscoreItem(item, team_id=away["tid"], team_abbreviation=away["ta"]) for item in away["pstsg"] ] self.items += [ DataNbaBoxscoreItem(item, team_id=home["tid"], team_abbreviation=home["ta"]) for item in home["pstsg"] ] self.items.append( DataNbaBoxscoreItem( away["tstsg"], team_id=away["tid"], team_abbreviation=away["ta"] ) ) self.items.append( DataNbaBoxscoreItem( home["tstsg"], team_id=home["tid"], team_abbreviation=home["ta"] ) ) @property def data(self): """ returns raw JSON response data """ return self.source_data["g"]