refactor: provide a proper TUI interface with textual
This commit is contained in:
@@ -0,0 +1,236 @@
|
||||
"""
|
||||
A script to randomly select files from a directory, open them, and allow the user to upload them to a Discord webhook, delete them, or skip to the next file.
|
||||
|
||||
You can specify a file size limit, whether to descend into subdirectories, and provide blacklist/whitelist patterns for file selection.
|
||||
|
||||
Here's an example usage:
|
||||
./rand.py -s 5 -d /path/to/directory --descend --blacklist "*.png" "*.mp4"
|
||||
"""
|
||||
|
||||
import pathlib
|
||||
import random
|
||||
|
||||
import typed_argparse as tap
|
||||
from textual.app import App, ComposeResult
|
||||
from textual.binding import BindingType
|
||||
from textual.widgets import Footer, Header, Label
|
||||
|
||||
from random_file_picker.core import collect_files, move_file_to_meme_folder, upload_file
|
||||
from random_file_picker.openers import open_file_in_dolphin, open_file_with_default_app, open_file_with_geeqie
|
||||
|
||||
IMG_EXTENSIONS = [
|
||||
".jpg", ".jpeg", ".webp", ".png", ".gif",
|
||||
]
|
||||
|
||||
class Args(tap.TypedArgs):
|
||||
file_size: int = tap.arg("-s", default=10, help="Maximum file size in MB")
|
||||
directory: str = tap.arg("-d", default=".", help="Directory to search for files")
|
||||
descend: bool = tap.arg("--descend", default=False, help="Whether to descend into subdirectories")
|
||||
use_geeqie: bool = tap.arg("--use-geeqie", default=False, help="Whether to use Geeqie as the image viewer")
|
||||
blacklist: list[str] | None = tap.arg("--blacklist", nargs="*", help="List of glob patterns to blacklist files")
|
||||
whitelist: list[str] | None = tap.arg("--whitelist", nargs="*", help="List of glob patterns to whitelist files")
|
||||
|
||||
class MenuApp(App[None]):
|
||||
# This defines the keys shown in the footer
|
||||
BINDINGS: list[BindingType] = [
|
||||
("left", "prev_file", "Previous File"),
|
||||
("right", "next_file", "Next File"),
|
||||
("r", "reopen_file", "Reopen"),
|
||||
("u", "upload", "Upload"),
|
||||
("d", "delete", "Delete"),
|
||||
("m", "move_to_meme", "Move Meme"),
|
||||
("f", "open_folder", "Open Folder"),
|
||||
("q", "quit", "Quit"),
|
||||
]
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
directory: str = ".",
|
||||
file_limit: int = 10,
|
||||
recurse: bool = False,
|
||||
blacklist: list[str] | None = None,
|
||||
whitelist: list[str] | None = None,
|
||||
use_geeqie: bool = False,
|
||||
):
|
||||
super().__init__()
|
||||
self.directory: pathlib.Path = pathlib.Path(directory)
|
||||
self.file_limit: int = file_limit
|
||||
self.recurse: bool = recurse
|
||||
self.blacklist_patterns: list[str] = blacklist or []
|
||||
self.whitelist_patterns: list[str] = whitelist or []
|
||||
self.use_geeqie: bool = use_geeqie
|
||||
self.usable_files: list[pathlib.Path] = []
|
||||
self.current_index: int = 0
|
||||
self.current_file: pathlib.Path | None = None
|
||||
|
||||
def compose(self) -> ComposeResult:
|
||||
yield Header()
|
||||
yield Label("Loading files...", id="status")
|
||||
yield Label("", id="current-file")
|
||||
yield Footer()
|
||||
|
||||
def on_mount(self) -> None:
|
||||
self.gather_files()
|
||||
if self.usable_files:
|
||||
self.current_index = 0
|
||||
self.display_current_file()
|
||||
|
||||
def display_current_file(self) -> None:
|
||||
"""Display the current file based on current_index."""
|
||||
if not self.usable_files or self.current_index >= len(self.usable_files):
|
||||
self.query_one("#status", Label).update("❌ No files available")
|
||||
self.current_file = None
|
||||
return
|
||||
|
||||
self.current_file = self.usable_files[self.current_index]
|
||||
file_count_str = f"({self.current_index + 1}/{len(self.usable_files)})"
|
||||
self.query_one("#current-file", Label).update(f"📄 {self.current_file} {file_count_str}")
|
||||
self.open_file()
|
||||
|
||||
def gather_files(self) -> None:
|
||||
"""Collect files from the directory based on filters."""
|
||||
self.usable_files = []
|
||||
|
||||
if not self.directory.exists() or not self.directory.is_dir():
|
||||
self.query_one("#status", Label).update(f"❌ Directory '{self.directory}' not found")
|
||||
return
|
||||
|
||||
self.usable_files = collect_files(
|
||||
search_path=self.directory,
|
||||
search_subdirectories=self.recurse,
|
||||
file_size_limit=self.file_limit,
|
||||
blacklist_patterns=self.blacklist_patterns,
|
||||
whitelist_patterns=self.whitelist_patterns,
|
||||
)
|
||||
|
||||
if not self.usable_files:
|
||||
self.query_one("#status", Label).update("❌ No usable files found")
|
||||
else:
|
||||
self.query_one("#status", Label).update(f"✅ Found {len(self.usable_files)} files")
|
||||
random.shuffle(self.usable_files)
|
||||
|
||||
def open_file(self) -> None:
|
||||
"""Open the current file - display images in terminal, play others externally."""
|
||||
if not self.current_file:
|
||||
return
|
||||
|
||||
self.query_one("#status", Label).update(f"📝 Open {self.current_file}")
|
||||
try:
|
||||
if self.use_geeqie and str(self.current_file).lower().endswith(tuple(IMG_EXTENSIONS)):
|
||||
open_file_with_geeqie(self.current_file)
|
||||
else:
|
||||
ok = open_file_with_default_app(self.current_file)
|
||||
if not ok:
|
||||
self.query_one("#status", Label).update("❌ Failed to open file with default app")
|
||||
except Exception as e:
|
||||
self.query_one("#status", Label).update(f"❌ Failed to open file: {str(e)}")
|
||||
|
||||
def send_file(self) -> None:
|
||||
"""Upload the current file to Discord webhook."""
|
||||
if not self.current_file:
|
||||
return
|
||||
|
||||
try:
|
||||
self.query_one("#status", Label).update(f"⏳ Uploading {self.current_file.name} to webhook...")
|
||||
upload_file(self.current_file)
|
||||
self.query_one("#status", Label).update(f"✅ Uploaded {self.current_file.name} to webhook")
|
||||
except Exception as e:
|
||||
self.query_one("#status", Label).update(f"❌ Upload failed: {str(e)}")
|
||||
|
||||
def delete_file(self) -> None:
|
||||
"""Delete the current file."""
|
||||
if not self.current_file or self.current_index >= len(self.usable_files):
|
||||
return
|
||||
|
||||
try:
|
||||
self.current_file.unlink()
|
||||
_ = self.usable_files.pop(self.current_index)
|
||||
self.query_one("#status", Label).update("✅ Deleted file")
|
||||
|
||||
# Adjust index if we deleted the last file
|
||||
if self.current_index >= len(self.usable_files) and self.usable_files:
|
||||
self.current_index = len(self.usable_files) - 1
|
||||
|
||||
if self.usable_files:
|
||||
self.display_current_file()
|
||||
else:
|
||||
self.query_one("#status", Label).update("❌ No more files")
|
||||
except Exception as e:
|
||||
self.query_one("#status", Label).update(f"❌ Delete failed: {str(e)}")
|
||||
|
||||
def move_to_meme_folder(self) -> None:
|
||||
"""Move the current file to the meme folder."""
|
||||
if not self.current_file or self.current_index >= len(self.usable_files):
|
||||
return
|
||||
|
||||
try:
|
||||
move_file_to_meme_folder(self.current_file)
|
||||
_ = self.usable_files.pop(self.current_index)
|
||||
self.query_one("#status", Label).update("✅ Moved to meme folder")
|
||||
|
||||
# Adjust index if we removed the last file
|
||||
if self.current_index >= len(self.usable_files) and self.usable_files:
|
||||
self.current_index = len(self.usable_files) - 1
|
||||
|
||||
if self.usable_files:
|
||||
self.display_current_file()
|
||||
else:
|
||||
self.query_one("#status", Label).update("❌ No more files")
|
||||
except Exception as e:
|
||||
self.query_one("#status", Label).update(f"❌ Move failed: {str(e)}")
|
||||
|
||||
def open_in_folder(self) -> None:
|
||||
"""Open the file's folder in file manager."""
|
||||
if not self.current_file:
|
||||
return
|
||||
|
||||
open_file_in_dolphin(self.current_file)
|
||||
self.query_one("#status", Label).update("📂 Opened file in folder")
|
||||
|
||||
def action_upload(self) -> None:
|
||||
self.send_file()
|
||||
|
||||
def action_delete(self) -> None:
|
||||
self.delete_file()
|
||||
|
||||
def action_move_to_meme(self) -> None:
|
||||
self.move_to_meme_folder()
|
||||
|
||||
def action_open_folder(self) -> None:
|
||||
self.open_in_folder()
|
||||
|
||||
def action_next_file(self) -> None:
|
||||
if not self.usable_files:
|
||||
return
|
||||
self.current_index = (self.current_index + 1) % len(self.usable_files)
|
||||
self.display_current_file()
|
||||
|
||||
def action_prev_file(self) -> None:
|
||||
if not self.usable_files:
|
||||
return
|
||||
self.current_index = (self.current_index - 1) % len(self.usable_files)
|
||||
self.display_current_file()
|
||||
|
||||
def action_reopen_file(self) -> None:
|
||||
if not self.current_file:
|
||||
return
|
||||
self.open_file()
|
||||
self.query_one("#status", Label).update("🔄 Reopened file")
|
||||
|
||||
|
||||
def main_with_tap(args: Args) -> None:
|
||||
app = MenuApp(
|
||||
directory=args.directory,
|
||||
file_limit=args.file_size,
|
||||
recurse=args.descend,
|
||||
blacklist=args.blacklist or [],
|
||||
whitelist=args.whitelist or [],
|
||||
use_geeqie=args.use_geeqie,
|
||||
)
|
||||
app.run()
|
||||
|
||||
def main() -> None:
|
||||
tap.Parser(Args).bind(main_with_tap).run()
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user