aljaz/hummingbot-dashboard
1
0
Fork 0
mirror of https://github.com/aljazceru/hummingbot-dashboard.git synced 2025-12-31 03:54:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

(feat) add type of controller to dict

Browse Source
This commit is contained in:
cardosofede
2023-10-09 16:53:34 -03:00
parent 6a8d5f086e
commit e30494affc
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
utils/os_utils.py
View File

@@ -6,6 +6,8 @@ import os
from hummingbot.smart_components.strategy_frameworks.directional_trading import DirectionalTradingControllerBase, DirectionalTradingControllerConfigBase
import yaml
from hummingbot.smart_components.strategy_frameworks.market_making import MarketMakingControllerBase, \
MarketMakingControllerConfigBase
def remove_files_from_directory(directory: str):
@@ -91,8 +93,14 @@ def load_controllers(path):
for name, cls in inspect.getmembers(module, inspect.isclass):
if issubclass(cls, DirectionalTradingControllerBase) and cls is not DirectionalTradingControllerBase:
controllers[module_name]["class"] = cls
controllers[module_name]["type"] = "directional_trading"
if issubclass(cls, DirectionalTradingControllerConfigBase) and cls is not DirectionalTradingControllerConfigBase:
controllers[module_name]["config"] = cls
if issubclass(cls, MarketMakingControllerBase) and cls is not MarketMakingControllerBase:
controllers[module_name]["class"] = cls
controllers[module_name]["type"] = "market_making"
if issubclass(cls, MarketMakingControllerConfigBase) and cls is not MarketMakingControllerConfigBase:
controllers[module_name]["config"] = cls
return controllers