import json
import os
from pbpstats import G_LEAGUE_STRING, NBA_STRING
from pbpstats.data_loader.stats_nba.web_loader import StatsNbaWebLoader
[docs]class StatsNbaPbpWebLoader(StatsNbaWebLoader):
"""
A ``StatsNbaPbpWebLoader`` object should be instantiated and passed into ``StatsNbaPbpLoader`` when loading data directly from the NBA Stats API
:param str file_directory: (optional, use it if you want to store the response data on disk)
Directory in which data should be either stored.
The specific file location will be `stats_<game_id>.json` in the `/pbp` subdirectory.
If not provided response data will not be saved on disk.
"""
def __init__(self, file_directory=None):
self.file_directory = file_directory
[docs] def load_data(self, game_id):
self.game_id = game_id
league_url_part = (
f"{G_LEAGUE_STRING}.{NBA_STRING}"
if self.league == G_LEAGUE_STRING
else self.league
)
self.base_url = f"https://stats.{league_url_part}.com/stats/playbyplayv2"
self.parameters = {
"GameId": self.game_id,
"StartPeriod": 0,
"EndPeriod": 10,
"RangeType": 2,
"StartRange": 0,
"EndRange": 55800,
}
return 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}/pbp/stats_{self.game_id}.json"
with open(file_path, "w") as outfile:
json.dump(self.source_data, outfile)