summaryrefslogtreecommitdiff
path: root/jfr_playoff/data.py
diff options
context:
space:
mode:
Diffstat (limited to 'jfr_playoff/data.py')
-rw-r--r--jfr_playoff/data.py59
1 files changed, 56 insertions, 3 deletions
diff --git a/jfr_playoff/data.py b/jfr_playoff/data.py
index 6fc3c4b..47d66f5 100644
--- a/jfr_playoff/data.py
+++ b/jfr_playoff/data.py
@@ -7,11 +7,19 @@ from jfr_playoff.db import PlayoffDB
from jfr_playoff.dto import Match, Phase, Team
+SWISS_TIE_WARNING = 'WARNING: tie detected in swiss %s.' + \
+ ' Make sure to resolve the tie by arranging teams' + \
+ ' in configuration file.'
+
+
class PlayoffData(object):
def __init__(self, settings):
self.database = PlayoffDB(settings.get('database'))
self.phases = settings.get('phases')
self.teams = settings.get('teams')
+ self.swiss = []
+ if settings.has_section('swiss'):
+ self.swiss = settings.get('swiss')
self.grid = []
self.match_info = {}
self.leaderboard = []
@@ -52,16 +60,30 @@ class PlayoffData(object):
phase['link'], self.match_info[match['id']].link)
return self.match_info
- def get_match_link(self, match):
+ def __get_link(self, database, suffix):
try:
- row = self.database.fetch(match['database'], p_sql.PREFIX, ())
+ row = self.database.fetch(database, p_sql.PREFIX, ())
if row is not None:
if len(row) > 0:
- return '%srunda%d.html' % (row[0], match['round'])
+ return '%s%s' % (row[0], suffix)
except mysql.connector.Error:
return None
return None
+ def get_match_link(self, match):
+ return self.__get_link(
+ match['database'], 'runda%d.html' % (match['round']))
+
+ def get_leaderboard_link(self, database):
+ return self.__get_link(database, 'leaderb.html')
+
+ def get_swiss_link(self, event):
+ swiss_link = self.get_leaderboard_link(event['database'])
+ if ('relative_path' in event) and (
+ event['relative_path'] is not None):
+ swiss_link = '%s/%s' % (event['relative_path'], swiss_link)
+ return swiss_link
+
def get_db_match_teams(self, match):
teams = [Team(), Team()]
row = self.database.fetch(
@@ -152,8 +174,33 @@ class PlayoffData(object):
for team in teams:
if len(team) > 3:
self.leaderboard[team[3]-1] = team[0]
+ self.fill_swiss_leaderboard(self.swiss, teams)
return self.leaderboard
+ def fill_swiss_leaderboard(self, swiss, teams):
+ teams = [team[0] for team in teams]
+ for event in swiss:
+ swiss_finished = self.database.fetch(
+ event['database'], p_sql.SWISS_ENDED, {})
+ if swiss_finished[0] > 0:
+ swiss_teams = self.database.fetch_all(
+ event['database'], p_sql.SWISS_RESULTS, {})
+ swiss_results = [(teams.index(team[0]), team[1], team[2]) for
+ team in swiss_teams]
+ swiss_results = sorted(swiss_results, key=lambda t: t[0])
+ swiss_results = sorted(
+ swiss_results, key=lambda t: t[1], reverse=True)
+ prev_result = None
+ place = -1
+ for team in sorted(
+ swiss_results, key=lambda team: team[2]):
+ if prev_result == team[1]:
+ print SWISS_TIE_WARNING % (event['database'])
+ prev_result = team[1]
+ self.leaderboard[
+ event['position'] + place] = teams[team[0]]
+ place += 1
+
def fill_leaderboard(self):
self.prefill_leaderboard(self.teams)
leaderboard_teams = {}
@@ -181,6 +228,12 @@ class PlayoffData(object):
self.leaderboard[position-1] = table_team[0]
return self.leaderboard
+ def get_swiss_info(self):
+ return [{
+ 'link': self.get_swiss_link(event),
+ 'position': event['position']
+ } for event in self.swiss]
+
def get_dimensions(self):
return (
len(self.phases),