From 8f2a0e674e0ad6dbfd7d8ccd0cd5869cf5601ed1 Mon Sep 17 00:00:00 2001 From: cardosofede Date: Fri, 4 Aug 2023 18:39:02 +0200 Subject: [PATCH] (feat) add specific file explorers --- ui_components/directional_strategies_file_explorer.py | 3 ++- ui_components/optimizations_file_explorer.py | 5 +++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/ui_components/directional_strategies_file_explorer.py b/ui_components/directional_strategies_file_explorer.py index 9f8f436..8073972 100644 --- a/ui_components/directional_strategies_file_explorer.py +++ b/ui_components/directional_strategies_file_explorer.py @@ -8,7 +8,8 @@ from utils.os_utils import get_python_files_from_directory class DirectionalStrategiesFileExplorer(FileExplorerBase): def add_tree_view(self): with mui.lab.TreeView(defaultExpandIcon=mui.icon.ChevronRight, defaultCollapseIcon=mui.icon.ExpandMore, - onNodeSelect=lambda event, node_id: self.set_selected_file(event, node_id)): + onNodeSelect=lambda event, node_id: self.set_selected_file(event, node_id), + defaultExpanded=["directional_strategies"]): with mui.lab.TreeItem(nodeId="directional_strategies", label=f"⚔️Directional Strategies"): strategies = get_python_files_from_directory(constants.DIRECTIONAL_STRATEGIES_PATH) for strategy in strategies: diff --git a/ui_components/optimizations_file_explorer.py b/ui_components/optimizations_file_explorer.py index 8c7c605..c0354ab 100644 --- a/ui_components/optimizations_file_explorer.py +++ b/ui_components/optimizations_file_explorer.py @@ -8,8 +8,9 @@ from utils.os_utils import get_python_files_from_directory class OptimizationsStrategiesFileExplorer(FileExplorerBase): def add_tree_view(self): with mui.lab.TreeView(defaultExpandIcon=mui.icon.ChevronRight, defaultCollapseIcon=mui.icon.ExpandMore, - onNodeSelect=lambda event, node_id: self.set_selected_file(event, node_id)): - with mui.lab.TreeItem(nodeId="optimization_strategies", label=f"🔬Optimizations"): + onNodeSelect=lambda event, node_id: self.set_selected_file(event, node_id), + defaultExpanded=["optimization_strategies"]): + with mui.lab.TreeItem(nodeId="optimization_strategies", label=f"🔬Studies"): optimizations = get_python_files_from_directory(constants.OPTIMIZATIONS_PATH) for optimization in optimizations: mui.lab.TreeItem(nodeId=optimization, label=f"🐍{optimization.split('/')[-1]}")