graphiti/examples/nba/runner.py
2024-08-26 20:04:35 -04:00

35 lines
1.5 KiB
Python

if __name__ == '__main__':
# {'id': 2544, 'full_name': 'LeBron James', 'first_name': 'LeBron', 'last_name': 'James', 'is_active': True}
all_teams = teams.get_teams()
all_players = players.get_players()
players_json = []
for t in all_teams:
name = t['full_name']
print(name)
if name == 'Golden State Warriors' or name == 'Boston Celtics' or name == 'Toronto Raptors':
roster = commonteamroster.CommonTeamRoster(team_id=t['id']).get_dict()
players_data = roster['resultSets'][0]
headers = players_data['headers']
row_set = players_data['rowSet']
players_json = []
for row in row_set:
player_dict = dict(zip(headers, row))
player_dict['team_name'] = name
print(player_dict)
meaningful_data = {
'team_name': name,
'player_id': player_dict['PLAYER_ID'],
'player_name': player_dict['PLAYER'],
'player_number': player_dict['NUM'],
'player_position': player_dict['POSITION'],
'player_school': player_dict['SCHOOL'],
}
players_json.append(meaningful_data)
print(len(players_json))
players_json.extend(players_json)
script_dir = Path(__file__).parent
filename = script_dir / 'current_nba_roster.json'
print(players_json)
with open(filename, 'w') as f:
json.dump(players_json, f, indent=2)