diff options
Diffstat (limited to 'jfr_playoff/data/info.py')
| -rw-r--r-- | jfr_playoff/data/info.py | 76 |
1 files changed, 51 insertions, 25 deletions
diff --git a/jfr_playoff/data/info.py b/jfr_playoff/data/info.py index aec78c2..230196d 100644 --- a/jfr_playoff/data/info.py +++ b/jfr_playoff/data/info.py @@ -1,4 +1,7 @@ +import glob +import inspect import re +from os.path import dirname, basename, isfile, join from urlparse import urljoin import jfr_playoff.sql as p_sql @@ -7,12 +10,46 @@ from jfr_playoff.remote import RemoteUrl as p_remote from jfr_playoff.logger import PlayoffLogger +class ResultInfoClient(object): + def __init__(self, settings, database=None): + self.settings = settings + self.database = database + + @property + def priority(self): + return 0 + + def is_capable(self): + return False + + def get_exceptions(self, method): + pass + + class ResultInfo(object): def __init__(self, *args): - self.clients = self.fill_client_list(*args) - - def fill_client_list(self, settings, database): - return [] + self.clients = self._fill_client_list(*args) + + @property + def submodule_path(self): + raise NotImplementedError() + + @property + def _client_classes(self): + module = __import__(self.submodule_path, fromlist=['']) + for f in glob.glob(join(dirname(module.__file__), "*.py")): + if isfile(f) and not f.endswith('__init__.py'): + submodule_name = basename(f)[:-3] + submodule_path = self.submodule_path + '.' + submodule_name + submodule = __import__(submodule_path, fromlist=['']) + for member in inspect.getmembers(submodule, inspect.isclass): + if member[1].__module__ == submodule_path: + yield member[1] + + def _fill_client_list(self, *args): + all_clients = [c(*args) for c in self._client_classes] + clients = [c for c in all_clients if c.is_capable()] + return sorted(clients, key=lambda c: c.priority, reverse=True) def call_client(self, method, default, *args): PlayoffLogger.get('resultinfo').info( @@ -35,34 +72,22 @@ class ResultInfo(object): return default -from jfr_playoff.data.tournament.jfrdb import JFRDbTournamentInfo -from jfr_playoff.data.tournament.jfrhtml import JFRHtmlTournamentInfo -from jfr_playoff.data.tournament.tcjson import TCJsonTournamentInfo - - class TournamentInfo(ResultInfo): def __init__(self, settings, database): - self.settings = settings ResultInfo.__init__(self, settings, database) + self.final_positions = settings.get('final_positions', []) - def fill_client_list(self, settings, database): - clients = [] - if (database is not None) and ('database' in settings): - clients.append(JFRDbTournamentInfo(settings, database)) - if 'link' in settings: - if settings['link'].endswith('leaderb.html'): - clients.append(JFRHtmlTournamentInfo(settings)) - clients.append(TCJsonTournamentInfo(settings)) - return clients + @property + def submodule_path(self): + return 'jfr_playoff.data.tournament' def get_tournament_results(self): teams = self.call_client('get_tournament_results', []) if self.is_finished(): - final_positions = self.settings.get('final_positions', []) PlayoffLogger.get('tournamentinfo').info( 'setting final positions from tournament results: %s', - final_positions) - for position in final_positions: + self.final_positions) + for position in self.final_positions: if len(teams) >= position: teams[position-1] = (teams[position-1] + [None] * 4)[0:4] teams[position-1][3] = position @@ -80,6 +105,7 @@ class MatchInfo(ResultInfo): matches = {} def __init__(self, match_config, teams, database, aliases=None): + ResultInfo.__init__(self, match_config, database) self.config = match_config self.teams = teams self.database = database @@ -88,13 +114,13 @@ class MatchInfo(ResultInfo): for team, team_aliases in aliases.iteritems(): for alias in team_aliases: self.aliases[alias] = team - ResultInfo.__init__(self, match_config, database) self.info = Match() self.__init_info() self.__fetch_match_link() - def fill_client_list(self, settings, database): - return [] + @property + def submodule_path(self): + return 'jfr_playoff.data.match' def __init_info(self): self.info.id = self.config['id'] |