1 files changed, 126 insertions, 0 deletions
diff --git a/jfr_playoff/generator.py b/jfr_playoff/generator.py
new file mode 100644
index 0000000..5868750
--- /dev/null
+++ b/jfr_playoff/generator.py
@@ -0,0 +1,126 @@
+from datetime import datetime
+
+import jfr_playoff.template as p_temp
+from jfr_playoff.data import PlayoffData
+
+
+class PlayoffGenerator(object):
+    def __init__(self, settings):
+        self.data = PlayoffData(settings)
+        self.page = settings.get('page')
+        self.canvas = {}
+        if settings.has_section('canvas'):
+            self.canvas = settings.get('canvas')
+
+    def generate_content(self):
+        return p_temp.PAGE % (
+            p_temp.PAGE_HEAD % (
+                p_temp.PAGE_HEAD_REFRESH % (
+                    self.page['refresh'])
+                if self.page['refresh'] > 0 else '',
+                self.page['title']),
+            p_temp.PAGE_BODY % (
+                self.page['logoh'],
+                self.get_match_grid(
+                    self.data.get_dimensions(),
+                    self.data.generate_phases(),
+                    self.data.fill_match_info()),
+                self.get_leaderboard_table(self.data.fill_leaderboard()),
+                p_temp.PAGE_BODY_FOOTER.decode('utf8') % (
+                    datetime.now().strftime('%Y-%m-%d o %H:%M'))))
+
+    def get_match_table(self, match):
+        rows = ''
+        for team in match.teams:
+            rows += p_temp.MATCH_TEAM_ROW % (
+                ' '.join([
+                    'winner' if team.name == match.winner else '',
+                    'loser' if team.name == match.loser else ''
+                ]).strip(),
+                match.link,
+                team.name,
+                ' / '.join([
+                    self.data.get_shortname(name) for name in
+                    team.name.split('<br />')]),
+                match.link,
+                team.score)
+        html = p_temp.MATCH_TABLE.decode('utf8') % (
+            int(self.page['width'] * 0.75),
+            int(self.page['width'] * 0.25),
+            rows)
+        if match.running > 0:
+            html += p_temp.MATCH_RUNNING % (match.link, match.running)
+        return html
+
+    def get_phase_header(self, phase, position):
+        if phase.running:
+            grid_header = p_temp.MATCH_GRID_PHASE_RUNNING
+        else:
+            grid_header = p_temp.MATCH_GRID_PHASE
+        return grid_header % (
+            phase.link,
+            self.page['width'],
+            position,
+            phase.title)
+
+    def get_match_box(self, match, position):
+        if match is not None:
+            return p_temp.MATCH_BOX % (
+                position[0], position[1],
+                match.id,
+                ' '.join([
+                    str(m) for m in match.winner_matches
+                ]) if match.winner_matches is not None else '',
+                ' '.join([
+                    str(m) for m in match.loser_matches
+                ]) if match.loser_matches is not None else '',
+                self.get_match_table(match))
+        return ''
+
+    def get_match_grid(self, dimensions, grid, matches):
+        canvas_size = (
+            dimensions[0] * (
+                self.page['width'] + self.page['margin']
+            ) - self.page['margin'],
+            dimensions[1] * (
+                self.page['height'] + self.page['margin']
+            ) - self.page['margin'])
+        grid_boxes = ''
+        col_no = 0
+        for phase in grid:
+            grid_x = col_no * (self.page['width'] + self.page['margin'])
+            grid_boxes += self.get_phase_header(phase, grid_x)
+            match_height = canvas_size[1] / len(phase.matches)
+            row_no = 0
+            for match in phase.matches:
+                grid_y = int(row_no * match_height +
+                             0.5 * (match_height - self.page['height']))
+                grid_boxes += self.get_match_box(
+                    matches[match] if match is not None else None,
+                    (grid_x, grid_y))
+                row_no += 1
+            col_no += 1
+        return p_temp.MATCH_GRID % (
+            canvas_size[0], canvas_size[1],
+            canvas_size[0], canvas_size[1],
+            ' '.join(['data-%s="%s"' % (
+                setting.replace('_', '-'), str(value)
+            ) for setting, value in self.canvas.iteritems()]),
+            grid_boxes
+        )
+
+    def get_leaderboard_table(self, leaderboard):
+        if len([t for t in leaderboard if t is not None]) == 0:
+            return ''
+        position = 1
+        rows = ''
+        for team in leaderboard:
+            rows += p_temp.LEADERBOARD_ROW % (
+                position, self.get_flag(team), team or '')
+            position += 1
+        html = p_temp.LEADERBOARD.decode('utf8') % (rows)
+        return html
+
+    def get_flag(self, team):
+        flag = self.data.get_team_image(team)
+        return '' if flag is None else p_temp.LEADERBOARD_ROW_FLAG % (flag)