summaryrefslogtreecommitdiff
path: root/test/apitest/api.py
blob: 4a5229653cf11131efc758982b74959c1a5beb1a (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
import json
import urlparse

from .apitest import ApiTest

class ApiCalculator(ApiTest):
    ranks = {
        'o++++': 7,
        'o+++': 6,
        'o++': 5,
        'o+': 4,
        'o-': 3,
        'r': 2,
        'o': 1,
        'k': 0
    }
    def get_url(self):
        return 'http://localhost/pkl/api.php'
    def get_method(self):
        return 'post'
    def get_request(self,
                    tour_type, tour_rank, tour_boards,
                    cont_count, cont_rank,
                    override=None):
        params = {}
        params['type'] = self.tourtypes[tour_type]
        params['over39_boards'] = tour_boards
        params['tournament_rank'] = self.ranks[tour_rank]
        params['contestants'] = cont_count
        params['title_sum'] = cont_rank
        if override:
            params['manual'] = {}
            if 'points' in override:
                params['manual[min_points]'] = override['points']
            if 'weight' in override:
                params['manual[tournament_weight]'] = override['weight']
            if 'players' in override:
                params['manual[players_coefficient]'] = override['players']
            if 'cutoff' in override:
                i = 0
                for c in override['cutoff']:
                    params['manual[points_cutoffs][' + str(i) + '][0]'] = c[0]
                    params['manual[points_cutoffs][' + str(i) + '][1]'] = c[1]
                    i += 1
        return params

    def get_response(self, text):
        return json.loads(text)