1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
import copy, json, os, sys
from bs4 import BeautifulSoup as bs4
from pyranking.fetch import fetch_ranking
output_directory = sys.argv[1]
dates = {}
for date_config in json.load(file('config/dates.json')):
dates[date_config['date']] = date_config['url']
players = {}
for date in sorted(dates.keys()):
for player, ranking in fetch_ranking(date, True).iteritems():
if player not in players:
players[player] = {'rankings':{}}
players[player]['name'] = ranking['player']
players[player]['club'] = ranking['club']
players[player]['rankings'][date] = {'change': 'N','change-class':'primary'}
for field in ['place', 'score']:
players[player]['rankings'][date][field] = ranking[field]
for field in ['gender', 'age', 'region']:
players[player]['rankings'][date][field] = ranking[field]
players[player]['rankings'][date][field + '-place'] = ranking[field + '-place']
players[player]['rankings'][date][field + '-change'] = 'N'
players[player]['rankings'][date][field + '-change-class'] = 'primary'
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()
for date in dates:
if date not in player['rankings']:
player['rankings'][date] = 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:
ranking['change'] = prev['place'] - ranking['place']
for field in ['gender', 'age', 'region']:
if prev[field] == ranking[field]:
ranking[field + '-change'] = prev[field + '-place'] - ranking[field + '-place']
for field in ['', 'gender-', 'age-', 'region-']:
if ranking[field+'change'] == 0:
ranking[field+'change'] = '='
ranking[field+'change-class'] = 'default'
elif ranking[field+'change'] == 'N':
ranking[field+'change-class'] = 'primary'
else:
ranking[field+'change-class'] = 'success' if ranking[field+'change'] > 0 else 'danger'
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])
rank_link['href'] = '../%s' % (dates[date])
if ranking is not None:
row.select('.score')[0].string = str(ranking['score'])
for field in ['region', 'age', 'gender']:
link = row.select('td.'+field+' a')[0]
link['href'] = rank_link['href'] + '#%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)
file(os.path.join(output_directory, '%d.html' % pid), 'w').write(template.prettify().encode('utf-8'))
|
