(feat) add optimizations file explorer card

This commit is contained in:
cardosofede
2023-08-04 15:05:11 +02:00
parent 587cb2be14
commit aa97cf658d

View File

@@ -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]}")