diff options
Diffstat (limited to 'jfr_playoff/data.py')
| -rw-r--r-- | jfr_playoff/data.py | 91 |
1 files changed, 29 insertions, 62 deletions
diff --git a/jfr_playoff/data.py b/jfr_playoff/data.py index 81f40d2..83f5d6c 100644 --- a/jfr_playoff/data.py +++ b/jfr_playoff/data.py @@ -1,14 +1,10 @@ import mysql from cached_property import cached_property -import jfr_playoff.sql as p_sql from jfr_playoff.db import PlayoffDB from jfr_playoff.dto import Phase from jfr_playoff.matchinfo import MatchInfo - -SWISS_TIE_WARNING = 'WARNING: tie detected in swiss %s.' + \ - ' Make sure to resolve the tie by arranging teams' + \ - ' in configuration file.' +from jfr_playoff.tournamentinfo import TournamentInfo class PlayoffData(object): @@ -29,13 +25,8 @@ class PlayoffData(object): def teams(self): if isinstance(self.team_settings, list): return self.team_settings - db_teams = self.get_swiss_results( - self.team_settings['database'], - self.team_settings['ties'] if 'ties' in self.team_settings else []) - if 'final_positions' in self.team_settings: - for position in self.team_settings['final_positions']: - db_teams[position-1].append(position) - return db_teams + tournament_info = TournamentInfo(self.team_settings, self.database) + return tournament_info.get_tournament_results() def generate_phases(self): self.grid = [] @@ -70,11 +61,9 @@ class PlayoffData(object): phase_obj.running = True return self.match_info - 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']) + event_info = TournamentInfo(event, self.database) + swiss_link = event_info.get_results_link() if ('relative_path' in event) and ( event['relative_path'] is not None): swiss_link = '%s/%s' % (event['relative_path'], swiss_link) @@ -93,52 +82,30 @@ class PlayoffData(object): if self.database is None: return for event in swiss: - swiss_finished = self.database.fetch( - event['database'], p_sql.SWISS_ENDED, {}) - if len(swiss_finished) > 0: - if swiss_finished[0] > 0: - swiss_position = ( - event['swiss_position'] - if 'swiss_position' in event - else 1 - ) - position_limit = ( - event['position_to'] - if 'position_to' in event - else 9999 - ) - place = 1 - swiss_results = self.get_swiss_results( - event['database'], teams) - for team in swiss_results: - if place >= swiss_position: - target_position = event['position'] \ - + place - swiss_position - if target_position <= min( - position_limit, len(self.leaderboard)): - self.leaderboard[ - target_position - 1] = team[0] - place += 1 - - def get_swiss_results(self, swiss, ties=None): - if ties is None: - ties = [] - if self.database is None: - return [] - swiss_teams = self.database.fetch_all( - swiss, p_sql.SWISS_RESULTS, {}) - swiss_results = sorted( - swiss_teams, - key=lambda t: ties.index(t[0]) if t[0] in ties else -1) - swiss_results = sorted( - swiss_results, key=lambda t: t[1], reverse=True) - swiss_results = sorted(swiss_results, key=lambda team: team[2]) - prev_result = None - for team in swiss_results: - if prev_result == team[1]: - print SWISS_TIE_WARNING % (swiss) - prev_result = team[1] - return [[team[0], team[3], team[4]] for team in swiss_results] + event['ties'] = teams + event_info = TournamentInfo(event, self.database) + if event_info.is_finished(): + swiss_position = ( + event['swiss_position'] + if 'swiss_position' in event + else 1 + ) + position_limit = ( + event['position_to'] + if 'position_to' in event + else 9999 + ) + place = 1 + swiss_results = event_info.get_tournament_results() + for team in swiss_results: + if place >= swiss_position: + target_position = event['position'] \ + + place - swiss_position + if target_position <= min( + position_limit, len(self.leaderboard)): + self.leaderboard[ + target_position - 1] = team[0] + place += 1 def fill_leaderboard(self): self.prefill_leaderboard(self.teams) |