diff options
Diffstat (limited to 'market.py')
-rw-r--r-- | market.py | 31 |
1 files changed, 12 insertions, 19 deletions
@@ -1,24 +1,17 @@ | |||
1 | import ccxt_wrapper as ccxt | 1 | import ccxt_wrapper as ccxt |
2 | from store import ReportStore | ||
2 | 3 | ||
3 | market = ccxt.poloniexE({ | 4 | def get_market(config): |
4 | "apiKey": "XXXXXXXX-XXXXXXXX-XXXXXXXX-XXXXXXXX", | 5 | market = ccxt.poloniexE(config) |
5 | "secret": "1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef", | ||
6 | }) | ||
7 | 6 | ||
7 | # For requests logging | ||
8 | market.session.origin_request = market.session.request | ||
8 | 9 | ||
9 | # For requests logging | 10 | def request_wrap(self, *args, **kwargs): |
10 | requests = [] | 11 | r = self.origin_request(*args, **kwargs) |
11 | market.session.origin_request = market.session.request | 12 | ReportStore.log_http_request(args[0], args[1], kwargs["data"], |
13 | kwargs["headers"], r) | ||
14 | return r | ||
15 | market.session.request = request_wrap.__get__(market.session, market.session.__class__) | ||
12 | 16 | ||
13 | def request_wrap(self, *args, **kwargs): | 17 | return market |
14 | r = self.origin_request(*args, **kwargs) | ||
15 | requests.append({ | ||
16 | "method": args[0], | ||
17 | "url": args[1], | ||
18 | "body": kwargs["data"], | ||
19 | "headers": kwargs["headers"], | ||
20 | "status": r.status_code, | ||
21 | "response": r.text, | ||
22 | }) | ||
23 | return r | ||
24 | market.session.request = request_wrap.__get__(market.session, market.session.__class__) | ||