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')