main.py

from flask import Flask, render_template
import requests
import os
import json
import arrow

cwd = os.path.abspath(os.getcwd()).replace("\\","/")
appCounter = 0
app = Flask(__name__)

def download_image(url, destination):
    data = requests.get(url)
    with open(destination, "wb") as f:
        f.write(data.content)

def hexToRGB(value):
    value = value.lstrip('#')
    lv = len(value)
    return tuple(int(value[i:i + lv // 3], 16) for i in range(0, lv, lv // 3))

def getThumbnails(day, month, year):
    driverData = requests.get('https://api.openf1.org/v1/drivers?session_key=latest').json()
    for i in driverData:
        code = i["name_acronym"]
        if not os.path.exists(f"{cwd}/static/driverThumbnails/{code}.png"):
            file = open(f"{cwd}/static/driverThumbnails/{code}.png","x")
            file.close()
            download_image(i["headshot_url"],f"{cwd}/static/driverThumbnails/{code}.png")
    with open(f"{cwd}/appData.json","r") as f:
        appData = json.load(f)
    appData["yearUpdated"] = year
    appData["monthUpdated"] = month
    appData["dayUpdated"] = day
    with open("appData.json","w") as f:
        json.dump(appData,f)

def setup():
    with open("appData.json","r") as f:
        appData = json.load(f)
    utc = arrow.utcnow()
    year = int(utc.format("YYYY"))
    month = int(utc.format("MM"))
    day = int(utc.format("DD"))
    if appData["yearUpdated"] != year or appData["monthUpdated"] != month or appData["dayUpdated"] != day:
        thumbnails = os.listdir(f"{cwd}/static/driverThumbnails")
        for file in thumbnails:
            filePath = f"{cwd}/static/driverThumbnails/{file}"
            if os.path.isfile(filePath):
                os.remove(filePath)
        getThumbnails(day, month, year)

def getChampionshipsStandings(): #requests all data and then returns needed data.
    #constructorStanding and driverStanding are the leaderboards,
    #driverData provides extra needed data like team colours.
    constructorStanding = requests.get('https://api.jolpi.ca/ergast/f1/2025/constructorstandings/').json()["MRData"]["StandingsTable"]["StandingsLists"][0]["ConstructorStandings"]
    driverStanding = requests.get('https://api.jolpi.ca/ergast/f1/2025/driverstandings/').json()["MRData"]["StandingsTable"]["StandingsLists"][0]["DriverStandings"]
    driverData = requests.get('https://api.openf1.org/v1/drivers?session_key=latest').json()

    constructorTable = []
    constructorTable.append(["Position", "Name", "Points"]) #adds header row to constructorTable
    for constructor in constructorStanding:
        name = constructor["Constructor"]["name"]
        if name.split(" ")[0] == "RB": #Renaming RB to Racing Bulls to match driverData
            name = "Racing Bulls"
        position = constructor["position"]
        points = constructor["points"]
        colour = "#FFFFFF" #Default if real colour couldn't be found.
        count = 0
        done = False
        while not done and count < len(driverData): #Use while so that can exit once colour has been found
            driver = driverData[count] # will select every driver from driverData until finds one from the current team.
            if (name != "Racing Bulls" and name.split()[0] in driver["team_name"]) or (name == "Racing Bulls" and driver["team_name"] == "Racing Bulls"): #Checks first word of current team's name is in the driverData driver's team name. For some reason Alpine didn't work otherwise. Exception is Racing Bulls as Racing is also in Red Bull Racing
                teamColour = "#" + driver["team_colour"]
                done = True #don't have to check more drivers as have found the team colour
            count += 1
        constructorTable.append([position, name, points, hexToRGB(teamColour)]) #adds all of the data as a row in the constructorTable list. teamColour won't be displayed, just used for row colour
    
    tempDriverTable = []
    tempDriverTable.append(["", "Position", "Number", "Points", "Wins",]) #adds header row to tempDriverTable
    for driver in driverStanding:
        name = f"{driver["Driver"]["givenName"]} {driver["Driver"]["familyName"]}"
        position = driver["position"]
        points = driver["points"]
        wins = driver["wins"]
        number = driver["Driver"]["permanentNumber"]
        constructor = driver["Constructors"][0]["name"]
        if name == "Max Verstappen":
            number = "1" #Max's permanent number is 33 but he is current world champion so gets number 1.
        code = driver["Driver"]["code"] #Eg VER for Verstappen.
        thumbnailPath = f"/static/driverThumbnails/{code}.png"
        teamColour = "#FFFFFF" #default colour if driver's team colour can't be found.
        count = 0
        done = False
        while not done and count < len(driverData): #Use while so that can exit once colour found.
            selectedDriver = driverData[count]
            if code == selectedDriver["name_acronym"]: #check is the same driver by comparing the driver code/acronym (eg VER for Verstappen) - more unique than first name.
                teamColour = "#" + selectedDriver["team_colour"]
                done = True #don't have to check more driverData drivers as have found the one we need.
            count += 1
        tempDriverTable.append([position,number,points,wins,thumbnailPath,hexToRGB(teamColour)]) #adds all of the data as a row in the tempDriverTable list. teamColour only used for row colour, not displayed as text.
    
    driverTable = [tempDriverTable[0]]
    del tempDriverTable[0]
    numOfDrivers = len(tempDriverTable)
    numOfRows = numOfDrivers//2
    for i in range(numOfRows):
        row = []
        row.append(tempDriverTable[i])
        row.append(tempDriverTable[i+numOfRows])
        driverTable.append(row)

    return constructorTable, driverTable #returns both lists

def getdata(): #gets all data in format to be passed to tables.
    constructorTable, driverTable = getChampionshipsStandings()
    constructorHeadings = constructorTable[0]
    constructorData = constructorTable[1:]
    driverHeadings = driverTable[0]
    driverData = driverTable[1:]
    return constructorHeadings, constructorData, driverHeadings, driverData

@app.route("/")
def table():
    global appCounter
    appCounter += 1
    if appCounter >= 144:
        setup()
        appCounter = 0
    constructorHeadings, constructorData, driverHeadings, driverData = getdata()
    print(driverData)
    timeout = 3600000 #change timeout based on whether it is a session or not.
    return render_template("tables.html", constructorHeadings=constructorHeadings, constructorData=constructorData, driverHeadings=driverHeadings, driverData=driverData, timeout=timeout) #renders the html file, passing it the table headers and data, and telling how long before reload.

setup()
app.run()