summaryrefslogtreecommitdiff
path: root/cfetch/plugins/bitcoinaverage.py
blob: 2e045af29fd1e24905a5ad2d9651c69ad8584bab (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
##
##  coinfetch-api-bitcoinaverage - BitcoinAverage API plugin for coinfetch
##  Copyright (C) 2015-2016 Delwink, LLC
##
##  This program is free software: you can redistribute it and/or modify
##  it under the terms of the GNU Affero General Public License as published by
##  the Free Software Foundation, version 3 only.
##
##  This program is distributed in the hope that it will be useful,
##  but WITHOUT ANY WARRANTY; without even the implied warranty of
##  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
##  GNU Affero General Public License for more details.
##
##  You should have received a copy of the GNU Affero General Public License
##  along with this program.  If not, see <http://www.gnu.org/licenses/>.
##

from cfetch import register_ticker, NoSuchKindException, NoSuchPairException
from cfetch import Ticker
from requests import get

class BitcoinAverageTicker(Ticker):
    def __init__(self, path):
        super().__init__(path)

    def get_pair_data(self, response):
        return response.json()

    def get_rate(self, a, b, amt=1, kind='24h_avg'):
        a = a.upper()
        b = b.upper()
        if 'BTC' not in (a, b) or 'USD' not in (a, b):
            self._fail(a, b)

        r = get(self.path)
        res = self.get_pair_data(r)

        try:
            if a == 'BTC':
                if b not in res:
                    self._fail(a, b)

                res = res[b]
                return float(res[kind]) * amt

            if a not in res:
                self._fail(a, b)

            res = res[a]
            return (float(res[kind]) ** -1) * amt
        except KeyError as e:
            raise NoSuchKindException(str(e))

    def _fail(self, a, b):
        raise NoSuchPairException('{}/{}'.format(a, b))

register_ticker('ba', 'The BitcoinAverage ticker (built-in)',
                BitcoinAverageTicker('https://api.bitcoinaverage.com/ticker/all'))