[Community] add get_exchange_type_from_internal_name

Drakkar-Software · Dec 27, 2024 · 8090ed1 · 8090ed1
1 parent 8443936
commit 8090ed1
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/octobot/community/__init__.py b/octobot/community/__init__.py
@@ -33,6 +33,7 @@
     StrategyData,
     get_exchange_type_from_availability,
     to_bot_exchange_internal_name,
+    get_exchange_type_from_internal_name,
     to_community_exchange_internal_name,
 )
 from octobot.community.supabase_backend import (
@@ -110,6 +111,7 @@
     "StrategyData",
     "get_exchange_type_from_availability",
     "to_bot_exchange_internal_name",
+    "get_exchange_type_from_internal_name",
     "to_community_exchange_internal_name",
     "SyncConfigurationStorage",
     "ASyncConfigurationStorage",

diff --git a/octobot/community/models/__init__.py b/octobot/community/models/__init__.py
@@ -48,6 +48,7 @@
     format_portfolio_with_profitability,
     get_exchange_type_from_availability,
     to_bot_exchange_internal_name,
+    get_exchange_type_from_internal_name,
     to_community_exchange_internal_name,
 )
 from octobot.community.models.community_public_data import (
@@ -71,6 +72,7 @@
     "format_portfolio_with_profitability",
     "get_exchange_type_from_availability",
     "to_bot_exchange_internal_name",
+    "get_exchange_type_from_internal_name",
     "to_community_exchange_internal_name",
     "CommunityPublicData",
     "StrategyData",

diff --git a/octobot/community/models/formatters.py b/octobot/community/models/formatters.py
@@ -155,6 +155,12 @@ def to_bot_exchange_internal_name(community_exchange_internal_name: str) -> str:
     return community_exchange_internal_name
 
 
+def get_exchange_type_from_internal_name(community_exchange_internal_name: str) -> str:
+    if community_exchange_internal_name.endswith(FUTURES_INTERNAL_NAME_SUFFIX):
+        return commons_constants.CONFIG_EXCHANGE_FUTURE
+    return commons_constants.CONFIG_EXCHANGE_SPOT
+
+
 def get_exchange_type_from_availability(exchange_availability: dict) -> str:
     if not exchange_availability:
         # use spot by default