create wrapper for poloniex class

1 files changed, 18 insertions, 1 deletions

diff --git a/market.py b/market.py
index deaab88..08838a7 100644
--- a/market.py
+++ b/market.py
@@ -1,7 +1,24 @@
 import ccxt_wrapper as ccxt
 
-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__)