From aa97cf658d0eafc2bc12502214dc49e6d4a641d0 Mon Sep 17 00:00:00 2001 From: cardosofede Date: Fri, 4 Aug 2023 15:05:11 +0200 Subject: [PATCH] (feat) add optimizations file explorer card --- ui_components/optimizations_file_explorer.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 ui_components/optimizations_file_explorer.py diff --git a/ui_components/optimizations_file_explorer.py b/ui_components/optimizations_file_explorer.py new file mode 100644 index 0000000..8c7c605 --- /dev/null +++ b/ui_components/optimizations_file_explorer.py @@ -0,0 +1,15 @@ +from streamlit_elements import mui + +import constants +from ui_components.file_explorer_base import FileExplorerBase +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"): + optimizations = get_python_files_from_directory(constants.OPTIMIZATIONS_PATH) + for optimization in optimizations: + mui.lab.TreeItem(nodeId=optimization, label=f"🐍{optimization.split('/')[-1]}")