team_stats.py

import json
import os
import requests
from bs4 import BeautifulSoup
import sqlite3
from datetime import date
import matplotlib.pyplot as plt

path = os.path.dirname(os.path.abspath(__file__))
conn = sqlite3.connect(path+'/'+'stats.db')
cur = conn.cursor()
#testing

def team_abrevs():
    abrevs = {}
    wiki = requests.get("https://en.wikipedia.org/wiki/Wikipedia:WikiProject_National_Basketball_Association/National_Basketball_Association_team_abbreviations")
    soup = BeautifulSoup(wiki.text, 'html.parser')
    table = soup.find('table')
    rows = table.find_all('tr')
    for x in range(1, len(rows)):
        row = rows[x]
        names = row.find_all('td')
        abr = names[0].text
        abr = abr.strip()
        team = names[1].text
        team = team.strip()
        abrevs[team] = abr
    abrevs['New Orleans Pelicans'] = 'NO'
    abrevs['Utah Jazz'] = 'UTAH'
    return abrevs

def get_team_stats(team):
    output = {}
    cur.execute(f"SELECT * FROM Teams where Team = '{team}'")
    ab = ""
    ident = 0
    for row in cur:
        ab = row[2]
        ident = row[0]
    output['Team'] = team
    output['Abbreviation'] = ab
    url = f"https://www.espn.com/nba/team/_/name/{ab}/{team}"
    r = requests.get(url)
    soup = BeautifulSoup(r.text, 'html.parser')
    column = soup.find('section', class_ = "col-c chk-height")
    sections = column.find_all('article')

    stats = sections[3]
    blocks = stats.find_all('div')
    for x in range(1,5):
        first = blocks[x].find_all('span')
        category = first[0].text
        value = first[1].text
        output[category] = value

    stats = column.find('article', class_ = "sub-module teamseasonhistory")
    blocks = stats.find_all('td')
    w = blocks[1].text
    l = blocks[2].text
    output['Wins'] = w
    output['Losses'] = l
    
    column = soup.find('section', class_ = "col-a chk-height")
    games = column.find_all('li')
    last_10 = []
    for x in range(1, 11):
        results = games[x].find_all('div')
        res = results[3].text
        last_10.append(res)
    w = 0
    for res in last_10:
        if res == 'W':
            w += 1
    win_per = w/10
    output['Last 10 Win Percentage'] = win_per
    return output

def stats(team):
    cur.execute(f'SELECT * FROM Teams WHERE Team = "{team}"')
    result = cur.fetchone()
    if result:
        ident = result[0]
    else:
        print("team not found")
        return
    now = date.today()
    now = str(now)
    cur.execute('CREATE TABLE IF NOT EXISTS Team_Stats (Team_id INTEGER, PPG REAL, RPG REAL, APG REAL, Points_Allowed REAL, Wins INTEGER, Losses INTEGER, Last_10_Win_Percentage REAL, Last_updated TEXT)')
    cur.execute(f"SELECT * FROM Team_Stats WHERE Team_id = '{ident}' AND Last_updated = '{now}'")
    result = cur.fetchone()
    if result and result[8] == now:
        val = {}
        cur.execute(f'SELECT * FROM Team_Stats JOIN Teams WHERE Teams.id = Team_Stats.Team_id AND Teams.Team = "{team}" AND Team_Stats.Last_updated = "{now}"')
        one = cur.fetchone()
        val['Team'] = one[10]
        val['Abbreviation'] = one[11]
        val['Points Per Game'] = one[1]
        val['Rebounds Per Game'] = one[2]
        val['Assists Per Game'] = one[3]
        val['Points Allowed'] = one[4]
        val['Wins'] = one[5]
        val['Losses'] = one[6]
        val['Last 10 Win Percentage'] = one[7]
        return val
    else:
        val = get_team_stats(team)
        cur.execute("INSERT INTO Team_Stats (Team_id, PPG, RPG, APG, Points_Allowed, Wins, Losses, Last_10_Win_Percentage, Last_updated) VALUES (?,?,?,?,?,?,?,?,?)",
        (ident, val['Points Per Game'], val['Rebounds Per Game'], val['Assists Per Game'], val['Points Allowed'], val['Wins'], val['Losses'], val['Last 10 Win Percentage'], now))
        conn.commit()
        return val

def make_id_table():
    abrevs = team_abrevs()
    cur.execute('CREATE TABLE IF NOT EXISTS Teams (id INTEGER PRIMARY KEY, Team TEXT, Abbreviation TEXT)')
    cur.execute("SELECT * FROM Teams")
    result = cur.fetchone()
    if result:
        return
    for key in abrevs:
        cur.execute('INSERT INTO Teams (Team, Abbreviation) VALUES (?, ?)', (key, abrevs[key]))
    conn.commit()

def compare_teams(team1, team2):
    stats1 = stats(team1)
    stats2 = stats(team2)
    ppg_offset = stats1["Points Per Game"] - stats2["Points Per Game"]
    rpg_offset = stats1["Rebounds Per Game"] - stats2["Rebounds Per Game"]
    apg_offset = stats1["Assists Per Game"] - stats2["Assists Per Game"]
    w_offset = stats1["Wins"] - stats2["Wins"]
    l10_offset = stats1["Last 10 Win Percentage"] * 10 - stats2["Last 10 Win Percentage"] * 10
    x_axis = ["Points", "Rebounds", "Assists", "Wins", "Wins in Last 10 Games"]
    y_axis = [ppg_offset, rpg_offset, apg_offset, w_offset, l10_offset]
    plt.xlabel("Stats")
    plt.ylabel("Difference in Stats (" + stats1["Abbreviation"] + " - " + stats2["Abbreviation"] + ")")
    plt.title(team1 + " vs. " + team2)
    plt.bar(x_axis, y_axis)
    plt.show()

#ab = team_abrevs()
#for keys in ab:
    #print(stats(keys))