X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=market.py;h=08838a7f971537975037790a63ef665cd0a82c05;hb=7eb9cb36c2ef01097fe8e99d40dd17308abdfe37;hp=63eff5a87ea089c6043964a72e7af1257422ebe9;hpb=5ab23e1c86de4caf0a34b9b91e5b9eddb0efa222;p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git

diff --git a/market.py b/market.py
index 63eff5a..08838a7 100644
--- a/market.py
+++ b/market.py
@@ -1,57 +1,24 @@
-import ccxt
-import decimal
+import ccxt_wrapper as ccxt
 
-def exchange_sum(self, *args):
-    return sum([arg for arg in args if isinstance(arg, (float, int, decimal.Decimal))])
-ccxt.Exchange.sum = exchange_sum
-def poloniex_fetch_balance(self, params={}):
-    self.load_markets()
-    balances = self.privatePostReturnCompleteBalances(self.extend({
-        'account': 'all',
-        }, params))
-    result = {'info': balances}
-    currencies = list(balances.keys())
-    for c in range(0, len(currencies)):
-        id = currencies[c]
-        balance = balances[id]
-        currency = self.common_currency_code(id)
-        account = {
-                'free': decimal.Decimal(balance['available']),
-                'used': decimal.Decimal(balance['onOrders']),
-                'total': decimal.Decimal(0.0),
-                }
-        account['total'] = self.sum(account['free'], account['used'])
-        result[currency] = account
-    return self.parse_balance(result)
-ccxt.poloniex.fetch_balance = poloniex_fetch_balance
-
-def poloniex_parse_ticker(self, ticker, market=None):
-    timestamp = self.milliseconds()
-    symbol = None
-    if market:
-        symbol = market['symbol']
-    return {
-        'symbol': symbol,
-        'timestamp': timestamp,
-        'datetime': self.iso8601(timestamp),
-        'high': decimal.Decimal(ticker['high24hr']),
-        'low': decimal.Decimal(ticker['low24hr']),
-        'bid': decimal.Decimal(ticker['highestBid']),
-        'ask': decimal.Decimal(ticker['lowestAsk']),
-        'vwap': None,
-        'open': None,
-        'close': None,
-        'first': None,
-        'last': decimal.Decimal(ticker['last']),
-        'change': decimal.Decimal(ticker['percentChange']),
-        'percentage': None,
-        'average': None,
-        'baseVolume': decimal.Decimal(ticker['quoteVolume']),
-        'quoteVolume': decimal.Decimal(ticker['baseVolume']),
-        'info': ticker,
-    }
-ccxt.poloniex.parse_ticker = poloniex_parse_ticker
-market = ccxt.poloniex({
+market = ccxt.poloniexE({
     "apiKey": "XXXXXXXX-XXXXXXXX-XXXXXXXX-XXXXXXXX",
     "secret": "1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef",
     })
+
+
+# For requests logging
+requests = []
+market.session.origin_request = market.session.request
+
+def request_wrap(self, *args, **kwargs):
+    r = self.origin_request(*args, **kwargs)
+    requests.append({
+        "method": args[0],
+        "url": args[1],
+        "body": kwargs["data"],
+        "headers": kwargs["headers"],
+        "status": r.status_code,
+        "response": r.text,
+        })
+    return r
+market.session.request = request_wrap.__get__(market.session, market.session.__class__)