(feat) use directory from constants

This commit is contained in:
cardosofede
2023-08-03 18:18:28 +02:00
parent 26291607c8
commit 76acf6735d

View File

@@ -1,5 +1,6 @@
from streamlit_elements import mui
import constants
from ui_components.file_explorer_base import FileExplorerBase
from utils.os_utils import get_directories_from_directory, get_python_files_from_directory, \
get_yml_files_from_directory, get_log_files_from_directory
@@ -7,18 +8,19 @@ from utils.os_utils import get_directories_from_directory, get_python_files_from
class BotsFileExplorer(FileExplorerBase):
def add_tree_view(self):
bots = [bot.split("/")[-2] for bot in get_directories_from_directory(self._directory) if
directory = constants.BOTS_FOLDER
bots = [bot.split("/")[-2] for bot in get_directories_from_directory(directory) if
"data_downloader" not in bot]
with mui.lab.TreeView(defaultExpandIcon=mui.icon.ChevronRight, defaultCollapseIcon=mui.icon.ExpandMore,
onNodeSelect=lambda event, node_id: self.set_selected_file(event, node_id)):
for bot in bots:
with mui.lab.TreeItem(nodeId=bot, label=f"🤖{bot}"):
with mui.lab.TreeItem(nodeId=f"scripts_{bot}", label="🐍Scripts"):
for file in get_python_files_from_directory(f"{self._directory}/{bot}/scripts"):
for file in get_python_files_from_directory(f"{directory}/{bot}/scripts"):
mui.lab.TreeItem(nodeId=file, label=f"📄{file.split('/')[-1]}")
with mui.lab.TreeItem(nodeId=f"strategies_{bot}", label="📜Strategies"):
for file in get_yml_files_from_directory(f"{self._directory}/{bot}/conf/strategies"):
for file in get_yml_files_from_directory(f"{directory}/{bot}/conf/strategies"):
mui.lab.TreeItem(nodeId=file, label=f"📄 {file.split('/')[-1]}")
with mui.lab.TreeItem(nodeId=f"logs_{bot}", label="🗄Logs"):
for file in get_log_files_from_directory(f"{self._directory}/{bot}/logs"):
for file in get_log_files_from_directory(f"{directory}/{bot}/logs"):
mui.lab.TreeItem(nodeId=file, label=f"📄 {file.split('/')[-1]}")