From c8fb1dcc0d493eff0909dc9372a2f33aaccb5d91 Mon Sep 17 00:00:00 2001
From: Igor Chubin <igor@chub.in>
Date: Sun, 7 Oct 2018 12:10:07 +0200
Subject: [PATCH] moved buttons related function to buttons.py

---
 lib/buttons.py              | 11 +++++++++++
 lib/{srv.py => wttr_srv.py} |  0
 2 files changed, 11 insertions(+)
 rename lib/{srv.py => wttr_srv.py} (100%)

diff --git a/lib/buttons.py b/lib/buttons.py
index 96180a8..9aac99b 100644
--- 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>')
diff --git a/lib/srv.py b/lib/wttr_srv.py
similarity index 100%
rename from lib/srv.py
rename to lib/wttr_srv.py