[Server] Add route

Author: Herklos

dev_requirements.txt

@@ -2,4 +2,3 @@ pytest
 
 redis
 Flask
-Flask-Caching

main.py

@@ -15,13 +15,13 @@
 
     # start arbitrage detection
     print("Scanning...")
-    best_opportunities, best_profit = asyncio.run(triangular_arbitrage.detector.run_detection())
+    best_opportunities, best_profit, exchange_name = asyncio.run(triangular_arbitrage.detector.run_detection())
     def opportunity_symbol(opportunity):
         return symbols.parse_symbol(str(opportunity.symbol))
 
     # Display arbitrage detection result
     print("-------------------------------------------")
-    print(f"Start by selling {str(opportunity_symbol(best_opportunities[0]).base)} to {str(opportunity_symbol(best_opportunities[0]).quote)} then sell {str(opportunity_symbol(best_opportunities[1]).base)} to {str(opportunity_symbol(best_opportunities[1]).quote)} and finally sell {str(opportunity_symbol(best_opportunities[2]).base)} to {str(opportunity_symbol(best_opportunities[2]).quote)} to make a profit of {(best_profit - 1) * 100}%")
+    print(f"[{exchange_name}] Start by selling {str(opportunity_symbol(best_opportunities[0]).base)} to {str(opportunity_symbol(best_opportunities[0]).quote)} then sell {str(opportunity_symbol(best_opportunities[1]).base)} to {str(opportunity_symbol(best_opportunities[1]).quote)} and finally sell {str(opportunity_symbol(best_opportunities[2]).base)} to {str(opportunity_symbol(best_opportunities[2]).quote)} to make a profit of {(best_profit - 1) * 100}%")
     print("-------------------------------------------")
 
     if benchmark:

server.py

@@ -2,23 +2,21 @@
 import os
 from dotenv import load_dotenv
 from flask import Flask, jsonify
-from flask_caching import Cache
 
 import triangular_arbitrage.detector
 
-cache = Cache(config={'CACHE_TYPE': 'SimpleCache'})
 app = Flask(__name__)
-cache.init_app(app)
 
-@app.route("/")
-@cache.cached(timeout=os.getenv('CACHE', 60*60)) # cache it 1h by default
-def get_data():
+@app.route("/", defaults={'exchange': 'binance'})
+@app.route("/<exchange>")
+def get_data(exchange):
     # start arbitrage detection
     print("Scanning...")
-    best_opportunities, best_profit = asyncio.run(triangular_arbitrage.detector.run_detection())
+    best_opportunities, best_profit, exchange_name = asyncio.run(triangular_arbitrage.detector.run_detection(exchange))
     return jsonify({
         'best_opportunity': [str(best_opportunity.symbol) for best_opportunity in best_opportunities],
-        'best_profit': best_profit
+        'best_profit': best_profit,
+        'exchange_name': exchange_name
     })
 
 

triangular_arbitrage/detector.py

@@ -92,20 +92,21 @@ def get_opportunity_symbol(a, b):
 
     return best_triplet, best_profit
 
-async def run_detection():
-    exchange = ccxt.binance()
+async def run_detection(exchange_name = "binance"):
+    exchange_class = getattr(ccxt, exchange_name)
+    exchange = exchange_class()
     try:
         tickers = await fetch_tickers(exchange)
         exchange_time = exchange.milliseconds()
         last_prices = get_last_prices(exchange_time, tickers)
         best_opportunity, best_profit = get_best_opportunity(last_prices)
         if os.getenv(REDIS_HOST_ENV, None) is not None:
-            upload_result(best_opportunity, best_profit)
+            upload_result(best_opportunity, best_profit, exchange.id)
     finally:
         await exchange.close()
-        return best_opportunity, best_profit
+        return best_opportunity, best_profit, exchange.name
 
-def upload_result(best_opportunities, best_profit):
+def upload_result(best_opportunities, best_profit, exchange_id):
     import redis
     redis_client = redis.Redis(
         host=os.getenv(REDIS_HOST_ENV, None),
@@ -117,6 +118,7 @@ def upload_result(best_opportunities, best_profit):
     data = {
         'best_opportunity': [str(best_opportunity.symbol) for best_opportunity in best_opportunities],
         'best_profit': best_profit,
+        'exchange_id': exchange_id,
         'timestamp': time.time()
     }
-    redis_client.json().set(f"{os.getenv(REDIS_KEY_ENV, None)}", '$', data)
+    redis_client.json().set(f"{os.getenv(REDIS_KEY_ENV, None)}:{exchange_id}", '$', data)