Graph-Fixturing/fixturing_2018b.py at master · robfox92/Graph-Fixturing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
from fixturelib import *


rootURL = 'https://docs.google.com/spreadsheets/d/10KrdFgNjH-L0NxBmXEl60beEEIgdtTzBElKM0k2Q3S8/export?format=xlsx&id=10KrdFgNjH-L0NxBmXEl60beEEIgdtTzBElKM0k2Q3S8'
season = "2018b"

print("Enter Round Number")
roundNumber=int(input())

print("Retrieving Results from remote")
mixedResults = getResults(URL=rootURL, table='Mixed-Scores')
ladiesResults = getResults(URL=rootURL, table='Ladies-Scores')

# Grab the rating data and unpack it
ladiesRatings = getRatings(URL=rootURL, table='Ladies-Starting Elos', teamNameCol='TEAM NAME', teamEloCol='STARTING ELO', teamKCol='K Value')
mixedRatings = getRatings(URL=rootURL, table='Mixed-Starting Elos', teamNameCol='TEAM NAME',teamEloCol='STARTING ELO', teamKCol='K Value')

ladiesFixtured = list(getDataFromRemote(URL=rootURL, table='Ladies-Fixtured Games')['Game Code'])
mixedFixtured = list(getDataFromRemote(URL=rootURL, table='Mixed-Fixtured Games')['Game Code'])
mixedRequested = list(getDataFromRemote(URL=rootURL, table='Mixed-Requests')['Game Code'])
ladiesRequested = list(getDataFromRemote(URL=rootURL, table='Ladies-Requests')['Game Code'])
# Init these as empty lists
ladiesAntiRequested = list()
mixedAntiRequested = list(getDataFromRemote(URL=rootURL,table='Mixed-Antirequests')['Game Code'])
print("Successfully retrieved Results from Remote")

mixedStartingElos = mixedRatings[0]
mixedKVals = mixedRatings[1]
mixedTeams = mixedRatings[2]
ladiesStartingElos = ladiesRatings[0]
ladiesKVals = ladiesRatings[1]
ladiesTeams = ladiesRatings[2]

# Process the results to update the Elos
print("Parsing results to update Ratings")
ladiesElos = updateElosFromResults(elos=ladiesStartingElos,results=ladiesResults,
        kValues=ladiesKVals)
mixedElos = updateElosFromResults(elos=mixedStartingElos,results=mixedResults,
        kValues=mixedKVals)
print("Successfully parsed results to update Ratings\nFinding Game Ratings")
# Create the Game Ratings Graph
mixedGameRatings = createGameRatingsGraph(fixturedGames=mixedFixtured,
        requestedGames=mixedRequested,antiRequestedGames=mixedAntiRequested,elosDict=mixedElos)
ladiesGameRatings = createGameRatingsGraph(fixturedGames=ladiesFixtured,
        requestedGames=ladiesRequested,antiRequestedGames=ladiesAntiRequested,elosDict=ladiesElos)
mixedHomeGameCounts = getHomeGameCounts(mixedTeams,mixedFixtured)
ladiesHomeGameCounts = getHomeGameCounts(ladiesTeams,ladiesFixtured)
print("Successfully found Game Ratings")

mixedFixture = None
ladiesFixture = None

print("Fixturing Ladies Teams")
if len(ladiesTeams)%2 == 0:
    ladiesRoundNumber = str(roundNumber)
    ladiesFixture = fixtureSingleRound(teams=ladiesTeams, elos=ladiesElos,
            fixtured=ladiesFixtured, requested=ladiesRequested,
            antiRequested=ladiesAntiRequested,rematchesAllowed=0)
elif len(ladiesTeams)%2 == 1 and roundNumber%2 == 1:
    ladiesRoundNumber = "%i-%i" %(roundNumber,roundNumber+1)
    ladiesFixture = fixtureDoubleRound(teams=ladiesTeams, elos=ladiesElos,
            fixtured=ladiesFixtured, requested=ladiesRequested,
            antiRequested=ladiesAntiRequested,rematchesAllowed=0)

print("Ladies Fixture Complete.\nFixturing Mixed Teams")
if len(mixedTeams)%2 == 0:
    mixedRoundNumber = str(roundNumber)
    mixedFixture = fixtureSingleRound(teams=mixedTeams, elos=mixedElos,
            fixtured=mixedFixtured, requested=mixedRequested,
            antiRequested=mixedAntiRequested,rematchesAllowed=0)
elif len(mixedTeams)%2 == 1 and roundNumber%2 == 1:
    mixedRoundNumber = "%i-%i" %(roundNumber, roundNumber+1)
    mixedFixture = fixtureDoubleRound(teams=mixedTeams, elos=mixedElos,
            fixtured=mixedFixtured, requested=mixedRequested,
            antiRequested=mixedAntiRequested,rematchesAllowed=0)
print("Mixed Fixture Complete.\nWriting to CSV")
if mixedFixture is not None:
    mixedPath = "Mixed Round %s Fixtures %s.csv" %(mixedRoundNumber, season)
    mixedFixture.to_csv(path_or_buf=mixedPath,encoding='utf-8')

if ladiesFixture is not None:
    ladiesPath = "Ladies Round %s Fixtures %s.csv" %(ladiesRoundNumber, season)
    ladiesFixture.to_csv(path_or_buf=ladiesPath,encoding='utf-8')
print("Script Complete.")