working on multiple editor types
This commit is contained in:
@@ -9,13 +9,18 @@ import os, shutil
|
||||
|
||||
|
||||
class CustomDirectoryTree(DirectoryTree):
|
||||
|
||||
ICON_NODE = " "
|
||||
ICON_NODE_EXPANDED = " "
|
||||
ICON_FILE = " "
|
||||
|
||||
def __init__(self, path, *, name = None, id = None, classes = None, disabled = False):
|
||||
super().__init__(path, name=name, id=id, classes=classes, disabled=disabled)
|
||||
self.right_clicked_node: TreeNode | None = None
|
||||
|
||||
|
||||
|
||||
def context_menu_chosen(self, result):
|
||||
def context_menu_chosen(self, file_name: str, result: str):
|
||||
if result == "Delete":
|
||||
def delete_confirm(will_delete: bool | None):
|
||||
if will_delete == True:
|
||||
@@ -44,7 +49,7 @@ class CustomDirectoryTree(DirectoryTree):
|
||||
self.right_clicked_node = None
|
||||
|
||||
self.app.push_screen(Prompt(f"Enter the new name for \"{self.right_clicked_node.label}\".", "string", "Rename"), rename_confirm)
|
||||
elif result == "New Folder":
|
||||
elif result == "New folder":
|
||||
def new_folder(folder_name: str | None):
|
||||
if folder_name == None: return
|
||||
if folder_name.strip() == "":
|
||||
@@ -60,7 +65,7 @@ class CustomDirectoryTree(DirectoryTree):
|
||||
return
|
||||
|
||||
self.app.push_screen(Prompt("Enter the name of the new folder.", "string", "Create New Folder"), new_folder)
|
||||
elif result == "New File":
|
||||
elif result == "New file":
|
||||
def new_file(file_name: str | None):
|
||||
if file_name == None: return
|
||||
if file_name.strip() == "":
|
||||
@@ -82,6 +87,24 @@ class CustomDirectoryTree(DirectoryTree):
|
||||
return
|
||||
|
||||
self.app.push_screen(Prompt("Enter the name of the new file", "string", "Create New File"), new_file)
|
||||
elif result == "Open with...":
|
||||
file_extension = file_name.rsplit(".")[1]
|
||||
default_editors = eval(self.app.config_handler.get("editor", "default_editors"))
|
||||
default_editor = default_editors.get(file_extension, "Text Editor")
|
||||
|
||||
def open_with(chosen_editor):
|
||||
self.notify(str(chosen_editor))
|
||||
|
||||
self.app.push_screen(Prompt(
|
||||
"Editor for file:",
|
||||
"list",
|
||||
"Choose editor for file",
|
||||
values=[
|
||||
default_editor + " (Default)",
|
||||
"Text Editor"
|
||||
],
|
||||
allow_blank=False
|
||||
), open_with)
|
||||
|
||||
def on_mouse_down(self, event: MouseDown):
|
||||
if event.button != 3 or not "line" in event.style.meta:
|
||||
@@ -89,16 +112,18 @@ class CustomDirectoryTree(DirectoryTree):
|
||||
selected_node = self.get_node_at_line(event.style.meta["line"])
|
||||
self.right_clicked_node = selected_node
|
||||
|
||||
spacer = NoSelectStatic(f'[d]{"-" * 17}[/]')
|
||||
|
||||
options = None
|
||||
if self._safe_is_dir(self.right_clicked_node.data.path):
|
||||
options = ["New Folder", "New File", NoSelectStatic(f'[d]{"-" * 17}[/]'), "Delete", "Rename"]
|
||||
options = ["New folder", "New file", spacer, "Delete", "Rename"]
|
||||
else:
|
||||
options = ["Delete", "Rename"]
|
||||
options = ["Delete", "Rename", spacer, "Open with..."]
|
||||
|
||||
file_name = str(self.right_clicked_node.label) if len(self.right_clicked_node.label) <= 17 else self.right_clicked_node.label[:14] + "..."
|
||||
|
||||
self.app.push_screen(ContextMenu(
|
||||
[NoSelectStatic(f"[b]{file_name}[/]"), NoSelectStatic(f'[d]{"-" * 17}[/]')] + options,
|
||||
event.screen_offset
|
||||
), self.context_menu_chosen)
|
||||
), lambda result: self.context_menu_chosen(file_name, result))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user