From 54729ff05267ffddaff919dfbdc3b13d8767cd5e Mon Sep 17 00:00:00 2001
From: emkael <emkael@tlen.pl>
Date: Sat, 21 Oct 2017 18:05:13 +0200
Subject: Ability to set custom link label for swiss tournament links

---
 jfr_playoff/generator.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

(limited to 'jfr_playoff/generator.py')

diff --git a/jfr_playoff/generator.py b/jfr_playoff/generator.py
index 479f33b..bb18bb9 100644
--- a/jfr_playoff/generator.py
+++ b/jfr_playoff/generator.py
@@ -124,10 +124,15 @@ class PlayoffGenerator(object):
         return html
 
     def get_swiss_links(self):
-        return '\n'.join([
-            p_temp.SWISS_LINK % (
-                event['link'], event['position']
-            ) for event in self.data.get_swiss_info()])
+        info = []
+        for event in self.data.get_swiss_info():
+            event_label = p_temp.SWISS_DEFAULT_LABEL % (event['position'])
+            if 'label' in event and event['label'] is not None:
+                event_label = event['label']
+            info.append(p_temp.SWISS_LINK % (
+                event['link'], event_label
+            ))
+        return '\n'.join(info)
 
     def get_flag(self, team):
         flag = self.data.get_team_image(team)
-- 
cgit v1.2.3