def test_get_functions(self):
"""Tests various get functions to assure they don't return nothing/empty
stringss"""
self.assertIsNotNone(get_color())
self.assertIsNotNone(get_full_name())
self.assertIsNotNone(get_team())
self.assertIsNotNone(get_season_year())
def _get_conf_division(self, team=None):
"""Method to get division of given team.
Args:
team: team url, ie raptors, sixers, for team who division you want
to acquire. If none, uses favourite team from config.
"""
if not team: team = get_team()
return self.TI.get_conf_division(team)
def get_conf_division(self, team=None, id_option=None):
if team == None:
team = get_team()
for teams in self.standard:
if teams['urlName'] == team.lower():
return (teams['confName'], teams['divName'])
elif team != None and id_option == None:
team = self.get_team(team)
for teams in self.standard:
if teams['urlName'] == team['urlName'].lower():
return (teams['confName'], teams['divName'])
elif team != None and id_option != None:
team = self.get_team(team, id_option=id_option, id_return=None)
print('Could not find division')
def __init__(self, season=None, team=None):
self.season = season if season != None \
else get_season_year()
self.team = team if team != None else get_team()
self.raw_data = get_schedule_data(season=self.season, team=self.team)
self._internal = self.raw_data['_internal']
self.league = self.raw_data['league']
self.standard = self.league['standard'] #list of all games
self.last_game_idx = self.league['lastStandardGamePlayedIndex']
self.last_game = self.standard[self.last_game_idx]
self.last_game_id = self.last_game['gameId']
self.last_game_date = self.last_game['gameUrlCode'][0:8]
self.last_game_id_date = (self.last_game_id, self.last_game_date)
def get_schedule_data(season=None, team=None):
"""Gets raw json data of given team schedule for given season.
Args:
season: year of season start date, YYYY format.
team: teamUrl for given team -> team name, ex: raptors, sixers
Returns:
Dict of raw json data from data.nba.net.../schedule.json endpoint
"""
if not team: season = get_season_year()
if not team: team = get_team()
team = handle_team_url_name(team)
url_start = 'http://data.nba.net/prod/v1/'
url = url_start + str(season) + '/teams/' + str(team) + '/schedule.json'
data = get_data(url)
return data
def get_team_leaders_data(team=None, season=None):
"""Gets raw json data for stat leaders for given team.
Args:
team: lowercase team name of team. ex: raptors, sixers, bulls
season: Year of season start date. Ex: 2019 for the 2019/2020 season.
Returns:
Dict of raw json data from data.nba.net /leaders.json endpoint
"""
if not team: team = get_team()
team = handle_team_url_name(team)
if not season: season = get_season_year()
url1 = 'http://data.nba.net/prod/v1/'
url_end = '/leaders.json'
url = str(url1) + str(season) + '/teams/' + str(
team.lower()) + str(url_end)
data = get_data(url)
return data