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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
import copy, json, sys
import mysql.connector
from bs4 import BeautifulSoup as bs4
settings = json.load(file('config.json'))
connection = mysql.connector.connect(
user=settings['user'],
password=settings['pass'],
host=settings['host'],
port=settings['port'],
database=settings['db']
)
cursor = connection.cursor(dictionary=True, buffered=True)
def fetch_ranking(date, assoc=False):
sql = '''SELECT
rankings.place,
rankings.pid,
CONCAT(players.name, " ", players.surname) player,
players.rank,
players.region, players.club,
REPLACE(players.flags, "K", "") age,
IF(players.flags LIKE 'K%', "K", "") gender,
rankings.score
FROM rankings
JOIN players
ON players.id = rankings.pid
WHERE rankings.date = %(date)s
ORDER BY rankings.place
'''
cursor.execute(sql, {'date': date})
ranks = {
'gender': {},
'age': {},
'region': {}
}
ranking = cursor.fetchall()
for row in ranking:
if row['gender'] == '':
row['gender'] = 'M'
for category in ['gender', 'age', 'region']:
if row[category] not in ranks[category]:
ranks[category][row[category]] = 0
ranks[category][row[category]] += 1
row[category + '-place'] = ranks[category][row[category]]
for category in ['place', 'gender', 'age', 'region']:
row[category + '-change'] = 'N'
row[category + '-change-class'] = 'info'
if assoc:
result = {}
for row in ranking:
result[row['pid']] = row
return result
return ranking
ranking = fetch_ranking(sys.argv[1])
old_ranking = fetch_ranking(sys.argv[2], True) if len(sys.argv) > 2 else {}
for row in ranking:
if row['pid'] in old_ranking:
row['place-change'] = old_ranking[row['pid']]['place'] - row['place']
row['place-change-class'] = 'success' if row['place-change'] > 0 else 'danger'
row['place-change'] = '%+d' % (row['place-change'])
if row['place-change'] == '+0':
row['place-change'] = '='
row['place-change-class'] = 'default'
for category in ['gender', 'age', 'region']:
if row[category] == old_ranking[row['pid']][category]:
row[category + '-change'] = old_ranking[row['pid']][category + '-place'] - row[category + '-place']
row[category + '-change-class'] = 'success' if row[category + '-change'] > 0 else 'danger'
row[category + '-change'] = '%+d' % (row[category + '-change'])
if row[category + '-change'] == '+0':
row[category + '-change'] = '='
row[category + '-change-class'] = 'default'
table = bs4(file('templates/ranking.html'), 'lxml')
table_body = table.select('tbody')[0]
table_row = table_body.select('tr')[0].extract()
table.select('.page-header h1 small')[0].string = 'z dnia %s' % (
'.'.join(sys.argv[1].split('-')[::-1])
)
for row in ranking:
new_row = copy.copy(table_row)
new_row.select('td.pid')[0].string = str(row['pid'])
new_row.select('td.pidlink a')[0]['href'] = 'https://msc.com.pl/cezar/?p=21&pid=%d' % (row['pid'])
new_row.select('td.name')[0].string = row['player']
new_row.select('td.club')[0].string = row['club']
for category in ['gender', 'age', 'region']:
new_row.select('td.' + category)[0].string = row[category]
new_row.select('td.' + category + '-place .rank')[0].string = str(row[category + '-place'])
badge = new_row.select('td.' + category + '-place .change')[0]
badge.string = row[category + '-change']
badge['class'] = badge['class'] + ['label-' + row[category + '-change-class']]
new_row.select('td.ranking')[0].string = str(row['score'])
new_row.select('td.place .rank')[0].string = str(row['place'])
badge = new_row.select('td.place .change')[0]
badge.string = row['place-change']
badge['class'] = badge['class'] + ['label-' + row['place-change-class']]
for category in ['gender', 'age', 'region']:
if row[category + '-place'] == 1:
new_row['class'] = new_row.get('class', []) + ['info']
table_body.append(new_row)
print table.prettify().encode('utf-8')
|