mirror of
https://github.com/shadoll/moma.git
synced 2026-08-28 03:27:34 +00:00
330 lines
12 KiB
Python
330 lines
12 KiB
Python
from textual.screen import Screen
|
|
from textual.widgets import Input, Button, Static
|
|
from textual.containers import Vertical, Horizontal, Center, Container
|
|
from textual.markup import escape
|
|
from pathlib import Path
|
|
import logging
|
|
|
|
|
|
class OpenScreen(Screen):
|
|
def compose(self):
|
|
yield Input(placeholder="Enter directory path", value=".", id="dir_input")
|
|
yield Button("OK", id="ok")
|
|
|
|
def on_button_pressed(self, event):
|
|
if event.button.id == "ok":
|
|
self.submit_path()
|
|
|
|
def on_input_submitted(self, event):
|
|
self.submit_path()
|
|
|
|
def submit_path(self):
|
|
path_str = self.query_one("#dir_input", Input).value
|
|
path = Path(path_str)
|
|
if not path.exists():
|
|
# Show error
|
|
self.query_one("#dir_input", Input).value = f"Path does not exist: {path_str}"
|
|
return
|
|
if not path.is_dir():
|
|
self.query_one("#dir_input", Input).value = f"Not a directory: {path_str}"
|
|
return
|
|
self.app.scan_dir = path # type: ignore
|
|
self.app.scan_files() # type: ignore
|
|
self.app.pop_screen()
|
|
|
|
|
|
class HelpScreen(Screen):
|
|
def compose(self):
|
|
try:
|
|
from importlib.metadata import version
|
|
app_version = version("renamer")
|
|
except Exception:
|
|
app_version = "unknown"
|
|
|
|
help_text = f"""
|
|
Media File Renamer v{app_version}
|
|
|
|
A powerful tool for analyzing and renaming media files with intelligent metadata extraction.
|
|
|
|
NAVIGATION:
|
|
• Use arrow keys to navigate the file tree
|
|
• Right arrow: Expand directory
|
|
• Left arrow: Collapse directory
|
|
• Enter/Space: Select file
|
|
|
|
ACTIONS:
|
|
• o: Open directory - Change the scan directory
|
|
• s: Scan - Refresh the current directory
|
|
• f: Refresh - Reload metadata for selected file
|
|
• r: Rename - Rename selected file with proposed name
|
|
• p: Expand/Collapse - Toggle expansion of selected directory
|
|
• m: Toggle Mode - Switch between technical and catalog display modes
|
|
• ctrl+s: Settings - Open settings window
|
|
• h: Help - Show this help screen
|
|
• q: Quit - Exit the application
|
|
|
|
FEATURES:
|
|
• Automatic metadata extraction from filenames
|
|
• MediaInfo integration for technical details
|
|
• Intelligent title, year, and format detection
|
|
• Support for special editions and collections
|
|
• Real-time file analysis and renaming suggestions
|
|
|
|
FILE ANALYSIS:
|
|
The app extracts various metadata including:
|
|
• Movie/series titles and years
|
|
• Video resolution and frame rates
|
|
• Audio languages and formats
|
|
• Special edition information
|
|
• Collection order numbers
|
|
• HDR and source information
|
|
|
|
Press any key to close this help screen.
|
|
""".strip()
|
|
|
|
with Vertical():
|
|
yield Static(help_text, id="help_content")
|
|
yield Button("Close", id="close")
|
|
|
|
def on_button_pressed(self, event):
|
|
if event.button.id == "close":
|
|
self.app.pop_screen()
|
|
|
|
def on_key(self, event):
|
|
# Close on any key press
|
|
self.app.pop_screen()
|
|
|
|
|
|
class RenameConfirmScreen(Screen):
|
|
CSS = """
|
|
#confirm_content {
|
|
text-align: center;
|
|
}
|
|
# Button {
|
|
# background: $surface;
|
|
# border: solid $surface;
|
|
# }
|
|
Button:focus {
|
|
background: $primary;
|
|
# color: $text-primary;
|
|
# border: solid $primary;
|
|
}
|
|
#buttons {
|
|
align: center middle;
|
|
}
|
|
#new_name_input {
|
|
width: 100%;
|
|
margin: 1 0;
|
|
}
|
|
#new_name_display {
|
|
text-align: center;
|
|
margin-bottom: 1;
|
|
}
|
|
#warning_content {
|
|
text-align: center;
|
|
margin-bottom: 0;
|
|
}
|
|
"""
|
|
|
|
def __init__(self, old_path: Path, new_name: str):
|
|
super().__init__()
|
|
self.old_path = old_path
|
|
self.new_name = new_name
|
|
self.new_path = old_path.parent / new_name
|
|
self.was_edited = False
|
|
|
|
def compose(self):
|
|
from .formatters.text_formatter import TextFormatter
|
|
|
|
confirm_text = f"""
|
|
{TextFormatter.bold(TextFormatter.red("RENAME CONFIRMATION"))}
|
|
|
|
RAW name: {escape(self.old_path.name)}
|
|
|
|
Current name: {TextFormatter.cyan(escape(self.old_path.name))}
|
|
Proposed name: {TextFormatter.green(escape(self.new_name))}
|
|
|
|
{TextFormatter.yellow("Edit the new name below:")}
|
|
""".strip()
|
|
|
|
warning_text = f"""
|
|
{TextFormatter.bold(TextFormatter.red("This action cannot be undone!"))}
|
|
Do you want to proceed with renaming?
|
|
""".strip()
|
|
|
|
with Center():
|
|
with Vertical():
|
|
yield Static(confirm_text, id="confirm_content", markup=True)
|
|
yield Input(value=self.new_name, id="new_name_input", placeholder="New file name")
|
|
yield Static(f"{TextFormatter.green(escape(self.new_name))}", id="new_name_display", markup=True)
|
|
yield Static(warning_text, id="warning_content", markup=True)
|
|
with Horizontal(id="buttons"):
|
|
yield Button("Rename (y)", id="rename")
|
|
yield Button("Cancel (n)", id="cancel")
|
|
|
|
def on_mount(self):
|
|
self.set_focus(self.query_one("#rename"))
|
|
|
|
def on_input_changed(self, event):
|
|
if event.input.id == "new_name_input":
|
|
self.new_name = event.input.value.replace("/", "-").replace("\\", "-")
|
|
self.new_path = self.old_path.parent / self.new_name
|
|
self.was_edited = True
|
|
# Update the display
|
|
from .formatters.text_formatter import TextFormatter
|
|
display = self.query_one("#new_name_display", Static)
|
|
display.update(f"{TextFormatter.green(escape(self.new_name))}")
|
|
|
|
def on_button_pressed(self, event):
|
|
if event.button.id == "rename":
|
|
try:
|
|
logging.info(f"Renaming {self.old_path} to {self.new_path}")
|
|
self.old_path.rename(self.new_path)
|
|
# Update the tree node
|
|
self.app.update_renamed_file(self.old_path, self.new_path) # type: ignore
|
|
self.app.pop_screen()
|
|
except Exception as e:
|
|
# Show error
|
|
content = self.query_one("#confirm_content", Static)
|
|
content.update(f"Error renaming file: {str(e)}")
|
|
elif event.button.id == "cancel":
|
|
self.app.pop_screen()
|
|
|
|
def on_key(self, event):
|
|
current = self.focused
|
|
if current and hasattr(current, 'id'):
|
|
if current.id == "new_name_input":
|
|
# When input is focused, let left/right move cursor, use up/down to change focus
|
|
if event.key == "up":
|
|
self.set_focus(self.query_one("#cancel"))
|
|
elif event.key == "down":
|
|
self.set_focus(self.query_one("#rename"))
|
|
elif current.id in ("rename", "cancel"):
|
|
if event.key == "left":
|
|
if current.id == "rename":
|
|
self.set_focus(self.query_one("#new_name_input"))
|
|
elif current.id == "cancel":
|
|
self.set_focus(self.query_one("#rename"))
|
|
elif event.key == "right":
|
|
if current.id == "new_name_input":
|
|
self.set_focus(self.query_one("#rename"))
|
|
elif current.id == "rename":
|
|
self.set_focus(self.query_one("#cancel"))
|
|
elif event.key == "up":
|
|
if current.id == "rename":
|
|
self.set_focus(self.query_one("#new_name_input"))
|
|
elif current.id == "cancel":
|
|
self.set_focus(self.query_one("#rename"))
|
|
elif event.key == "down":
|
|
if current.id == "new_name_input":
|
|
self.set_focus(self.query_one("#rename"))
|
|
elif current.id == "rename":
|
|
self.set_focus(self.query_one("#cancel"))
|
|
elif current.id == "cancel":
|
|
self.set_focus(self.query_one("#new_name_input"))
|
|
|
|
# Hotkeys only work when not focused on input
|
|
if not current or not hasattr(current, 'id') or current.id != "new_name_input":
|
|
if event.key == "y":
|
|
# Trigger rename
|
|
try:
|
|
logging.info(f"Hotkey renaming {self.old_path} to {self.new_path}")
|
|
self.old_path.rename(self.new_path)
|
|
# Update the tree node
|
|
self.app.update_renamed_file(self.old_path, self.new_path) # type: ignore
|
|
self.app.pop_screen()
|
|
except Exception as e:
|
|
# Show error
|
|
content = self.query_one("#confirm_content", Static)
|
|
content.update(f"Error renaming file: {str(e)}")
|
|
elif event.key == "n":
|
|
# Cancel
|
|
self.app.pop_screen()
|
|
|
|
|
|
class SettingsScreen(Screen):
|
|
CSS = """
|
|
#settings_content {
|
|
text-align: center;
|
|
}
|
|
Button:focus {
|
|
background: $primary;
|
|
}
|
|
#buttons {
|
|
align: center middle;
|
|
}
|
|
.input_field {
|
|
width: 100%;
|
|
margin: 1 0;
|
|
}
|
|
.label {
|
|
text-align: left;
|
|
margin-bottom: 0;
|
|
}
|
|
"""
|
|
|
|
def compose(self):
|
|
from .formatters.text_formatter import TextFormatter
|
|
|
|
settings = self.app.settings # type: ignore
|
|
|
|
content = f"""
|
|
{TextFormatter.bold("SETTINGS")}
|
|
|
|
Configure application settings.
|
|
""".strip()
|
|
|
|
with Center():
|
|
with Vertical():
|
|
yield Static(content, id="settings_content", markup=True)
|
|
|
|
# Mode selection
|
|
yield Static("Display Mode:", classes="label")
|
|
with Horizontal():
|
|
yield Button("Technical", id="mode_technical", variant="primary" if settings.get("mode") == "technical" else "default")
|
|
yield Button("Catalog", id="mode_catalog", variant="primary" if settings.get("mode") == "catalog" else "default")
|
|
|
|
# TTL inputs
|
|
yield Static("Cache TTL - Extractors (hours):", classes="label")
|
|
yield Input(value=str(settings.get("cache_ttl_extractors") // 3600), id="ttl_extractors", classes="input_field")
|
|
|
|
yield Static("Cache TTL - TMDB (hours):", classes="label")
|
|
yield Input(value=str(settings.get("cache_ttl_tmdb") // 3600), id="ttl_tmdb", classes="input_field")
|
|
|
|
yield Static("Cache TTL - Posters (days):", classes="label")
|
|
yield Input(value=str(settings.get("cache_ttl_posters") // 86400), id="ttl_posters", classes="input_field")
|
|
|
|
with Horizontal(id="buttons"):
|
|
yield Button("Save", id="save")
|
|
yield Button("Cancel", id="cancel")
|
|
|
|
def on_button_pressed(self, event):
|
|
if event.button.id == "save":
|
|
self.save_settings()
|
|
self.app.pop_screen() # type: ignore
|
|
elif event.button.id == "cancel":
|
|
self.app.pop_screen() # type: ignore
|
|
elif event.button.id.startswith("mode_"):
|
|
# Toggle mode buttons
|
|
mode = event.button.id.split("_")[1]
|
|
self.app.settings.set("mode", mode) # type: ignore
|
|
# Update button variants
|
|
tech_btn = self.query_one("#mode_technical", Button)
|
|
cat_btn = self.query_one("#mode_catalog", Button)
|
|
tech_btn.variant = "primary" if mode == "technical" else "default"
|
|
cat_btn.variant = "primary" if mode == "catalog" else "default"
|
|
|
|
def save_settings(self):
|
|
try:
|
|
# Get values and convert to seconds
|
|
ttl_extractors = int(self.query_one("#ttl_extractors", Input).value) * 3600
|
|
ttl_tmdb = int(self.query_one("#ttl_tmdb", Input).value) * 3600
|
|
ttl_posters = int(self.query_one("#ttl_posters", Input).value) * 86400
|
|
|
|
self.app.settings.set("cache_ttl_extractors", ttl_extractors) # type: ignore
|
|
self.app.settings.set("cache_ttl_tmdb", ttl_tmdb) # type: ignore
|
|
self.app.settings.set("cache_ttl_posters", ttl_posters) # type: ignore
|
|
|
|
self.app.notify("Settings saved!", severity="information", timeout=2) # type: ignore
|
|
except ValueError:
|
|
self.app.notify("Invalid TTL values. Please enter numbers only.", severity="error", timeout=3) # type: ignore |