summaryrefslogtreecommitdiff
path: root/scripts
diff options
context:
space:
mode:
Diffstat (limited to 'scripts')
-rw-r--r--scripts/players-compile.py52
1 files changed, 18 insertions, 34 deletions
diff --git a/scripts/players-compile.py b/scripts/players-compile.py
index f4c012a..66750d1 100644
--- a/scripts/players-compile.py
+++ b/scripts/players-compile.py
@@ -1,4 +1,5 @@
-import copy, json, os, sys
+import copy, os, sys
+import simplejson as json
from bs4 import BeautifulSoup as bs4
from math import ceil
from pyranking.fetch import fetch_ranking
@@ -6,8 +7,13 @@ from pyranking.fetch import fetch_ranking
output_directory = sys.argv[1]
pagesize = 100.0
+template = bs4(file('templates/player.html'), 'lxml')
menu_file = sys.argv[2]
menu_content = bs4(file(menu_file), 'html.parser')
+menu = template.select('div.static-menu')[0]
+menu.append(copy.copy(menu_content))
+
+file(os.path.join(output_directory, 'index.html'), 'w').write(template.prettify().encode('utf-8'))
dates = {}
for date_config in json.load(file('config/dates.json')):
@@ -33,20 +39,14 @@ for date in sorted(dates.keys()):
pcount = 0
for pid, player in players.iteritems():
- template = bs4(file('templates/player.html'), 'lxml')
- template.select('h2.name')[0].insert(0, player['name'])
- template.select('h3.club')[0].string = player['club']
- template.select('a.pid-link')[0]['href'] = 'https://msc.com.pl/cezar/?p=21&pid=%d' % (pid)
-
- missing_row = template.select('tr.missing')[0].extract()
- normal_row = template.select('tr.normal')[0].extract()
-
+ player['url'] = 'https://msc.com.pl/cezar/?p=21&pid=%d' % (pid)
for date in dates:
if date not in player['rankings']:
- player['rankings'][date] = None
+ player['rankings'][date] = {'place': None}
+
prev = None
for date, ranking in sorted(player['rankings'].iteritems(), lambda x,y: cmp(x[0], y[0])):
- if prev is not None and ranking is not None:
+ if prev is not None and prev['place'] is not None and ranking['place'] is not None:
ranking['change'] = prev['place'] - ranking['place']
for field in ['gender', 'age', 'region']:
if prev[field] == ranking[field]:
@@ -62,42 +62,26 @@ for pid, player in players.iteritems():
ranking[field+'change'] = '%+d' % (ranking[field+'change'])
prev = ranking
- row = copy.copy(missing_row) if ranking is None else copy.copy(normal_row)
- rank_link = row.find('td').a
- rank_link.string = '.'.join(date.split('-')[::-1])
base_rank_link = '../%s' % (dates[date])
- if ranking is not None and ranking['place'] > int(pagesize):
- rank_link['href'] = '../%s#page:%d' % (
+ if ranking['place'] is not None and ranking['place'] > int(pagesize):
+ ranking['href'] = '../%s#page:%d' % (
dates[date], ceil(ranking['place'] / pagesize)
)
else:
- rank_link['href'] = base_rank_link
- if ranking is not None:
- score_cell = row.select('.score span')[0]
- score_cell.string = '%.2f' % (ranking['score'])
- score_cell['title'] = str(ranking['score'])
+ ranking['href'] = base_rank_link
+ if ranking['place'] is not None:
for field in ['region', 'age', 'gender']:
- link = row.select('td.'+field+' a')[0]
if ranking[field+'-place'] > int(pagesize):
- link['href'] = base_rank_link + '#%s:%s;page:%d' % (
+ ranking[field+'-href'] = base_rank_link + '#%s:%s;page:%d' % (
field, ranking[field], ceil(ranking[field+'-place'] / pagesize)
)
else:
- link['href'] = base_rank_link + '#%s:%s' % (
+ ranking[field+'-href'] = base_rank_link + '#%s:%s' % (
field, ranking[field]
)
- link.string = ranking[field] if len(ranking[field]) else '-'
- for field in ['', 'region-', 'age-', 'gender-']:
- row.select('td.'+field+'place')[0].string = '%d.' % (ranking[field+'place'])
- change_label = row.select('td.'+field+'place-change span.label')[0]
- change_label.string = ranking[field+'change']
- change_label['class'] = change_label['class'] + ['label-'+ranking[field+'change-class']]
- template.select('table.table tbody')[0].insert(0, row)
- menu = template.select('div.static-menu')[0]
- menu.append(copy.copy(menu_content))
+ json.dump(player, file(os.path.join(output_directory, '%d.json' % pid), 'w'))
- file(os.path.join(output_directory, '%d.html' % pid), 'w').write(template.prettify().encode('utf-8'))
pcount += 1
sys.stdout.write("[%d/%d]\r" % (pcount, len(players)))
sys.stdout.flush()