OctoBot-Trading 2.4.115__tar.gz → 2.4.117__tar.gz

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/CHANGELOG.md

@@ -4,6 +4,15 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [2.4.117] - 2024-10-24
+### Fixed
+[Exchanges] cache key conflict
+
+## [2.4.116] - 2024-10-23
+### Added
+[Exchanges] handle proxy config
+[Exchanges] handle access token auth
+
 ## [2.4.115] - 2024-10-21
 ### Added
 [Orders] add get_valid_split_orders

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117/OctoBot_Trading.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: OctoBot-Trading
-Version: 2.4.115
+Version: 2.4.117
 Summary: OctoBot project trading package
 Home-page: https://github.com/Drakkar-Software/OctoBot-Trading
 Author: Drakkar-Software
@@ -16,7 +16,7 @@ Classifier: Operating System :: POSIX
 Requires-Python: >=3.8
 License-File: LICENSE
 
-# OctoBot-Trading [2.4.115](https://github.com/Drakkar-Software/OctoBot-Trading/blob/master/CHANGELOG.md)
+# OctoBot-Trading [2.4.117](https://github.com/Drakkar-Software/OctoBot-Trading/blob/master/CHANGELOG.md)
 [![Codacy Badge](https://api.codacy.com/project/badge/Grade/903b6b22bceb4661b608a86fea655f69)](https://app.codacy.com/gh/Drakkar-Software/OctoBot-Trading?utm_source=github.com&utm_medium=referral&utm_content=Drakkar-Software/OctoBot-Trading&utm_campaign=Badge_Grade_Dashboard)
 [![PyPI](https://img.shields.io/pypi/v/OctoBot-Trading.svg)](https://pypi.python.org/pypi/OctoBot-Trading/)
 [![Coverage Status](https://coveralls.io/repos/github/Drakkar-Software/OctoBot-Trading/badge.svg?branch=master)](https://coveralls.io/github/Drakkar-Software/OctoBot-Trading?branch=master)

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/OctoBot_Trading.egg-info/SOURCES.txt

@@ -96,6 +96,7 @@ octobot_trading/exchanges/adapters/abstract_adapter.py
 octobot_trading/exchanges/config/__init__.py
 octobot_trading/exchanges/config/backtesting_exchange_config.py
 octobot_trading/exchanges/config/exchange_config_data.py
+octobot_trading/exchanges/config/proxy_config.py
 octobot_trading/exchanges/connectors/__init__.py
 octobot_trading/exchanges/connectors/ccxt/__init__.py
 octobot_trading/exchanges/connectors/ccxt/ccxt_adapter.py

{OctoBot-Trading-2.4.115/OctoBot_Trading.egg-info → OctoBot-Trading-2.4.117}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: OctoBot-Trading
-Version: 2.4.115
+Version: 2.4.117
 Summary: OctoBot project trading package
 Home-page: https://github.com/Drakkar-Software/OctoBot-Trading
 Author: Drakkar-Software
@@ -16,7 +16,7 @@ Classifier: Operating System :: POSIX
 Requires-Python: >=3.8
 License-File: LICENSE
 
-# OctoBot-Trading [2.4.115](https://github.com/Drakkar-Software/OctoBot-Trading/blob/master/CHANGELOG.md)
+# OctoBot-Trading [2.4.117](https://github.com/Drakkar-Software/OctoBot-Trading/blob/master/CHANGELOG.md)
 [![Codacy Badge](https://api.codacy.com/project/badge/Grade/903b6b22bceb4661b608a86fea655f69)](https://app.codacy.com/gh/Drakkar-Software/OctoBot-Trading?utm_source=github.com&utm_medium=referral&utm_content=Drakkar-Software/OctoBot-Trading&utm_campaign=Badge_Grade_Dashboard)
 [![PyPI](https://img.shields.io/pypi/v/OctoBot-Trading.svg)](https://pypi.python.org/pypi/OctoBot-Trading/)
 [![Coverage Status](https://coveralls.io/repos/github/Drakkar-Software/OctoBot-Trading/badge.svg?branch=master)](https://coveralls.io/github/Drakkar-Software/OctoBot-Trading?branch=master)

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/README.md

@@ -1,4 +1,4 @@
-# OctoBot-Trading [2.4.115](https://github.com/Drakkar-Software/OctoBot-Trading/blob/master/CHANGELOG.md)
+# OctoBot-Trading [2.4.117](https://github.com/Drakkar-Software/OctoBot-Trading/blob/master/CHANGELOG.md)
 [![Codacy Badge](https://api.codacy.com/project/badge/Grade/903b6b22bceb4661b608a86fea655f69)](https://app.codacy.com/gh/Drakkar-Software/OctoBot-Trading?utm_source=github.com&utm_medium=referral&utm_content=Drakkar-Software/OctoBot-Trading&utm_campaign=Badge_Grade_Dashboard)
 [![PyPI](https://img.shields.io/pypi/v/OctoBot-Trading.svg)](https://pypi.python.org/pypi/OctoBot-Trading/)
 [![Coverage Status](https://coveralls.io/repos/github/Drakkar-Software/OctoBot-Trading/badge.svg?branch=master)](https://coveralls.io/github/Drakkar-Software/OctoBot-Trading?branch=master)

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/__init__.py

@@ -15,4 +15,4 @@
 #  License along with this library.
 
 PROJECT_NAME = "OctoBot-Trading"
-VERSION = "2.4.115"  # major.minor.revision
+VERSION = "2.4.117"  # major.minor.revision

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/constants.py

@@ -66,7 +66,11 @@ DEFAULT_FAILED_REQUEST_RETRY_TIME = 1
 FAILED_NETWORK_REQUEST_RETRY_ATTEMPTS = 5
 DEFAULT_REQUEST_TIMEOUT = int(os.getenv("DEFAULT_REQUEST_TIMEOUT", "20000"))    # default ccxt is 10s, use 20
 ENABLE_EXCHANGE_HTTP_PROXY_FROM_ENV = os_util.parse_boolean_environment_var(
-    "ENABLE_EXCHANGE_HTTP_PROXY_FROM_ENV", "True")
+    "ENABLE_EXCHANGE_HTTP_PROXY_FROM_ENV", "True"
+)
+EXCHANGE_HTTP_PROXY_AUTHENTICATED_URL = os.getenv("EXCHANGE_HTTP_PROXY_AUTHENTICATED_URL")
+EXCHANGE_HTTPS_PROXY_AUTHENTICATED_URL = os.getenv("EXCHANGE_HTTPS_PROXY_AUTHENTICATED_URL")
+EXCHANGE_SOCKS_PROXY_AUTHENTICATED_URL = os.getenv("EXCHANGE_SOCKS_PROXY_AUTHENTICATED_URL")
 ENABLE_CCXT_VERBOSE = os_util.parse_boolean_environment_var("ENABLE_CCXT_VERBOSE", "False")
 ENABLE_CCXT_RATE_LIMIT = os_util.parse_boolean_environment_var("ENABLE_CCXT_RATE_LIMIT", "True")
 CCXT_DEFAULT_CACHE_LIMIT = int(os.getenv("CCXT_DEFAULT_CACHE_LIMIT", "1000"))  # 1000: default ccxt value

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/__init__.py

@@ -94,6 +94,7 @@ from octobot_trading.exchanges import config
 from octobot_trading.exchanges.config import (
     ExchangeConfig,
     BacktestingExchangeConfig,
+    ProxyConfig,
 )
 from octobot_trading.exchanges import traders
 from octobot_trading.exchanges.traders import (
@@ -133,6 +134,7 @@ __all__ = [
     "AbstractAdapter",
     "ExchangeConfig",
     "BacktestingExchangeConfig",
+    "ProxyConfig",
     "ExchangeManager",
     "ExchangeBuilder",
     "create_exchange_builder_instance",

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/abstract_websocket_exchange.py

@@ -219,7 +219,7 @@ class AbstractWebsocketExchange:
         raise NotImplementedError(f"get_feeds_count is not implemented")
 
     def _should_authenticate(self):
-        api_key, api_secret, _, _ = self.get_exchange_credentials()
+        api_key, api_secret, _, _, _ = self.get_exchange_credentials()
         return not self.exchange_manager.without_auth \
             and not self.exchange_manager.is_trader_simulated \
             and api_key and api_secret

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/config/__init__.py

@@ -23,8 +23,13 @@ from octobot_trading.exchanges.config import backtesting_exchange_config
 from octobot_trading.exchanges.config.backtesting_exchange_config import (
     BacktestingExchangeConfig,
 )
+from octobot_trading.exchanges.config import proxy_config
+from octobot_trading.exchanges.config.proxy_config import (
+    ProxyConfig,
+)
 
 __all__ = [
     "ExchangeConfig",
     "BacktestingExchangeConfig",
+    "ProxyConfig",
 ]

OctoBot-Trading-2.4.117/octobot_trading/exchanges/config/proxy_config.py

@@ -0,0 +1,40 @@
+#  Drakkar-Software OctoBot-Trading
+#  Copyright (c) Drakkar-Software, All rights reserved.
+#
+#  This library is free software; you can redistribute it and/or
+#  modify it under the terms of the GNU Lesser General Public
+#  License as published by the Free Software Foundation; either
+#  version 3.0 of the License, or (at your option) any later version.
+#
+#  This library is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+#  Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public
+#  License along with this library.
+import dataclasses
+import typing
+import octobot_trading.constants
+
+
+@dataclasses.dataclass
+class ProxyConfig:
+    # REST proxy
+    http_proxy: typing.Optional[str] = None
+    http_proxy_callback: typing.Optional[typing.Callable] = None
+    https_proxy: typing.Optional[str] = None
+    https_proxy_callback: typing.Optional[typing.Callable] = None
+    # Websocket proxy
+    socks_proxy : typing.Optional[str] = None
+    socks_proxy_callback: typing.Optional[typing.Callable] = None
+    # enable trust_env in exchange's aiohttp.ClientSession
+    aiohttp_trust_env: bool = octobot_trading.constants.ENABLE_EXCHANGE_HTTP_PROXY_FROM_ENV
+
+    @classmethod
+    def default_env_var_config(cls):
+        return cls(
+            http_proxy=octobot_trading.constants.EXCHANGE_HTTP_PROXY_AUTHENTICATED_URL or None,
+            https_proxy=octobot_trading.constants.EXCHANGE_HTTPS_PROXY_AUTHENTICATED_URL or None,
+            socks_proxy=octobot_trading.constants.EXCHANGE_SOCKS_PROXY_AUTHENTICATED_URL or None,
+        )

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/connectors/ccxt/ccxt_client_util.py

@@ -18,6 +18,7 @@ import logging
 import typing
 import ccxt
 import ccxt.pro as ccxt_pro
+import ccxt.async_support as async_ccxt
 
 import octobot_commons.time_frame_manager as time_frame_manager
 import octobot_trading.constants as constants
@@ -25,6 +26,7 @@ import octobot_trading.enums as enums
 import octobot_trading.errors as errors
 import octobot_trading.exchanges.connectors.ccxt.enums as ccxt_enums
 import octobot_trading.exchanges.connectors.ccxt.ccxt_clients_cache as ccxt_clients_cache
+import octobot_trading.exchanges.config.proxy_config as proxy_config_import
 import octobot_trading.exchanges.util.exchange_util as exchange_util
 
 
@@ -45,15 +47,23 @@ def create_client(
                     f"exchange with ccxt in version {ccxt.__version__}")
     if exchange_manager.ignore_config or exchange_manager.check_config(exchange_manager.exchange_name):
         try:
-            key, secret, password, uid = exchange_manager.get_exchange_credentials(exchange_manager.exchange_name)
+            auth_token_header_prefix = None
+            key, secret, password, uid, auth_token = exchange_manager.get_exchange_credentials(
+                exchange_manager.exchange_name
+            )
             if keys_adapter:
-                key, secret, password, uid = keys_adapter(key, secret, password, uid)
+                key, secret, password, uid, auth_token, auth_token_header_prefix = keys_adapter(
+                    key, secret, password, uid, auth_token
+                )
             if not (key and secret) and not exchange_manager.is_simulated and not exchange_manager.ignore_config:
                 logger.warning(f"No exchange API key set for {exchange_manager.exchange_name}. "
                                f"Enter your account details to enable real trading on this exchange.")
             if should_authenticate and not exchange_manager.is_backtesting:
-                client = exchange_class(_get_client_config(options, headers, additional_config,
-                                                           key, secret, password, uid))
+                client = exchange_class(_get_client_config(
+                    options, headers, additional_config,
+                    api_key=key, secret=secret, password=password, uid=uid,
+                    auth_token=auth_token, auth_token_header_prefix=auth_token_header_prefix
+                ))
                 is_authenticated = True
                 if exchange_manager.check_credentials:
                     client.checkRequiredCredentials()
@@ -62,14 +72,16 @@ def create_client(
         except (ccxt.AuthenticationError, Exception) as e:
             if unauthenticated_exchange_fallback is None:
                 return get_unauthenticated_exchange(
-                    exchange_class, options, headers, additional_config
+                    exchange_class, options, headers, additional_config, exchange_manager.proxy_config
                 ), False
             return unauthenticated_exchange_fallback(e), False
     else:
-        client = get_unauthenticated_exchange(exchange_class, options, headers, additional_config)
+        client = get_unauthenticated_exchange(
+            exchange_class, options, headers, additional_config, exchange_manager.proxy_config
+        )
         logger.error("configuration issue: missing login information !")
     client.logger.setLevel(logging.INFO)
-    _use_http_proxy_if_necessary(client)
+    _use_proxy_if_necessary(client, exchange_manager.proxy_config)
     return client, is_authenticated
 
 
@@ -95,9 +107,11 @@ async def close_client(client):
     client.orderbooks = {}
 
 
-def get_unauthenticated_exchange(exchange_class, options, headers, additional_config):
+def get_unauthenticated_exchange(
+    exchange_class, options, headers, additional_config, proxy_config: proxy_config_import.ProxyConfig
+):
     client = exchange_class(_get_client_config(options, headers, additional_config))
-    _use_http_proxy_if_necessary(client)
+    _use_proxy_if_necessary(client, proxy_config)
     return client
 
 
@@ -240,11 +254,29 @@ def converted_ccxt_common_errors(f):
     return converted_ccxt_common_errors_wrapper
 
 
-def _use_http_proxy_if_necessary(client):
-    client.aiohttp_trust_env = constants.ENABLE_EXCHANGE_HTTP_PROXY_FROM_ENV
-
-
-def _get_client_config(options, headers, additional_config, api_key=None, secret=None, password=None, uid=None):
+def _use_proxy_if_necessary(client, proxy_config: proxy_config_import.ProxyConfig):
+    client.aiohttp_trust_env = proxy_config.aiohttp_trust_env
+    if proxy_config.http_proxy:
+        client.http_proxy = proxy_config.http_proxy
+    if proxy_config.http_proxy_callback:
+        client.http_proxy_callback = proxy_config.http_proxy_callback
+    if proxy_config.https_proxy:
+        client.https_proxy = proxy_config.https_proxy
+    if proxy_config.https_proxy_callback:
+        client.https_proxy_callback = proxy_config.https_proxy_callback
+    if proxy_config.socks_proxy:
+        client.socks_proxy = proxy_config.socks_proxy
+    if proxy_config.socks_proxy_callback:
+        client.socks_proxy_callback = proxy_config.socks_proxy_callback
+
+
+def _get_client_config(
+    options, headers, additional_config,
+    api_key=None, secret=None, password=None, uid=None,
+    auth_token=None, auth_token_header_prefix=None
+):
+    if auth_token is not None:
+        headers["Authorization"] = f"{auth_token_header_prefix or ''}{auth_token}"
     config = {
         'verbose': constants.ENABLE_CCXT_VERBOSE,
         'enableRateLimit': constants.ENABLE_CCXT_RATE_LIMIT,
@@ -262,3 +294,7 @@ def _get_client_config(options, headers, additional_config, api_key=None, secret
         config['uid'] = uid
     config.update(additional_config or {})
     return config
+
+
+def ccxt_exchange_class_factory(exchange_name):
+    return getattr(async_ccxt, exchange_name)

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/connectors/ccxt/ccxt_clients_cache.py

@@ -14,6 +14,7 @@
 #  You should have received a copy of the GNU Lesser General Public
 #  License along with this library.
 import cachetools
+import json
 
 import octobot_commons.constants as commons_constants
 
@@ -22,12 +23,12 @@ _MARKETS_BY_EXCHANGE = cachetools.TTLCache(maxsize=50, ttl=commons_constants.DAY
 
 
 def get_client_key(client) -> str:
-    return client.__class__.__name__
+    return f"{client.__class__.__name__}:{json.dumps(client.urls.get('api'))}"
 
 
-def get_exchange_parsed_markets(exchange: str):
-    return _MARKETS_BY_EXCHANGE[exchange]
+def get_exchange_parsed_markets(client_key: str):
+    return _MARKETS_BY_EXCHANGE[client_key]
 
 
-def set_exchange_parsed_markets(exchange: str, markets):
-    _MARKETS_BY_EXCHANGE[exchange] = markets
+def set_exchange_parsed_markets(client_key: str, markets):
+    _MARKETS_BY_EXCHANGE[client_key] = markets

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/connectors/ccxt/ccxt_connector.py

@@ -200,7 +200,7 @@ class CCXTConnector(abstract_exchange.AbstractExchange):
 
     def _create_exchange_type(self):
         if self.is_supporting_exchange(self.rest_name):
-            self.exchange_type = getattr(ccxt, self.rest_name)
+            self.exchange_type = ccxt_client_util.ccxt_exchange_class_factory(self.rest_name)
         else:
             self.exchange_type = self.rest_name
 
@@ -256,7 +256,8 @@ class CCXTConnector(abstract_exchange.AbstractExchange):
             self.handle_token_error(err)
         return ccxt_client_util.get_unauthenticated_exchange(
             self.exchange_type,
-            self.options, self.headers, self.additional_config
+            self.options, self.headers, self.additional_config,
+            self.exchange_manager.proxy_config
         )
 
     def get_market_status(self, symbol, price_example=None, with_fixer=True):

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/connectors/simulator/ccxt_client_simulation.py

@@ -13,12 +13,11 @@
 #
 #  You should have received a copy of the GNU Lesser General Public
 #  License along with this library.
-import ccxt.async_support
 import octobot_trading.exchanges.connectors.ccxt.ccxt_client_util as ccxt_client_util
 
 
 def parse_markets(exchange_name, market_filter) -> dict:
-    client = getattr(ccxt.async_support, exchange_name)()
+    client = ccxt_client_util.ccxt_exchange_class_factory(exchange_name)()
     ccxt_client_util.load_markets_from_cache(client, market_filter)
     return client.markets
 

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/exchange_builder.py

@@ -269,6 +269,10 @@ class ExchangeBuilder:
         self.exchange_manager.bot_id = bot_id
         return self
 
+    def set_proxy_config(self, proxy_config):
+        self.exchange_manager.proxy_config = proxy_config
+        return self
+
     def disable_trading_mode(self):
         self._is_using_trading_modes = False
         return self

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/exchange_manager.py

@@ -58,6 +58,7 @@ class ExchangeManager(util.Initializable):
         self.without_auth: bool = False
         self.check_credentials: bool = True
         self.enable_storage: bool = True
+        self.proxy_config: exchanges.ProxyConfig = exchanges.ProxyConfig.default_env_var_config()
 
         # exchange_only is True when exchange channels are not required (therefore not created)
         self.exchange_only: bool = False
@@ -306,7 +307,7 @@ class ExchangeManager(util.Initializable):
 
     def get_exchange_credentials(self, exchange_name):
         if self.ignore_config or not self.should_decrypt_token() or self.without_auth:
-            return "", "", "", ""
+            return "", "", "", "", ""
         config_exchange = self.config[common_constants.CONFIG_EXCHANGES][exchange_name]
         key = configuration.decrypt_element_if_possible(
             common_constants.CONFIG_EXCHANGE_KEY, config_exchange, None
@@ -321,7 +322,8 @@ class ExchangeManager(util.Initializable):
             configuration.decrypt_element_if_possible(
                 common_constants.CONFIG_EXCHANGE_PASSWORD, config_exchange, None
             ),
-            config_exchange.get(common_constants.CONFIG_EXCHANGE_UID, "")
+            config_exchange.get(common_constants.CONFIG_EXCHANGE_UID, ""),
+            config_exchange.get(common_constants.CONFIG_EXCHANGE_ACCESS_TOKEN, "")
         )
 
     def get_exchange_sub_account_id(self, exchange_name):

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/exchanges/util/exchange_util.py

@@ -33,6 +33,7 @@ import octobot_trading.constants as constants
 import octobot_trading.exchanges.types as exchanges_types
 import octobot_trading.exchanges.implementations as exchanges_implementations
 import octobot_trading.exchanges.connectors.ccxt.enums as ccxt_enums
+import octobot_trading.exchanges.connectors.ccxt.ccxt_client_util as ccxt_client_util
 import octobot_trading.exchanges.exchange_details as exchange_details
 import octobot_trading.exchanges.exchange_builder as exchange_builder
 
@@ -128,7 +129,7 @@ async def get_exchange_details(
                 aiohttp_session, auto_filled_exchanges[exchange_name][1], exchange_name
             )
     try:
-        exchange = getattr(ccxt, exchange_name)()
+        exchange = ccxt_client_util.ccxt_exchange_class_factory(exchange_name)()
         return exchange_details.ExchangeDetails(
             exchange.id,
             exchange.name,

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/octobot_trading/util/test_tools/exchange_data.py

@@ -33,7 +33,7 @@ class ExchangeAuthDetails(octobot_commons.dataclasses.FlexibleDataclass, octobot
     api_key: str = ""
     api_secret: str = ""
     api_password: str = ""
-    oauth_token: str = ""
+    access_token: str = ""
     exchange_type: str = ""
     sandboxed: bool = False
     broker_enabled: bool = False

{OctoBot-Trading-2.4.115 → OctoBot-Trading-2.4.117}/tests/exchanges/__init__.py

@@ -31,10 +31,12 @@ import octobot_trading.exchanges.connectors.ccxt.ccxt_clients_cache as ccxt_clie
 from octobot_commons.tests.test_config import load_test_config
 from octobot_trading.api.exchange import create_exchange_builder, cancel_ccxt_throttle_task
 from octobot_trading.exchanges.exchange_manager import ExchangeManager
+from octobot_trading.exchanges.exchanges import Exchanges
 from octobot_trading.exchanges.implementations.default_rest_exchange import DefaultRestExchange
 from octobot_trading.exchanges.connectors.ccxt.ccxt_connector import CCXTConnector
 from octobot_trading.exchanges.traders.trader_simulator import TraderSimulator
 import octobot_trading.personal_data as personal_data
+import octobot_trading.exchanges.connectors.ccxt.ccxt_client_util as ccxt_client_util
 from octobot_trading.enums import FeePropertyColumns, ExchangeConstantsMarketPropertyColumns, \
     ExchangeConstantsMarketPropertyColumns
 
@@ -62,10 +64,10 @@ class MockedCCXTConnector(CCXTConnector):
         return DEFAULT_EXCHANGE_NAME
 
     async def load_symbol_markets(self, reload=False, market_filter=None):
-        if forced_markets := mock_exchanges_data.MOCKED_EXCHANGE_SYMBOL_DETAILS.get(
+        if mock_exchanges_data.MOCKED_EXCHANGE_SYMBOL_DETAILS.get(
             self.exchange_manager.exchange_name, None
         ):
-            ccxt_clients_cache.set_exchange_parsed_markets(self.exchange_manager.exchange_name, forced_markets)
+            register_market_status_mocks(self.exchange_manager.exchange_name)
         await super().load_symbol_markets(
             reload=reload,
         )
@@ -365,3 +367,25 @@ async def future_backtesting_trader(backtesting_config, backtesting_exchange_man
     trader_instance = TraderSimulator(backtesting_config, backtesting_exchange_manager)
     await trader_instance.initialize()
     return backtesting_config, backtesting_exchange_manager, trader_instance
+
+
+@contextlib.asynccontextmanager
+async def cached_markets_exchange_manager(config, exchange_name, exchange_only=False):
+    exchange_manager = None
+    try:
+        exchange_manager = ExchangeManager(config, exchange_name)
+        register_market_status_mocks(exchange_name)
+        exchange_manager.exchange_only = exchange_only
+        await exchange_manager.initialize(exchange_config_by_exchange=None)
+        Exchanges.instance().add_exchange(exchange_manager, "")
+        yield exchange_manager
+    finally:
+        if exchange_manager is not None:
+            await exchange_manager.stop()
+
+
+def register_market_status_mocks(exchange_name):
+    ccxt_clients_cache.set_exchange_parsed_markets(
+        ccxt_clients_cache.get_client_key(ccxt_client_util.ccxt_exchange_class_factory(exchange_name)()),
+        mock_exchanges_data.MOCKED_EXCHANGE_SYMBOL_DETAILS[exchange_name]
+    )