diff options
Diffstat (limited to 'jfr_playoff/data/__init__.py')
| -rw-r--r-- | jfr_playoff/data/__init__.py | 207 |
1 files changed, 207 insertions, 0 deletions
diff --git a/jfr_playoff/data/__init__.py b/jfr_playoff/data/__init__.py new file mode 100644 index 0000000..494eaef --- /dev/null +++ b/jfr_playoff/data/__init__.py @@ -0,0 +1,207 @@ +from cached_property import cached_property + +from jfr_playoff.db import PlayoffDB +from jfr_playoff.dto import Phase +from jfr_playoff.matchinfo import MatchInfo +from jfr_playoff.data.tournament import TournamentInfo +from jfr_playoff.logger import PlayoffLogger + + +class PlayoffData(object): + def __init__(self, settings=None): + if settings is not None: + self.database = PlayoffDB(settings.get('database')) \ + if settings.has_section('database') else None + if self.database is None: + PlayoffLogger.get('db').warning( + PlayoffDB.DATABASE_NOT_CONFIGURED_WARNING) + self.team_settings = settings.get('teams') + self.custom_final_order = [] + if settings.has_section('custom_final_order'): + self.custom_final_order = settings.get('custom_final_order') + self.custom_final_order = [ + t for t in [ + self.teams[i-1] if isinstance(i, int) + else self.get_team_data_by_name(i) + for i in self.custom_final_order] + if t is not None] + self.phases = settings.get('phases') + self.swiss = [] + if settings.has_section('swiss'): + self.swiss = settings.get('swiss') + self.aliases = {} + if settings.has_section('team_aliases'): + self.aliases = settings.get('team_aliases') + self.grid = [] + self.match_info = {} + self.leaderboard = [] + + def fetch_team_list(self, settings, db_interface): + if isinstance(settings, list): + PlayoffLogger.get('data').info( + 'team list pre-defined: %s', settings) + return settings + tournament_info = TournamentInfo(settings, db_interface) + team_list = tournament_info.get_tournament_results() + if len(team_list) == 0: + PlayoffLogger.get('data').warning('team list is empty!') + return team_list if 'max_teams' not in settings \ + else team_list[0:settings['max_teams']] + + @cached_property + def teams(self): + return self.fetch_team_list(self.team_settings, self.database) + + def generate_phases(self): + self.grid = [] + for phase in self.phases: + dummies = phase.get('dummies', []) + phase_count = len(phase['matches']) + len(dummies) + phase_object = Phase() + phase_object.title = phase['title'] + phase_object.link = phase.get('link', None) + phase_object.matches = [None] * phase_count + phase_pos = 0 + for match in phase['matches']: + while phase_pos in dummies: + phase_pos += 1 + phase_object.matches[phase_pos] = match['id'] + phase_pos += 1 + PlayoffLogger.get('data').info('phase object: %s', phase_object) + self.grid.append(phase_object) + return self.grid + + def fill_match_info(self): + self.match_info = {} + for phase in self.phases: + for match in phase['matches']: + match_info = MatchInfo(match, self.teams, self.database, self.aliases) + if 'link' in phase: + match_info.set_phase_link(phase['link']) + self.match_info[match['id']] = match_info.get_info() + if self.match_info[match['id']].running > 0: + for phase_obj in self.grid: + if match['id'] in phase_obj.matches: + phase_obj.running = True + PlayoffLogger.get('data').info( + 'match object: %s', self.match_info[match['id']]) + return self.match_info + + def get_swiss_link(self, event): + event_info = TournamentInfo(event, self.database) + swiss_link = event_info.get_results_link() + if event.get('relative_path', None): + swiss_link = '%s/%s' % (event['relative_path'], swiss_link) + PlayoffLogger.get('data').info('swiss link: %s', swiss_link) + return swiss_link + + def prefill_leaderboard(self, teams): + self.leaderboard = [None] * len(teams) + for team in teams: + if len(team) > 3 and team[3] is not None: + self.leaderboard[team[3]-1] = team[0] + self.fill_swiss_leaderboard(self.swiss, teams) + PlayoffLogger.get('data').info('leaderboard pre-filled: %s', self.leaderboard) + return self.leaderboard + + def fill_swiss_leaderboard(self, swiss, teams): + teams = [team[0] for team in teams] + for event in swiss: + event['ties'] = teams + event_info = TournamentInfo(event, self.database) + if event_info.is_finished(): + swiss_position = event.get('swiss_position', 1) + position_limit = event.get('position_to', 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 + PlayoffLogger.get('data').info( + 'leaderboard after %s swiss: %s', event, self.leaderboard) + + def fill_leaderboard(self): + self.prefill_leaderboard(self.teams) + leaderboard_teams = {} + for phase in self.phases: + for match in phase['matches']: + if 'winner' in match: + winner_key = tuple(match['winner']) + if winner_key not in leaderboard_teams: + leaderboard_teams[winner_key] = [] + leaderboard_teams[winner_key].append( + self.match_info[match['id']].winner) + if 'loser' in match: + loser_key = tuple(match['loser']) + if loser_key not in leaderboard_teams: + leaderboard_teams[loser_key] = [] + leaderboard_teams[loser_key].append( + self.match_info[match['id']].loser) + final_order = self.custom_final_order + [ + t for t in self.teams if t not in self.custom_final_order] + PlayoffLogger.get('data').info( + 'custom order for final positions: %s', self.custom_final_order) + PlayoffLogger.get('data').info( + 'order of teams to fill leaderboard positions: %s', + final_order) + for positions, position_teams in leaderboard_teams.iteritems(): + positions = list(positions) + PlayoffLogger.get('data').info( + 'filling leaderboard positions %s with teams %s', + positions, position_teams) + if len(positions) == len([ + team for team in position_teams if team is not None]): + for table_team in final_order: + if table_team[0] in position_teams: + position = positions.pop(0) + self.leaderboard[position-1] = table_team[0] + PlayoffLogger.get('data').info( + 'team %s in position %d', table_team[0], position) + PlayoffLogger.get('data').info( + 'leaderboard filled: %s', self.leaderboard) + return self.leaderboard + + def get_swiss_info(self): + swiss_info = [{ + 'link': self.get_swiss_link(event), + 'position': event['position'], + 'label': event.get('label', None), + 'finished': TournamentInfo(event, self.database).is_finished() + } for event in self.swiss] + PlayoffLogger.get('data').info('swiss info: %s', swiss_info) + return swiss_info + + def get_dimensions(self): + dimensions = ( + len(self.phases), + max([ + len(phase['matches']) + len(phase.get('dummies', [])) + for phase in self.phases + ] or [0]) + ) + PlayoffLogger.get('data').info('grid dimensions: %s', dimensions) + return dimensions + + def get_team_data_by_name(self, fullname): + for team in self.teams: + if team[0] == fullname: + return team + return None + + def get_shortname(self, fullname): + for team in self.teams: + if team[0] == fullname: + return team[1] + return fullname + + def get_team_image(self, fullname): + for team in self.teams: + if team[0] == fullname and len(team) > 2: + return team[2] + return None |