implementing some of the right click options

This commit is contained in:
SpookyDervish
2025-10-31 06:47:02 +11:00
parent a5ee621b46
commit 5e38909640
2 changed files with 111 additions and 1 deletions

View File

@@ -1,16 +1,46 @@
from textual.widgets import DirectoryTree, Rule
from textual.widgets.tree import TreeNode
from textual.events import MouseDown
from prompt import Prompt
from context_menu import ContextMenu, NoSelectStatic
import os, shutil
class CustomDirectoryTree(DirectoryTree):
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):
self.right_clicked_node = None
if result == "Open":
self.select_node(self.right_clicked_node)
self.right_clicked_node = None
elif result == "Delete":
def delete_confirm(will_delete: bool | None):
if will_delete == True:
shutil.rmtree(self.right_clicked_node.data.path)
self.reload()
self.notify(f"Deleted \"{self.right_clicked_node.data.path}\".")
self.right_clicked_node = None
self.app.push_screen(Prompt(f"Are you sure you want to delete \"{self.right_clicked_node.label}\"?", "confirm", "Confirm deletion"), delete_confirm)
elif result == "Rename":
def rename_confirm(new_name: str | None):
if new_name == None: return
os.rename(self.right_clicked_node.data.path, os.path.join(os.path.dirname(self.right_clicked_node.data.path), new_name))
self.reload()
self.notify("Renamed successfully.")
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)
def on_mouse_down(self, event: MouseDown):
if event.button != 3 or not "line" in event.style.meta: