Implemented calculation of arbitrage opportunities

bot/core.py

@@ -1,5 +1,5 @@
 #!/usr/bin/python
-from . import market_data_crawler
+from . import market_data_crawler, market_data_analyzer
 
 from tornado import gen
 from tornado.ioloop import IOLoop
@@ -13,7 +13,7 @@ class MainHandler(tornado.web.RequestHandler):
     def post(self):
         print("POST received from IP {0}".format(self.request.remote_ip))
 
-        response = {'error': False, 'msg': ""}
+        response = {'error': False, 'msg': "None"}
         request = json.loads(self.request.body.decode('utf-8'))
 
         if "token" not in request or request["token"] != "abc":
@@ -26,6 +26,11 @@ class MainHandler(tornado.web.RequestHandler):
 
             if request["command"] == "init_market_data":
                 yield self.update_market_data(request, response)
+            elif request["command"] == "get_market_data":
+                yield self.get_market_data(request, response)
+            elif request["command"] == "calc_arbitr_opport":
+                yield self.update_market_data(request, response)
+                yield self.get_arbitrage_opportunities(request, response)
 
         self.write(json.dumps(response))
 
@@ -35,6 +40,16 @@ class MainHandler(tornado.web.RequestHandler):
         response["msg"] = "Market Data initialized"
         response["data"] = market_data_crawler.market_data
 
+    @gen.coroutine
+    def get_market_data(self, request, response):
+        response["msg"] = "Market Data Retrieved"
+        response["data"] = market_data_crawler.market_data
+
+    @gen.coroutine
+    def get_arbitrage_opportunities(self, request, response):
+        response["msg"] = "Arbitrage Oportunities Retrieved"
+        response["data"] = market_data_analyzer.calculate_arbitrage_opportunities(request["basecoin"])
+
     @gen.coroutine
     def delete(self):
         print("Stopping server...")