summaryrefslogtreecommitdiff
path: root/cfetch/plugins/bitstamp.py
blob: c80899562c1204b1e8acb5b7c8c8ed903e5b1b12 (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
##
##  coinfetch-api-bitstamp - Bitstamp 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 BitstampTicker(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='vwap'):
        r = get(self.path)
        res = self.get_pair_data(r)
        if kind not in res:
            raise NoSuchKindException('Kind {} not available'.format(kind))

        if a == 'btc' and b == 'usd':
            return float(res[kind]) * amt
        elif a == 'usd' and b == 'btc':
            return (float(res[kind]) ** -1) * amt
        else:
            raise NoSuchPairException('{}/{}'.format(a, b))

register_ticker('bs', 'The Bitstamp ticker (built-in)',
                BitstampTicker('https://www.bitstamp.net/api/ticker/'))