PyShift/accounts.py at master · nytrobound/PyShift · GitHub

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
import requests

import constants
from helpers import requests_error, response_to_dict


@requests_error
def accounts_open(secret):
    payload = {'secret': secret}
    response = requests.post(constants.ACCOUNTS_OPEN,
                             data=payload,
                             timeout=constants.TIMEOUT
                             )
    return response_to_dict(response)


@requests_error
def accounts_get_balance(address):
    payload = {'address': address}
    response = requests.get(constants.ACCOUNTS_GET_BALANCE,
                            params=payload,
                            timeout=constants.TIMEOUT
                            )
    return response_to_dict(response)


@requests_error
def accounts_get_publickey(address):
    payload = {'address': address}
    response = requests.get(constants.ACCOUNTS_GET_PUBLICKEY,
                            params=payload,
                            timeout=constants.TIMEOUT
                            )
    return response_to_dict(response)


@requests_error
def accounts_gen_publickey(secret):
    payload = {'secret': secret}
    response = requests.post(constants.ACCOUNTS_GEN_PUBLICKEY,
                             data=payload,
                             timeout=constants.TIMEOUT
                             )
    return response_to_dict(response)


@requests_error
def accounts_get(address):
    payload = {'address': address}
    response = requests.get(constants.ACCOUNTS_GET,
                            params=payload,
                            timeout=constants.TIMEOUT
                            )
    return response_to_dict(response)


@requests_error
def accounts_get_delegates(address):
    payload = {'address': address}
    response = requests.get(constants.ACCOUNTS_GET_DELEGATES,
                            params=payload,
                            timeout=constants.TIMEOUT
                            )
    return response_to_dict(response)


@requests_error
def accounts_put_delegates(secret, delegates, **kwargs):
    PARAMS = [
        'secret',
        'publicKey',
        'secondSecret',
        'delegates'
    ]
    if len(delegates) > 33:
        raise ValueError("You are only allowed to vote for 33 delegates at once. You have voted for:", len(delegates))

    payload = {
        'secret': secret,
        'delegates': delegates
    }

    for name in kwargs:
        if name not in PARAMS:
            raise ValueError("%s is not a valid parameter." % name)
        payload[name] = kwargs[name]

    response = requests.put(constants.ACCOUNTS_PUT_DELEGATES,
                            data=payload,
                            timeout=constants.TIMEOUT
                            )
    return response_to_dict(response)