Ability to compile multiple panels into single static page

2 files changed, 30 insertions, 9 deletions

diff --git a/static.py b/static.py
index 3168843..16d82ae 100644
--- a/static.py
+++ b/static.py
@@ -1,13 +1,34 @@
-import os, sys
+import copy, os, sys
 from bs4 import BeautifulSoup as bs4
 
-content_file = sys.argv[1]
-page_header = sys.argv[2] if len(sys.argv) > 2 else ''
+content_files = []
+page_header = ''
+
+arguments = sys.argv[1:]
+
+while True:
+    page_header = arguments[0]
+    arguments = arguments[1:]
+    if os.path.exists(page_header):
+        content_files.append(page_header)
+        page_header = ''
+    if len(arguments) == 0:
+        break
 
 template = bs4(file('templates/static.html'), 'lxml')
+
+content_wrapper = template.find('div', {'id': 'wrapper'}).extract()
+del content_wrapper['id']
+
 template.select('h2 small')[0].string = page_header
-template.find('div', {'id': 'content'}).append(
-    bs4(file(content_file).read(), 'lxml')
-)
+
+footer = template.find('div', {'id': 'footer'})
+
+for content_file in content_files:
+    content = copy.copy(content_wrapper)
+    content.div.append(
+        bs4(file(content_file).read(), 'html.parser')
+    )
+    footer.insert_before(content)
 
 print template.prettify().encode('utf-8')
diff --git a/templates/static.html b/templates/static.html
index 736c354..b115e0f 100644
--- a/templates/static.html
+++ b/templates/static.html
@@ -19,11 +19,11 @@
           <small></small>
         </h2>
       </div>
-      <div class="panel panel-default">
-        <div class="panel-body" id="content">
+      <div class="panel panel-default" id="wrapper">
+        <div class="panel-body">
         </div>
       </div>
-      <div class="well well-sm">
+      <div class="well well-sm" id="footer">
         &copy; 2017 - dla <a href="http://pzbs.pl">PZBS</a> zrobił <a href="https://emkael.info">mkl.</a>
         Też sobie <a href="https://github.com/emkael/pzbs-ranking">zrób.</a>
       </div>