Add main running file and fetch information from database

[perso/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Trader.git] / market.py

diff --git a/market.py b/market.py
index a34636eab42aeefcb8e62311e33840a2ae4dee4b..224cc32bae3b19b51f20cc2ee2158062664f86ad 100644 (file)
--- a/market.py
+++ b/market.py
@@ -1,6 +1,17 @@
-import ccxt
+import ccxt_wrapper as ccxt
+from store import ReportStore
 
-market = ccxt.poloniex({
-    "apiKey": "XXXXXXXX-XXXXXXXX-XXXXXXXX-XXXXXXXX",
-    "secret": "1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef",
-    })
+def get_market(config):
+    market = ccxt.poloniexE(config)
+
+    # For requests logging
+    market.session.origin_request = market.session.request
+
+    def request_wrap(self, *args, **kwargs):
+        r = self.origin_request(*args, **kwargs)
+        ReportStore.log_http_request(args[0], args[1], kwargs["data"],
+                kwargs["headers"], r)
+        return r
+    market.session.request = request_wrap.__get__(market.session, market.session.__class__)
+
+    return market