From 8d6e613c4a759859ee76cbba303c8a9e4af8dbc9 Mon Sep 17 00:00:00 2001 From: cardosofede Date: Thu, 7 Sep 2023 17:38:44 +0800 Subject: [PATCH] (feat) rename controllers for strategies --- utils/os_utils.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/utils/os_utils.py b/utils/os_utils.py index 07c99d7..abb0863 100644 --- a/utils/os_utils.py +++ b/utils/os_utils.py @@ -12,9 +12,11 @@ def remove_files_from_directory(directory: str): for file in os.listdir(directory): os.remove(f"{directory}/{file}") + def remove_file(file_path: str): os.remove(file_path) + def remove_directory(directory: str): process = subprocess.Popen(f"rm -rf {directory}", shell=True) process.wait() @@ -76,22 +78,22 @@ def get_yml_files_from_directory(directory: str) -> list: return yml -def load_directional_strategies(path): - strategies = {} +def load_controllers(path): + controllers = {} for filename in os.listdir(path): if filename.endswith('.py') and "__init__" not in filename: module_name = filename[:-3] # strip the .py to get the module name - strategies[module_name] = {"module": module_name} + controllers[module_name] = {"module": module_name} file_path = os.path.join(path, filename) spec = importlib.util.spec_from_file_location(module_name, file_path) module = importlib.util.module_from_spec(spec) spec.loader.exec_module(module) for name, cls in inspect.getmembers(module, inspect.isclass): if issubclass(cls, DirectionalTradingControllerBase) and cls is not DirectionalTradingControllerBase: - strategies[module_name]["class"] = cls + controllers[module_name]["class"] = cls if issubclass(cls, DirectionalTradingControllerConfigBase) and cls is not DirectionalTradingControllerConfigBase: - strategies[module_name]["config"] = cls - return strategies + controllers[module_name]["config"] = cls + return controllers def get_function_from_file(file_path: str, function_name: str):