diff options
| -rw-r--r-- | jfr_playoff/data/match/tcjson.py | 91 |
1 files changed, 90 insertions, 1 deletions
diff --git a/jfr_playoff/data/match/tcjson.py b/jfr_playoff/data/match/tcjson.py index fbc28ce..2c4ea0a 100644 --- a/jfr_playoff/data/match/tcjson.py +++ b/jfr_playoff/data/match/tcjson.py @@ -1,6 +1,8 @@ +import json import urlparse from jfr_playoff.data.match import MatchInfoClient +from jfr_playoff.remote import RemoteUrl as p_remote from jfr_playoff.logger import PlayoffLogger @@ -15,12 +17,99 @@ class TCJsonMatchInfo(MatchInfoClient): def get_exceptions(self, method): return (TypeError, IndexError, KeyError, IOError, ValueError) + def _get_results_link(self, suffix): + return urlparse.urljoin(self.settings['link'], suffix) + def _get_round_from_link(self, link): fragment = urlparse.urlparse(link).fragment - return fragment[11:14], fragment[14:17] + return int(fragment[11:14]), int(fragment[14:17]) def get_match_link(self): PlayoffLogger.get('match.tcjson').info( 'match #%d link pre-defined: %s', self.settings['id'], self.settings['link']) return self.settings['link'] + + def _get_results(self, link, table_no): + session_no, round_no = self._get_round_from_link(link) + PlayoffLogger.get('match.tcjson').info( + 'link %s -> session %d, round %d', link, session_no, round_no) + round_results = json.loads( + p_remote.fetch_raw( + self._get_results_link( + 'o%d-%d.json' % (session_no, round_no)))) + tables = [] + for result in round_results['Results']: + tables.append(result['Table']) + table_id = sorted(tables)[table_no-1] + PlayoffLogger.get('match.tcjson').info( + 'table no %d from %s: %s', table_no, tables, table_id) + for result in round_results['Results']: + if result['Table'] == table_id: + return result + PlayoffLogger.get('match.tcjson').info( + 'results for table %d not found in %s', table_no, link) + raise ValueError('results not found') + + def fetch_teams(self, teams): + results = self._get_results( + self.settings['link'], self.settings['table']) + for idx, side in enumerate(['Ns', 'Ew']): + teams[idx].name = [results[side]['_name']] + teams[idx].known_teams = 1 + teams[idx].score = results['Sum1'+side] + PlayoffLogger.get('match.tcjson').info( + 'scores fetched: %s', teams) + return teams + + def board_count(self): + results = self._get_results( + self.settings['link'], self.settings['table']) + played = 0 + finished = True + for segment in results['Segments']: + PlayoffLogger.get('match.tcjson').info( + 'segment %d: played boards=%d, live=%s, towel=%s', + segment['Segment']+1, segment['BoardsCounted'], + segment['Live'], segment['Towel']) + played += segment['BoardsCounted'] + if segment['Live'] or not ( + (segment['BoardsCounted'] or segment['Towel'])): + PlayoffLogger.get('match.tcjson').info( + 'segment %d not finished', segment['Segment']+1) + finished = False + PlayoffLogger.get('match.tcjson').info( + 'board count: %d, finished: %s', played, finished) + if finished and not played: + # toweled match + played = 1 + return played, played if finished else played+1 + + def _get_segment_link(self, base_link, segment_id, table_id): + table_id = '{t:{pad}>6}'.format(t=table_id, pad='0') + session_id, round_id = self._get_round_from_link(base_link) + fragment = '#000SS000000%03d%03d%03d%s' % ( + session_id, round_id, segment_id, table_id) + return urlparse.urljoin(base_link, fragment) + + def running_link(self): + settings = json.loads(p_remote.fetch_raw( + self._get_results_link('settings.json'))) + if settings['ShowOnlyResults']: + PlayoffLogger.get('match.tcjson').info( + 'ShowOnlyResults active, no running segment link available') + return self.settings['link'] + results = self._get_results( + self.settings['link'], self.settings['table']) + for segment in results['Segments']: + if segment['Live']: + link = self._get_segment_link( + self.settings['link'], + segment['Segment'], results['TableFull']) + PlayoffLogger.get('match.tcjson').info( + 'running segment link for segment %d, table %s: %s', + segment['Segment']+1, results['TableFull'], link) + return link + PlayoffLogger.get('match.tcjson').info( + 'no running segment link available') + return self.settings['link'] |