summaryrefslogtreecommitdiff
path: root/test/apitest/pzbs.py
blob: 84488fffe26d235eee6fa232a597b80b589d4a02 (plain)
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
import urlparse

from bs4 import BeautifulSoup as bs

from .apitest import ApiTest

class PzbsCalculator(ApiTest):
    ranks = {
        'o++++': 1,
        'o+++': 2,
        'o++': 3,
        'o+': 4,
        'o-': 5,
        'r': 6,
        'o': 7,
        'k': 8
    }
    def get_url(self):
        return 'http://pzbs.pl/sedziowie/pkl/pkle2018.php'
    def get_method(self):
        return 'post'
    def get_request(self,
                    tour_type, tour_rank, tour_boards,
                    cont_count, cont_rank,
                    override=None):
        params = {
            'rgg': 25, 'rgp': 15, 'rot': 10, 'rok1': 7, 'rok': 5, 'rtp': 4, 'rto': 2, 'rtk': 1,
            'min1': 200, 'min2': 150, 'min3': 75, 'min4': 50, 'min5': 0, 'min6': 0, 'min7': 0, 'min8': 0,
            'rgg_': 40, 'rgp_': 25, 'rot_': 15, 'rok1_': 10, 'rok_': 7, 'rtp_': 5, 'rto_': 3, 'rtk_': 2,
            'min1_': 300, 'min2_': 200, 'min3_': 100, 'min4_': 70, 'min5_': 0, 'min6_': 0, 'min7_': 0, 'min8_': 0,
            'zaw': 0.05, 'prp1': 90, 'prp2': 20, 'pru1': 2, 'pru2': 20, 'pru3': 50,
            'typ': 0,
            'rozdan': 0,
            'rng': 0,
            'iuc': 0,
            'izw': 0,
            'swk': 0,
            'srd': 0
        }
        params['typ'] = self.tourtypes[tour_type]
        params['rozdan'] = tour_boards
        params['rng'] = self.ranks[tour_rank]
        params['iuc'] = cont_count
        params['izw'] = cont_count * self.tourtypes[tour_type]
        params['swk'] = cont_rank
        params['srd'] = max(0.15, cont_rank / (cont_count * self.tourtypes[tour_type]))
        if override:
            if 'points' in override:
                params['min' + str(params['rng']) + '_'*params['rozdan']] = override['points']
            if 'weight' in override:
                params['r' + ['', 'gg', 'gp', 'ot', 'ok1', 'ok', 'tp', 'to', 'tk'][params['rng']] + '_'*params['rozdan']] = override['weight']
            if 'players' in override:
                params['zaw'] = override['players']
            if 'cutoff' in override:
                for i in range(0, 3):
                    params['pru' + str(i+1)] = override['cutoff'][i][0] * 100
                    params['prp' + str(i+1)] = override['cutoff'][i][1] * 100
        return params

    def get_response(self, text):
        results = {}
        content = bs(text, 'lxml')
        for row in content.select('table')[-1].select('tr'):
            cells = row.select('td')
            if cells[0].text.isdigit():
                results[cells[0].text] = int(cells[1].text)
            elif cells[0].text == 'SUMA PKL':
                results[u'sum'] = int(cells[1].text)
        return results