Pre-generating menu content for ranking pages

1 files changed, 6 insertions, 1 deletions

diff --git a/scripts/ranking.py b/scripts/ranking.py
index e3276da..52b036f 100644
--- a/scripts/ranking.py
+++ b/scripts/ranking.py
@@ -1,7 +1,7 @@
 import os, sys
 from bs4 import BeautifulSoup as bs4
 
-ranking_date = sys.argv[3]
+ranking_date = sys.argv[4]
 subtitle = 'notowanie %s (%s), stan na %s' % (
     sys.argv[1], sys.argv[2], '.'.join(ranking_date.split('-')[::-1])
 )
@@ -18,4 +18,9 @@ script_src['src'] = '%s?%d' % ('_res/ranking.js', os.path.getmtime('http/_res/ra
 rawlink = table.select('a#rawlink')[0]
 rawlink['href'] = '%s/%s.csv' % (rawlink['href'], ranking_date)
 
+menu_file = file(sys.argv[3])
+menu = table.select('div.static-menu')[0]
+menu.clear()
+menu.append(bs4(menu_file, 'html.parser'))
+
 print table.prettify().encode('utf-8')