def setUp(self):
self.atlanta_box_score_2017_01_01 = requests.get(
'https://www.basketball-reference.com/boxscores/201701010ATL.html'
).text
self.team_abbreviation_parser = TeamAbbreviationParser(
abbreviations_to_teams=TEAM_ABBREVIATIONS_TO_TEAM, )
self.page = BoxScoresPage(
html.fromstring(self.atlanta_box_score_2017_01_01))
self.combined_team_totals = [
TeamTotal(team_abbreviation=table.team_abbreviation,
totals=table.team_totals)
for table in self.page.basic_statistics_tables
]
self.parser = TeamTotalsParser(
team_abbreviation_parser=self.team_abbreviation_parser)
self.team_totals = self.parser.parse(
first_team_totals=self.combined_team_totals[0],
second_team_totals=self.combined_team_totals[1],
)
def team_box_score(game_url_path):
url = "{BASE_URL}/{game_url_path}".format(BASE_URL=BASE_URL, game_url_path=game_url_path)
response = requests.get(url=url)
response.raise_for_status()
page = BoxScoresPage(html.fromstring(response.content))
combined_team_totals = [
TeamTotal(team_abbreviation=table.team_abbreviation, totals=table.team_totals)
for table in page.basic_statistics_tables
]
parser = TeamTotalsParser(team_abbreviation_parser=TeamAbbreviationParser(
abbreviations_to_teams=TEAM_ABBREVIATIONS_TO_TEAM,
))
return parser.parse(
first_team_totals=combined_team_totals[0],
second_team_totals=combined_team_totals[1],
)
class TestTeamTotalsParser(TestCase):
def setUp(self):
self.parser = TeamTotalsParser(
team_abbreviation_parser=TeamAbbreviationParser(
abbreviations_to_teams=TEAM_ABBREVIATIONS_TO_TEAM))
def test_parse_none_outcome_when_points_are_same(self):
team_totals = TeamTotal(team_abbreviation="BOS",
totals=MagicMock(points="100"))
opposing_team_totals = TeamTotal(team_abbreviation="GSW",
totals=MagicMock(points="100"))
self.assertIsNone(
self.parser.parse_totals(
team_totals=team_totals,
opposing_team_totals=opposing_team_totals)["outcome"])
class TestParseTeams(TestCase):
def setUp(self):
self.atlanta_box_score_2017_01_01 = requests.get(
'https://www.basketball-reference.com/boxscores/201701010ATL.html'
).text
self.team_abbreviation_parser = TeamAbbreviationParser(
abbreviations_to_teams=TEAM_ABBREVIATIONS_TO_TEAM, )
self.page = BoxScoresPage(
html.fromstring(self.atlanta_box_score_2017_01_01))
self.combined_team_totals = [
TeamTotal(team_abbreviation=table.team_abbreviation,
totals=table.team_totals)
for table in self.page.basic_statistics_tables
]
self.parser = TeamTotalsParser(
team_abbreviation_parser=self.team_abbreviation_parser)
self.team_totals = self.parser.parse(
first_team_totals=self.combined_team_totals[0],
second_team_totals=self.combined_team_totals[1],
)
def test_parse_two_team_totals(self):
self.assertEqual(len(self.team_totals), 2)
def test_parse_san_antonio_team_totals(self):
sas_team_totals = self.team_totals[0]
self.assertEqual(sas_team_totals["team"], Team.SAN_ANTONIO_SPURS)
self.assertEqual(sas_team_totals["outcome"], Outcome.LOSS)
self.assertEqual(sas_team_totals["minutes_played"], 265)
self.assertEqual(sas_team_totals["made_field_goals"], 42)
self.assertEqual(sas_team_totals["attempted_field_goals"], 90)
self.assertEqual(sas_team_totals["made_three_point_field_goals"], 9)
self.assertEqual(sas_team_totals["attempted_three_point_field_goals"],
27)
self.assertEqual(sas_team_totals["made_free_throws"], 19)
self.assertEqual(sas_team_totals["attempted_free_throws"], 22)
self.assertEqual(sas_team_totals["offensive_rebounds"], 9)
self.assertEqual(sas_team_totals["defensive_rebounds"], 38)
self.assertEqual(sas_team_totals["assists"], 27)
self.assertEqual(sas_team_totals["steals"], 5)
self.assertEqual(sas_team_totals["blocks"], 6)
self.assertEqual(sas_team_totals["turnovers"], 12)
self.assertEqual(sas_team_totals["personal_fouls"], 21)
self.assertEqual(sas_team_totals["points"], 112)
def test_parse_atlanta_team_totals(self):
atl_team_totals = self.team_totals[1]
self.assertEqual(atl_team_totals["team"], Team.ATLANTA_HAWKS)
self.assertEqual(atl_team_totals["outcome"], Outcome.WIN)
self.assertEqual(atl_team_totals["minutes_played"], 265)
self.assertEqual(atl_team_totals["made_field_goals"], 42)
self.assertEqual(atl_team_totals["attempted_field_goals"], 92)
self.assertEqual(atl_team_totals["made_three_point_field_goals"], 14)
self.assertEqual(atl_team_totals["attempted_three_point_field_goals"],
28)
self.assertEqual(atl_team_totals["made_free_throws"], 16)
self.assertEqual(atl_team_totals["attempted_free_throws"], 27)
self.assertEqual(atl_team_totals["offensive_rebounds"], 11)
self.assertEqual(atl_team_totals["defensive_rebounds"], 35)
self.assertEqual(atl_team_totals["assists"], 25)
self.assertEqual(atl_team_totals["steals"], 6)
self.assertEqual(atl_team_totals["blocks"], 6)
self.assertEqual(atl_team_totals["turnovers"], 11)
self.assertEqual(atl_team_totals["personal_fouls"], 21)
self.assertEqual(atl_team_totals["points"], 114)
def __init__(self):
self.team_abbreviation_parser = TeamAbbreviationParser(
abbreviations_to_teams=TEAM_ABBREVIATIONS_TO_TEAM)
self.league_abbreviation_parser = LeagueAbbreviationParser(
abbreviations_to_league=LEAGUE_ABBREVIATIONS_TO_LEAGUE)
self.location_abbreviation_parser = LocationAbbreviationParser(
abbreviations_to_locations=LOCATION_ABBREVIATIONS_TO_POSITION, )
self.outcome_abbreviation_parser = OutcomeAbbreviationParser(
abbreviations_to_outcomes=OUTCOME_ABBREVIATIONS_TO_OUTCOME, )
self.outcome_parser = PlayerBoxScoreOutcomeParser(
outcome_abbreviation_parser=self.outcome_abbreviation_parser)
self.period_details_parser = PeriodDetailsParser(
regulation_periods_count=4)
self.period_timestamp_parser = PeriodTimestampParser(
timestamp_format=ParserService.PLAY_BY_PLAY_TIMESTAMP_FORMAT)
self.position_abbreviation_parser = PositionAbbreviationParser(
abbreviations_to_positions=POSITION_ABBREVIATIONS_TO_POSITION, )
self.seconds_played_parser = SecondsPlayedParser()
self.scores_parser = ScoresParser(
scores_regex=ParserService.PLAY_BY_PLAY_SCORES_REGEX)
self.search_result_name_parser = SearchResultNameParser()
self.search_result_location_parser = ResourceLocationParser(
resource_location_regex=ParserService.
SEARCH_RESULT_RESOURCE_LOCATION_REGEX)
self.team_name_parser = TeamNameParser(
team_names_to_teams=TEAM_NAME_TO_TEAM)
self.play_by_plays_parser = PlayByPlaysParser(
period_details_parser=self.period_details_parser,
period_timestamp_parser=self.period_timestamp_parser,
scores_parser=self.scores_parser,
)
self.player_box_scores_parser = PlayerBoxScoresParser(
team_abbreviation_parser=self.team_abbreviation_parser,
location_abbreviation_parser=self.location_abbreviation_parser,
outcome_abbreviation_parser=self.outcome_abbreviation_parser,
seconds_played_parser=self.seconds_played_parser)
self.player_data_parser = PlayerDataParser(
search_result_location_parser=self.search_result_location_parser,
league_abbreviation_parser=self.league_abbreviation_parser,
)
self.player_season_box_scores_parser = PlayerSeasonBoxScoresParser(
team_abbreviation_parser=self.team_abbreviation_parser,
location_abbreviation_parser=self.location_abbreviation_parser,
outcome_parser=self.outcome_parser,
seconds_played_parser=self.seconds_played_parser)
self.player_season_totals_parser = PlayerSeasonTotalsParser(
position_abbreviation_parser=self.position_abbreviation_parser,
team_abbreviation_parser=self.team_abbreviation_parser,
)
self.player_advanced_season_totals_parser = PlayerAdvancedSeasonTotalsParser(
team_abbreviation_parser=self.team_abbreviation_parser,
position_abbreviation_parser=self.position_abbreviation_parser,
)
self.scheduled_start_time_parser = ScheduledStartTimeParser()
self.scheduled_games_parser = ScheduledGamesParser(
start_time_parser=self.scheduled_start_time_parser,
team_name_parser=self.team_name_parser,
)
self.search_results_parser = SearchResultsParser(
search_result_name_parser=self.search_result_name_parser,
search_result_location_parser=self.search_result_location_parser,
league_abbreviation_parser=self.league_abbreviation_parser,
)
self.team_totals_parser = TeamTotalsParser(
team_abbreviation_parser=self.team_abbreviation_parser)
self.division_name_parser = DivisionNameParser(divisions=Division)
self.team_standings_parser = TeamStandingsParser(teams=Team)
self.conference_division_standings_parser = ConferenceDivisionStandingsParser(
division_name_parser=self.division_name_parser,
team_standings_parser=self.team_standings_parser,
divisions_to_conferences=DIVISIONS_TO_CONFERENCES,
)
class ParserService:
PLAY_BY_PLAY_TIMESTAMP_FORMAT = "%M:%S.%f"
PLAY_BY_PLAY_SCORES_REGEX = "(?P<away_team_score>[0-9]+)-(?P<home_team_score>[0-9]+)"
SEARCH_RESULT_RESOURCE_LOCATION_REGEX = '(https?:\/\/www\.basketball-reference\.com\/)?(?P<resource_type>.+?(?=\/)).*\/(?P<resource_identifier>.+).html'
def __init__(self):
self.team_abbreviation_parser = TeamAbbreviationParser(
abbreviations_to_teams=TEAM_ABBREVIATIONS_TO_TEAM)
self.league_abbreviation_parser = LeagueAbbreviationParser(
abbreviations_to_league=LEAGUE_ABBREVIATIONS_TO_LEAGUE)
self.location_abbreviation_parser = LocationAbbreviationParser(
abbreviations_to_locations=LOCATION_ABBREVIATIONS_TO_POSITION, )
self.outcome_abbreviation_parser = OutcomeAbbreviationParser(
abbreviations_to_outcomes=OUTCOME_ABBREVIATIONS_TO_OUTCOME, )
self.outcome_parser = PlayerBoxScoreOutcomeParser(
outcome_abbreviation_parser=self.outcome_abbreviation_parser)
self.period_details_parser = PeriodDetailsParser(
regulation_periods_count=4)
self.period_timestamp_parser = PeriodTimestampParser(
timestamp_format=ParserService.PLAY_BY_PLAY_TIMESTAMP_FORMAT)
self.position_abbreviation_parser = PositionAbbreviationParser(
abbreviations_to_positions=POSITION_ABBREVIATIONS_TO_POSITION, )
self.seconds_played_parser = SecondsPlayedParser()
self.scores_parser = ScoresParser(
scores_regex=ParserService.PLAY_BY_PLAY_SCORES_REGEX)
self.search_result_name_parser = SearchResultNameParser()
self.search_result_location_parser = ResourceLocationParser(
resource_location_regex=ParserService.
SEARCH_RESULT_RESOURCE_LOCATION_REGEX)
self.team_name_parser = TeamNameParser(
team_names_to_teams=TEAM_NAME_TO_TEAM)
self.play_by_plays_parser = PlayByPlaysParser(
period_details_parser=self.period_details_parser,
period_timestamp_parser=self.period_timestamp_parser,
scores_parser=self.scores_parser,
)
self.player_box_scores_parser = PlayerBoxScoresParser(
team_abbreviation_parser=self.team_abbreviation_parser,
location_abbreviation_parser=self.location_abbreviation_parser,
outcome_abbreviation_parser=self.outcome_abbreviation_parser,
seconds_played_parser=self.seconds_played_parser)
self.player_data_parser = PlayerDataParser(
search_result_location_parser=self.search_result_location_parser,
league_abbreviation_parser=self.league_abbreviation_parser,
)
self.player_season_box_scores_parser = PlayerSeasonBoxScoresParser(
team_abbreviation_parser=self.team_abbreviation_parser,
location_abbreviation_parser=self.location_abbreviation_parser,
outcome_parser=self.outcome_parser,
seconds_played_parser=self.seconds_played_parser)
self.player_season_totals_parser = PlayerSeasonTotalsParser(
position_abbreviation_parser=self.position_abbreviation_parser,
team_abbreviation_parser=self.team_abbreviation_parser,
)
self.player_advanced_season_totals_parser = PlayerAdvancedSeasonTotalsParser(
team_abbreviation_parser=self.team_abbreviation_parser,
position_abbreviation_parser=self.position_abbreviation_parser,
)
self.scheduled_start_time_parser = ScheduledStartTimeParser()
self.scheduled_games_parser = ScheduledGamesParser(
start_time_parser=self.scheduled_start_time_parser,
team_name_parser=self.team_name_parser,
)
self.search_results_parser = SearchResultsParser(
search_result_name_parser=self.search_result_name_parser,
search_result_location_parser=self.search_result_location_parser,
league_abbreviation_parser=self.league_abbreviation_parser,
)
self.team_totals_parser = TeamTotalsParser(
team_abbreviation_parser=self.team_abbreviation_parser)
self.division_name_parser = DivisionNameParser(divisions=Division)
self.team_standings_parser = TeamStandingsParser(teams=Team)
self.conference_division_standings_parser = ConferenceDivisionStandingsParser(
division_name_parser=self.division_name_parser,
team_standings_parser=self.team_standings_parser,
divisions_to_conferences=DIVISIONS_TO_CONFERENCES,
)
def parse_division_standings(self, standings):
return self.conference_division_standings_parser.parse(
division_standings=standings)
def parse_play_by_plays(self, play_by_plays, away_team_name,
home_team_name):
return self.play_by_plays_parser.parse(
play_by_plays=play_by_plays,
away_team=self.team_name_parser.parse_team_name(
team_name=away_team_name),
home_team=self.team_name_parser.parse_team_name(
team_name=home_team_name),
)
def parse_player_box_scores(self, box_scores):
return self.player_box_scores_parser.parse(box_scores=box_scores)
def parse_player_season_box_scores(self, box_scores):
return self.player_season_box_scores_parser.parse(
box_scores=box_scores)
def parse_player_advanced_season_totals_parser(self, totals):
return self.player_advanced_season_totals_parser.parse(totals=totals)
def parse_player_season_totals(self, totals):
return self.player_season_totals_parser.parse(totals=totals)
def parse_scheduled_games(self, games):
return self.scheduled_games_parser.parse_games(games)
def parse_team_totals(self, first_team_totals, second_team_totals):
return self.team_totals_parser.parse(
first_team_totals=first_team_totals,
second_team_totals=second_team_totals)
def parse_player_search_results(self, nba_aba_baa_players):
return self.search_results_parser.parse(
nba_aba_baa_players=nba_aba_baa_players)
def parse_player_data(self, player):
return self.player_data_parser.parse(player=player)
def setUp(self):
self.parser = TeamTotalsParser(
team_abbreviation_parser=TeamAbbreviationParser(
abbreviations_to_teams=TEAM_ABBREVIATIONS_TO_TEAM))