def get_team_stats(self, team_id=None, team_name=None, season=None): """Get team stats. Return team stats object""" if not team_id: team_id = _team_id(team_name) endpoint = f"teams/{team_id}?expand=team.stats" data = _api_request(endpoint) return data['teams'][0]
def __init__(self, team=None): super().__init__() self.team = team self.id = _team_id(self.team) self.info = self.get_team_info(team_id=self.id) self.name = self.info['name'] self.venue = self.info['venue']['name'] self.conference = self.info['conference']['name'] self.division = self.info['division']['name'] self.stats = self.get_team_stats(self.id) self.roster = self.get_team_roster(self.id) self.schedule = _parse_schedule( self.get_team_schedule(team_id=self.id)) self.remaining_games = self.schedule.unplayed self.record = self.team_records.get(self.name) self.wins = self.record['record']['wins'] self.losses = self.record['record']['losses'] self.otl = self.record['record']['ot'] self.games_played = self.record['games_played'] self.points = self.record['points'] self.division_rank = self.division_standings[self.division][self.name] self.conference_rank = self.conference_standings[self.conference][ self.name] self.overall_rank = self.league_standings.get(self.name) self.year_by_year_records = None
def get_team_schedule(self, team_name=None, team_id=None, season=None): """Get team schedule. Return list of game objects""" team_id = _team_id(team_name) if not team_id else team_id season = self.current_season if not season else season endpoint = f"schedule?teamId={team_id}&season={season}" data = _api_request(endpoint) game_list = data['dates'] yield from game_list
def get_team_roster(self, team_id=None, team_name=None, season=None): """Get team roster. Return list of player objects""" team_id = _team_id(team_name) if not team_id else team_id season = self.current_season if not season else season endpoint = f"teams/{team_id}/roster?season={season}" data = _api_request(endpoint) player_list = data['roster'] return player_list
def get_team_info(self, team_id=None, team_name=None): """Get general team information""" if not team_id: team_id = _team_id(team_name) endpoint = f"teams/{team_id}" data = _api_request(endpoint) if data: team_info = data['teams'][0] return team_info
def test_team_id(self): team_id = _helpers._team_id(self.team_city) self.assertEqual(team_id, 6, 'Incorrect Team ID')