from math import ceil import re import jfr_playoff.sql as p_sql from jfr_playoff.remote import RemoteUrl as p_remote from jfr_playoff.logger import PlayoffLogger SWISS_TIE_WARNING = 'tie detected in swiss %s.' + \ ' Make sure to resolve the tie by arranging teams' + \ ' in configuration file.' class TournamentInfo: def __init__(self, settings, database): self.settings = settings self.database = database def __get_html_results(self): if 'link' not in self.settings: raise KeyError('link not configured') if not self.settings['link'].endswith('leaderb.html'): raise ValueError('invalid link to tournament results') PlayoffLogger.get('tournamentinfo').info( 'fetching tournament results from leaderboard URL: %s', self.settings['link']) leaderboard = p_remote.fetch(self.settings['link']) result_links = [row.select('a[onmouseover]') for row in leaderboard.find_all('tr') if len(row.select('a[onmouseover]')) > 0] results = [None] * (len(result_links) * max([len(links) for links in result_links])) for i in range(0, len(result_links)): for j in range(0, len(result_links[i])): results[len(result_links) * j + i] = result_links[i][j] teams = [] team_links = {} for team in results: if team is not None: team_info = [] fullname = team.text.strip(u'\xa0') team_links[team['href']] = fullname team_info.append(fullname) team_info.append('') team_image = team.find('img') if team_image is not None: team_info.append(team_image['src'].replace('images/', '')) teams.append(team_info) PlayoffLogger.get('tournamentinfo').info( 'read tournament results from leaderboard: %s', teams) for table in range(1, int(ceil(len(teams)/2.0))+1): table_url = self.get_results_link('1t%d-1.html' % (table)) table_content = p_remote.fetch(table_url) PlayoffLogger.get('tournamentinfo').info( 'reading team shortnames from traveller: %s', table_url) for link in table_content.select('a.br'): if link['href'] in team_links: for team in teams: if team[0] == team_links[link['href']]: team[1] = link.text.strip(u'\xa0') PlayoffLogger.get('tournamentinfo').info( 'shortname for %s: %s', team[0], team[1]) break PlayoffLogger.get('tournamentinfo').info( 'tournament results from HTML: %s', teams) return teams def __get_db_results(self): if self.database is None: raise KeyError('database not configured') if 'database' not in self.settings: raise KeyError('database not configured') if 'ties' not in self.settings: self.settings['ties'] = [] swiss_teams = self.database.fetch_all( self.settings['database'], p_sql.SWISS_RESULTS, {}) swiss_results = sorted( swiss_teams, key=lambda t: self.settings['ties'].index(t[0]) \ if t[0] in self.settings['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]) PlayoffLogger.get('tournamentinfo').info( 'fetched tournament results from database %s: %s', self.settings['database'], swiss_results) prev_result = None for team in swiss_results: if prev_result == team[1]: PlayoffLogger.get('tournamentinfo').warning( SWISS_TIE_WARNING, self.settings['database']) prev_result = team[1] db_teams = [[team[0], team[3], team[4]] for team in swiss_results] PlayoffLogger.get('tournamentinfo').info( 'fetched team list from database %s: %s', self.settings['database'], db_teams) return db_teams def __get_html_finished(self): if 'link' not in self.settings: raise KeyError('link not configured') if not self.settings['link'].endswith('leaderb.html'): raise ValueError('invalid tournament leaderboard link') PlayoffLogger.get('tournamentinfo').info( 'fetching tournament finished status from HTML: %s', self.settings['link']) leaderboard = p_remote.fetch(self.settings['link']) leaderb_heading = leaderboard.select('td.bdnl12')[0].text contains_digits = any(char.isdigit() for char in leaderb_heading) PlayoffLogger.get('tournamentinfo').info( 'tournament header from HTML: %s, %s', leaderb_heading, 'contains digits' if contains_digits else "doesn't contain digits") non_zero_scores = [imps.text for imps in leaderboard.select('td.bdc small') if imps.text != '0-0'] PlayoffLogger.get('tournamentinfo').info( 'tournament leaderboard from HTML: has %d non-zero scores', len(non_zero_scores)) finished = (not contains_digits) and (len(non_zero_scores) > 0) PlayoffLogger.get('tournamentinfo').info( 'tournament leaderboard from HTML indicates finished: %s', finished) return finished def __get_db_finished(self): if self.database is None: raise KeyError('database not configured') if 'database' not in self.settings: raise KeyError('database not configured') finished = self.database.fetch( self.settings['database'], p_sql.SWISS_ENDED, {}) PlayoffLogger.get('tournamentinfo').info( 'fetching tournament finished status from DB %s: %s', self.settings['database'], finished) return (len(finished) > 0) and (finished[0] > 0) def __get_html_link(self, suffix='leaderb.html'): if 'link' not in self.settings: raise KeyError('link not configured') if not self.settings['link'].endswith('leaderb.html'): raise ValueError('invalid tournament leaderboard link') link = re.sub(r'leaderb.html$', suffix, self.settings['link']) PlayoffLogger.get('tournamentinfo').info( 'generating tournament-specific link from leaderboard link %s: %s -> %s', self.settings['link'], suffix, link) return link def __get_db_link(self, suffix='leaderb.html'): if self.database is None: raise KeyError('database not configured') if 'database' not in self.settings: raise KeyError('database not configured') row = self.database.fetch( self.settings['database'], p_sql.PREFIX, ()) if row is not None: if len(row) > 0: link = row[0] + suffix PlayoffLogger.get('tournamentinfo').info( 'generating tournament-specific link from DB %s prefix: %s -> %s', self.settings['database'], suffix, link) return link raise ValueError('unable to fetch db link') def get_tournament_results(self): teams = [] try: teams = self.__get_db_results() except (IOError, TypeError, IndexError, KeyError) as e: PlayoffLogger.get('tournamentinfo').warning( 'cannot determine tournament results from DB: %s(%s)', type(e).__name__, e.message) try: teams = self.__get_html_results() except (TypeError, IndexError, KeyError, IOError, ValueError) as e: PlayoffLogger.get('tournamentinfo').warning( 'cannot determine tournament results from HTML: %s(%s)', type(e).__name__, e.message) if self.is_finished() and 'final_positions' in self.settings: PlayoffLogger.get('tournamentinfo').info( 'setting final positions from tournament results: %s', self.settings['final_positions']) for position in self.settings['final_positions']: if len(teams) >= position: teams[position-1].append(position) return teams def is_finished(self): try: return self.__get_db_finished() except (IOError, TypeError, IndexError, KeyError) as e: PlayoffLogger.get('tournamentinfo').warning( 'cannot determine tournament finished status from DB: %s(%s)', type(e).__name__, e.message) try: return self.__get_html_finished() except (TypeError, IndexError, KeyError, IOError, ValueError) as e: PlayoffLogger.get('tournamentinfo').warning( 'cannot determine tournament finished status from HTML: %s(%s)', type(e).__name__, e.message) PlayoffLogger.get('tournamentinfo').info( 'assuming tournament is finished') return True def get_results_link(self, suffix='leaderb.html'): try: return self.__get_db_link(suffix) except (IOError, TypeError, IndexError, KeyError) as e: PlayoffLogger.get('tournamentinfo').warning( 'cannot determine tournament link from DB: %s(%s)', type(e).__name__, e.message) try: return self.__get_html_link(suffix) except (KeyError, ValueError): PlayoffLogger.get('tournamentinfo').warning( 'cannot determine tournament link from HTML: %s(%s)', type(e).__name__, e.message) return None