moved buttons related function to buttons.py

6 years ago · c8fb1dcc0d
parent 2fce9af0ea
commit c8fb1dcc0d
2 changed files with 11 additions and 0 deletions
--- a/lib/buttons.py
+++ b/lib/buttons.py
@ -23,3 +23,14 @@ GITHUB_BUTTON_FOOTER = """
 <script async defer id="github-bjs" src="https://buttons.github.io/buttons.js"></script>
 """

+def add_buttons(output):
+    """
+    Add buttons to html output
+    """
+
+    return output.replace('</body>',
+                          (TWITTER_BUTTON
+                           + GITHUB_BUTTON
+                           + GITHUB_BUTTON_3
+                           + GITHUB_BUTTON_2
+                           + GITHUB_BUTTON_FOOTER) + '</body>')
--- a/lib/wttr_srv.py
+++ b/lib/wttr_srv.py