[RestExchange] filter symbols depending on exchange type

tmp


tmp

Author: techfreaque

octobot_trading/exchanges/types/rest_exchange.py

@@ -22,7 +22,9 @@
 import sys
 
 import ccxt.async_support as ccxt
+import octobot_commons as octobot_commons
 import octobot_commons.enums as commons_enums
+import octobot_commons.symbols.symbol_util as symbol_util
 import octobot_commons.tree as commons_tree
 
 from octobot_commons import number_util
@@ -93,7 +95,7 @@ def _create_connector(self, config, exchange_manager, connector_class):
 
     async def initialize_impl(self):
         await self.connector.initialize()
-        self.symbols = self.connector.symbols
+        self.symbols = self.get_exchange_pairs()
         self.time_frames = self.connector.time_frames
 
     async def stop(self) -> None:
@@ -558,6 +560,17 @@ def get_split_pair_from_exchange(self, pair) -> (str, str):
     def get_exchange_pair(self, pair) -> str:
         return self.connector.get_exchange_pair(pair)
 
+    def get_exchange_pairs(self) -> set:
+        # make sure spot doesnt use futures pairs and vice versa
+        if self.exchange_manager.is_spot_only or self.exchange_manager.is_margin: 
+            return {symbol for symbol in self.connector.symbols 
+                    if symbol_util.parse_symbol(symbol).is_spot()}
+        elif self.exchange_manager.is_future:
+            return {symbol for symbol in self.connector.symbols 
+                    if symbol_util.parse_symbol(symbol).is_future()}
+        else:
+            return self.connector.symbols 
+        
     def get_pair_cryptocurrency(self, pair) -> str:
         return self.connector.get_pair_cryptocurrency(pair)