update_teams_players.py

# update_teams_players.py
"""
This should be run every season, or maybe more often if the players are changing

I wish there was a good way to get the team abbreviations at the same time
Probably I could dig up an API somewhere that has them
"""
from modules.api_wrappers import TheOdds, APIFootball
import json
from dotenv import load_dotenv
import os
import sys

load_dotenv()
API_KEY_APIFOOTBALL = os.getenv('API_KEY_APIFOOTBALL')
API_KEY_THEODDS = os.getenv('API_KEY_THEODDS')
SEASON = int(os.getenv('SEASON'))    # storing in .env so it's easy to modify without making the local repo out of sync

todds = TheOdds(api_key=API_KEY_THEODDS)
apif = APIFootball(api_key=API_KEY_APIFOOTBALL, season=SEASON)

# Get the league ID for the current prem season
response = apif.get_league_search('premier_league').json()
league_id = None
for league in response['api']['leagues']:
    if league['country_code'] == 'GB' and league['season'] == SEASON:
        league_id = league['league_id']
if not league_id:
    print("Can't find the league ID")
    sys.exit()

# Get prem team names and ids from APIFootball
response = apif.get_teams(league_id).json()
apif_teams = [team['team_id'] for team in response['api']['teams']]
apif_names = sorted([team['name'] for team in response['api']['teams']])

# Iterate through team IDs to get a list of all prem player IDs
apif_player_ids = []
for team_id in apif_teams:
    response = apif.get_squad(team_id).json()
    squad_player_ids = [player['player_id'] for player in response['api']['players']]
    apif_player_ids += squad_player_ids
"""
I figured out how to do the above in 1 line with a list comprehension but it is really tough to read:
apif_player_ids = [player["player_id"] for team_id in apif_teams for player in apif.get_squad(team_id).json()['api']['players']]
"""

# Get prem team names from TheOdds
response = todds.get_odds_theodds().json()
todds_teams = [fixture['teams'][0] for fixture in response['data']]\
              + [fixture['teams'][1] for fixture in response['data']]
todds_teams = list(set(todds_teams))    # remove duplicates
todds_teams = (sorted(todds_teams))

# Store all of this as a json file
json_storage = {
    'APIFootball_league_ID': league_id,
    'APIFootball_team_names': apif_names,
    'APIFootball_player_IDs': apif_player_ids,
    'TheOdds_team_names': todds_teams
}
with open('data/league_data.txt', 'w') as outfile:
    json.dump(json_storage, outfile)