5ecb9537b7
Adds an input field and button to create a new folder inside the currently selected directory, then refreshes the tree. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
from pathlib import Path
|
|
|
|
from textual.app import ComposeResult
|
|
from textual.screen import ModalScreen
|
|
from textual.widgets import DirectoryTree, Static, Button, Input
|
|
from textual.containers import Horizontal, Vertical
|
|
|
|
|
|
class _DirOnly(DirectoryTree):
|
|
def filter_paths(self, paths):
|
|
return [p for p in paths if p.is_dir()]
|
|
|
|
|
|
class FolderPicker(ModalScreen[str | None]):
|
|
|
|
BINDINGS = [("escape", "cancel", "Cancel")]
|
|
|
|
def __init__(self, title: str = "Select folder", start: str = "/"):
|
|
super().__init__()
|
|
self._title = title
|
|
self._start = start
|
|
|
|
def compose(self) -> ComposeResult:
|
|
with Vertical(id="folder-picker"):
|
|
yield Static(self._title, id="fp-title")
|
|
yield _DirOnly(self._start, id="fp-tree")
|
|
with Horizontal(id="fp-new-row"):
|
|
yield Input(placeholder="New folder name", id="fp-new-name")
|
|
yield Button("Create Folder", id="fp-create")
|
|
with Horizontal(id="fp-buttons"):
|
|
yield Button("Select", variant="primary", id="fp-select")
|
|
yield Button("Cancel", variant="default", id="fp-cancel")
|
|
|
|
def _get_selected_path(self) -> Path | None:
|
|
tree = self.query_one("#fp-tree", DirectoryTree)
|
|
node = tree.cursor_node
|
|
if node and node.data and node.data.path:
|
|
return node.data.path
|
|
return None
|
|
|
|
def on_button_pressed(self, event: Button.Pressed) -> None:
|
|
if event.button.id == "fp-select":
|
|
path = self._get_selected_path()
|
|
self.dismiss(str(path) if path else None)
|
|
elif event.button.id == "fp-create":
|
|
self._create_folder()
|
|
else:
|
|
self.dismiss(None)
|
|
|
|
def _create_folder(self) -> None:
|
|
name = self.query_one("#fp-new-name", Input).value.strip()
|
|
if not name:
|
|
self.notify("Enter a folder name", severity="error")
|
|
return
|
|
if "/" in name or "\0" in name:
|
|
self.notify("Invalid folder name", severity="error")
|
|
return
|
|
parent = self._get_selected_path()
|
|
if not parent:
|
|
parent = Path(self._start)
|
|
new_dir = parent / name
|
|
try:
|
|
new_dir.mkdir(parents=True, exist_ok=True)
|
|
self.notify(f"Created: {new_dir}")
|
|
tree = self.query_one("#fp-tree", DirectoryTree)
|
|
tree.reload()
|
|
self.query_one("#fp-new-name", Input).value = ""
|
|
except OSError as e:
|
|
self.notify(f"Failed: {e}", severity="error")
|
|
|
|
def action_cancel(self) -> None:
|
|
self.dismiss(None)
|