-
Notifications
You must be signed in to change notification settings - Fork 0
/
exchange.py
93 lines (66 loc) · 1.92 KB
/
exchange.py
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
93
from abc import ABC, abstractmethod
import requests as http
EXCHANGE_COMMANDS = {}
def http_get(url, headers=None):
r = http.get(url, headers=headers) if headers else http.get(url)
return r.json()
def http_post(url, data, headers=None):
r = http.post(url, data=data, headers=headers) if headers else http.post(url, data)
return r.json()
def command_name(name=None):
def wrapper(func):
function_name = func.__name__
if name:
EXCHANGE_COMMANDS[name] = function_name
else:
EXCHANGE_COMMANDS[function_name] = function_name
return func
return wrapper
class ExchangeError(Exception):
pass
class ExchangeMethodNotImplementedError(ExchangeError):
pass
class ExchangeNotFoundError(ExchangeError):
pass
class Exchange(ABC):
def __init__(self, name):
self.name = name
self._api_keys = {}
def connect_to_api(self, api_keys):
self._api_keys = api_keys
@abstractmethod
@command_name('info')
def get_general_coin_info(self, coins):
pass
@abstractmethod
@command_name('order_history')
def get_order_history(self, coins, depth):
pass
@abstractmethod
@command_name('trade_history')
def get_trade_history(self, coins, start_date, end_date):
pass
@abstractmethod
@command_name('chart')
def get_chart_info(self, coins, start_date, end_date, candlestick_period):
pass
@abstractmethod
@command_name('balance')
def get_balance(self, coin):
pass
@abstractmethod
@command_name('open_orders')
def get_open_orders(self, coins):
pass
@abstractmethod
@command_name()
def buy(self, coins, quantity, price):
pass
@abstractmethod
@command_name()
def sell(self, coins, quantity, price):
pass
@abstractmethod
@command_name()
def cancel_order(self, order_number):
pass