NBA.py

import json
import sqlite3
import os
import requests
import time 

#Create function to get dictionary of NBA Teams and their respecitve ID's from the API 
def get_nba_teams_id():
    team={}
    for l in range(1,1315):
        if len(team)==30:
            break
        url_2="https://www.balldontlie.io/api/v1/stats"
        try:
            ne=requests.get(url_2,
            params={
            'start_date':'2018-10-15',
            'end_date':'2019-06-11',
            'page':l}).json()
            x=ne.get('data')
            for y in x:
                team_id=y['game']['home_team_id']
                players_team_id=y['team']['id']
                team_name=y['team']['name']
                if team_id==players_team_id:
                    if team_id not in team.keys():
                        team[team_id]=team_name
        except:
            None
    return team

#Create function to pull every NBA game from the 2018-2019 season 
#with the date, home team, home team score, away team, away team score, and winner 
def get_nba_api_data():
    nba_data=[]
    team_and_id=get_nba_teams_id()
    game_id=0
    for l in range(1,1315):
        url_2="https://www.balldontlie.io/api/v1/stats"
        try:
            ne=requests.get(url_2,
            params={
            'start_date':'2018-10-15',
            'end_date':'2019-06-11',
            'page':l}).json()
            x=ne.get('data')  
            first=x[0]

            date=first['game']['date']
            date=date.split('T')
            date=date[0]
            #date is a str
            
            
            home_team_id=first['game']['home_team_id']
            home_team=team_and_id[home_team_id]
            home_team_score=first['game']['home_team_score']


            away_team_id=first['game']['visitor_team_id']
            away_team=team_and_id[away_team_id]
            away_team_score=first['game']['visitor_team_score']
            game_id+=1

            winner=''
            if home_team_score>away_team_score:
                winner=home_team
            else:
                winner=away_team

            
            data=(game_id,date,home_team,home_team_score,away_team,away_team_score,winner)    
            nba_data.append(data)
        except:
            None
    return nba_data
        
           


#create database and limit to 25 
def table_setup():
    path = os.path.dirname(os.path.abspath(__file__))
    db = os.path.join(path, 'final.db')
    conn= sqlite3.connect(db)
    cur = conn.cursor() 
    cur.execute("CREATE TABLE IF NOT EXISTS NBA_Season (Game_ID INTEGER, Date TEXT, Home TEXT, Home_Score INTEGER, Away TEXT, Away_Score INTEGER, Winner TEXT)")
    data_1=get_nba_api_data()
    count=0
    cur.execute("SELECT Game_ID FROM NBA_Season")
    game_id=cur.fetchall()
    for x in data_1:
        if count<=24:
            if x[0] not in game_id:
                cur.execute("INSERT INTO NBA_Season (Game_ID, Date, Home, Home_Score, Away, Away_Score, Winner) VALUES (?, ?, ?, ?, ?, ?, ?)",(x[0],x[1], x[2], x[3], x[4], x[5], x[6]))
                count+=1
            else:
                continue
    conn.commit()
table_setup()