Source code for pbpstats.data_loader.stats_nba.file_loader

import json
from pathlib import Path

from pbpstats.data_loader.stats_nba.base import StatsNbaLoaderBase


class StatsNbaFileLoader(StatsNbaLoaderBase):
    """
    Base class for loading stats.nba.com files saved on disk.

    All stats.nba.com data loader classes should inherit from this class.

    This class should not be instantiated directly.
    """

    def _load_data_from_file(self):
        data_file = Path(self.file_path)
        if not data_file.is_file():
            raise Exception(f"{self.file_path} does not exist")
        with open(self.file_path) as json_data:
            self.source_data = json.load(json_data)
            return self.source_data