diff options
Diffstat (limited to 'jfr_playoff/data')
| -rw-r--r-- | jfr_playoff/data/__init__.py | 19 | ||||
| -rw-r--r-- | jfr_playoff/data/info.py | 12 |
2 files changed, 25 insertions, 6 deletions
diff --git a/jfr_playoff/data/__init__.py b/jfr_playoff/data/__init__.py index 3e63e98..b5c9ceb 100644 --- a/jfr_playoff/data/__init__.py +++ b/jfr_playoff/data/__init__.py @@ -8,6 +8,7 @@ from jfr_playoff.logger import PlayoffLogger class PlayoffData(object): def __init__(self, settings=None): + self._team_list_certain = True if settings is not None: self.database = PlayoffDB(settings.get('database')) \ if settings.has_section('database') else None @@ -31,6 +32,8 @@ class PlayoffData(object): self.aliases = {} if settings.has_section('team_aliases'): self.aliases = settings.get('team_aliases') + self._predict_teams = int(settings.get('page').get( + 'team_boxes', {}).get('predict_teams', 0)) > 0 self.grid = [] self.match_info = {} self.leaderboard = [] @@ -39,17 +42,20 @@ class PlayoffData(object): if isinstance(settings, list): PlayoffLogger.get('data').info( 'team list pre-defined: %s', settings) - return settings + return settings, True 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 \ + teams = team_list if 'max_teams' not in settings \ else team_list[0:settings['max_teams']] + return teams, tournament_info.is_finished() @cached_property def teams(self): - return self.fetch_team_list(self.team_settings, self.database) + team_list, certain = self.fetch_team_list(self.team_settings, self.database) + self._team_list_certain = certain + return team_list def generate_phases(self): self.grid = [] @@ -76,8 +82,13 @@ class PlayoffData(object): for match in phase['matches']: PlayoffLogger.get('data').info( 'getting match info for #%d', match['id']) + teams = self.teams + certain_starting_positions = self._team_list_certain \ + if self._predict_teams > 0 else True match_info = MatchInfo( - match, self.teams, self.database, self.aliases) + match, teams, self.database, + self.aliases, + certain_starting_positions) if 'link' in phase: match_info.set_phase_link(phase['link']) self.match_info[match['id']] = match_info.get_info() diff --git a/jfr_playoff/data/info.py b/jfr_playoff/data/info.py index 55cc926..f79499b 100644 --- a/jfr_playoff/data/info.py +++ b/jfr_playoff/data/info.py @@ -100,7 +100,8 @@ class MatchInfo(ResultInfo): matches = {} - def __init__(self, match_config, teams, database, aliases=None): + def __init__(self, match_config, teams, database, + aliases=None, starting_positions_certain=True): ResultInfo.__init__(self, match_config, database) self.config = match_config self.teams = teams @@ -110,6 +111,7 @@ class MatchInfo(ResultInfo): for team, team_aliases in aliases.iteritems(): for alias in team_aliases: self.aliases[alias] = team + self._starting_positions_certain = starting_positions_certain self.info = Match() self._init_info() self._fetch_match_link() @@ -193,9 +195,15 @@ class MatchInfo(ResultInfo): MatchInfo.matches[loser_match].possible_loser for loser_match in self.config['teams'][i]['loser']] if 'place' in self.config['teams'][i]: - match_teams += [ + placed_teams = [ self.teams[place-1][0] for place in self.config['teams'][i]['place']] + if self._starting_positions_certain: + match_teams += placed_teams + possible_teams = [None] * len(placed_teams) + else: + possible_teams += placed_teams + match_teams = [None] * len(placed_teams) teams[i].name = match_teams teams[i].possible_name = possible_teams teams[i].known_teams = len([ |