data.py

import requests
import json
import csv
import datetime
import os

os.chdir(os.path.dirname(os.path.abspath(__file__))) # MUST COME BEFORE RELATIVE IMPORTS

class ResponseError(Exception):
    def __init__(self, status_code, data):
        self.status_code = status_code
        self.body = data.text
        self.headers = data.headers
        self.request_headers = data.request.headers
        super().__init__(self.body)

    def __str__(self):
        return f'--- Response Error ---\nStatus Code: {self.status_code}\n\nHeaders:\nRequest: {self.request_headers}\nResponse: {self.headers}\n\nBody:\n{self.body}'


class MatchData():
    df = 'matchdata.csv'

    def fetch_data(self):
        http_headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.212 Safari/537.36'}
        csvHeader = "FanFootyDrawID,year,competition,round,gametime,day,homeTeam,awayTeam,ground,timeslot,TVCoverage,homeSupergoals,homeGoals,homeBehinds,homePoints,awaySupergoals,awayGoals,awayBehinds,awayPoints,matchStatus\n"

        # Get data from URL
        url = 'http://www.fanfooty.com.au/resource/draw.php'
        r = requests.get(url, headers=http_headers)

        # Write to file
        if r.status_code == 200:
            f = open(self.df, 'w', newline='')
            f.write(csvHeader)
            f.write(r.text.strip())
            f.close()
        else:
            raise ResponseError(r.status_code, r)
        return

    def get_rounds(self, year):
        currentDate = datetime.datetime.now()

        ha = []
        preliminary = []
        finals = []

        currentRound = 0
        currentComp = None
        cRound = -1
        cRoundBeenSet = False

        with open(self.df) as csvfile:
            reader = csv.DictReader(csvfile)
            for row in reader:
                if row['year'] == year:
                    if row['competition'][0] == "P" and int(row['round']) == 0 and currentComp != row['competition']:
                        currentComp = row['competition']
                        preliminary.append(row['competition'])

                    if row['competition'] == "HA" and int(row['round']) != currentRound:
                        currentRound = int(row['round'])
                        ha.append(row['round'])

                    if row['competition'][-1] == "F":
                        final = ""

                        if currentComp != row['competition'] and currentRound != int(row['round']):
                            final = row['competition']

                        if currentComp != row['competition'] and currentRound == int(row['round']):
                            if finals[-1].find(row['competition']) == -1:
                                finals[-1] += " and " + row['competition']

                        if final != "":
                            finals.append(final)

                        currentRound = int(row['round'])
                        currentComp = row['competition']

                if datetime.datetime.strptime(row['gametime'], "%Y-%m-%d %H:%M:%S") > currentDate and not cRoundBeenSet:
                    cRound = str(currentRound)
                    cRoundBeenSet = True

            dump = json.dumps(
                {"preliminary": preliminary, "HA": ha, "finals": finals, "currentRound": cRound})

        return dump

    def get_matches(self, year, round):
        matches = []

        # Rewrite this peice of shit, possibly an array of rounds (competitions)
        if round == "EF and QF" or round == "QF and EF":
            with open(self.df) as csvfile:
                reader = csv.DictReader(csvfile)
                for row in reader:
                    if row['year'] == year and (row['round'] == round or row['competition'] == "EF" or row['competition'] == "QF"):
                        matches.append(row)

            dump = json.dumps(matches)

            return dump

        with open(self.df) as csvfile:
            reader = csv.DictReader(csvfile)
            for row in reader:
                if row['year'] == year and (row['round'] == round or row['competition'] == round):
                    matches.append(row)

        dump = json.dumps(matches)

        return dump