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 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_player_info(self, player_id=None, player_name=None):
"""Get individual stats for a player"""
if not player_id:
player_id = _player_id(player_name)
endpoint = f"people/{player_id}"
data = _api_request(endpoint)
if data:
info = data['people'][0]
return info
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 get_career_stats(self, player_id=None, player_name=None):
"""Get career stats for a player"""
if not player_id:
player_id = _player_id(player_name)
stats_endpoint = "stats?stats=yearByYear"
endpoint = f"people/{player_id}/{stats_endpoint}"
data = _api_request(endpoint)
if data:
seasons = data['stats'][0]['splits']
return seasons
def get_player_stats(self, player_id=None, player_name=None, season=None):
"""Get individual stats for a player"""
if not player_id:
player_id = _player_id(player_name)
player_info = self.get_player_info(player_id)
player_active = player_info.get('active')
team = player_info.get('currentTeam')
if team:
team = team['id']
if not season and not player_active:
raise JockBotNHLException('Season required for inactive players')
if not season:
season = _current_season()
season_endpoint = f"stats?stats=statsSingleSeason&season={season}"
endpoint = f"people/{player_id}/{season_endpoint}"
data = _api_request(endpoint)
if data:
stats = data['stats'][0]['splits'][0]
stats['team'] = team
return stats