twd-m4sc0 3.0.4__tar.gz → 3.1.1__tar.gz

{twd_m4sc0-3.0.4 → twd_m4sc0-3.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: twd-m4sc0
-Version: 3.0.4
+Version: 3.1.1
 Summary: TWD by m4sc0
 Author: m4sc0
 License: MIT

{twd_m4sc0-3.0.4 → twd_m4sc0-3.1.1}/pyproject.toml

@@ -1,6 +1,7 @@
 [project]
 name = "twd-m4sc0"
-version = "3.0.4"
+# version = "3.1.0"
+dynamic = ["version"]
 description = "TWD by m4sc0"
 authors = [{name = "m4sc0"}]
 requires-python = ">=3.13"
@@ -32,3 +33,6 @@ where = ["src"]
 
 [tool.setuptools.package-data]
 twd = ["*.tcss"]
+
+[tool.setuptools.dynamic]
+version = {attr = "twd.__version__"}

twd_m4sc0-3.1.1/src/twd/__init__.py

@@ -0,0 +1 @@
+__version__ = "3.1.1"

{twd_m4sc0-3.0.4 → twd_m4sc0-3.1.1}/src/twd/cli.py

@@ -8,7 +8,7 @@ from .tui import TWDApp
 
 @click.group(invoke_without_command=True)
 @click.pass_context
-@click.version_option(version=__version__)
+@click.version_option(version=__version__, prog_name="twd")
 def cli(ctx):
     """
     TWD - Temp / Tracked Working Directory
@@ -39,7 +39,18 @@ def cli(ctx):
 @click.pass_context
 def save(ctx, path, alias, name):
     """
-    Save a new twd with PATH, [opt] ALIAS and [opt] NAME
+    Save a new TWD with an Alias (UUID like) and Name
+
+    Examples:
+
+        twd save ./Movies movies Media
+
+        twd save / root "Root Directory -- /dev/sda3"
+
+        twd save . finances-q1 "Finance Documents Q1"
+
+    The naming conventions for the alias align with kebab-casing.
+    There are no naming conventions for the name.
     """
     manager: TwdManager = ctx.obj['manager']
 
@@ -61,7 +72,13 @@ def save(ctx, path, alias, name):
 @click.pass_context
 def get(ctx, alias):
     """
-    get path for ALIAS
+    cd into a directory by the given Alias
+
+    Examples:
+
+        twd get movies
+
+    There's really not much to it.
     """
     manager = ctx.obj['manager']
 
@@ -81,7 +98,13 @@ def get(ctx, alias):
 @click.pass_context
 def remove(ctx, alias):
     """
-    remove TWD by ALIAS
+    Remove TWD by the given Alias
+
+    Examples:
+
+        twd remove movies
+
+    That's it.
     """
     manager = ctx.obj['manager']
     
@@ -95,7 +118,9 @@ def remove(ctx, alias):
 @cli.command('list')
 @click.pass_context
 def list_twds(ctx):
-    """List all TWDs"""
+    """
+    List all TWDs
+    """
     manager = ctx.obj['manager']
     
     entries = manager.list_all()
@@ -103,5 +128,65 @@ def list_twds(ctx):
         click.echo("No TWDs saved yet. Use 'twd save <path> <alias>' to add one.")
         return
     
+    invalid_found = False
     for entry in entries:
+        if not invalid_found and not os.path.exists(entry.path):
+            invalid_found = True
         click.echo(f"{entry.alias:20} {entry.name:30} {entry.path}")
+
+    if invalid_found:
+        click.echo(f"\nInvalid TWD paths found. Run <twd clean> to remove them.")
+
+
+@cli.command('clean')
+@click.option('--yes', '-y', is_flag=True, help="Remove all invalid entries without asking")
+@click.pass_context
+def clean(ctx, yes):
+    """
+    Clean the records of rogue TWDs
+
+    Sometimes it's possible that TWDs are not properly cleaned up.
+    That's what this subcommand does.
+    """
+    manager = ctx.obj['manager']
+
+    # all entries, valid and invalid
+    entries = manager.list_all()
+
+    # only invalids
+    invalids = []
+    for entry in entries:
+        if os.path.exists(entry.path):
+            continue
+
+        invalids.append(entry)
+
+    if len(invalids) == 0:
+        click.echo("No invalid TWDs found.")
+        return
+
+    click.echo(f"Found {len(invalids)} invalid TWDs\n")
+
+    # only valid
+    valid_entries = []
+    for entry in entries:
+        if entry not in invalids:
+            valid_entries.append(entry)
+
+    # remove all
+    if yes:
+        click.echo("Removing all invalid entries...")
+        manager._write_all(valid_entries)
+        click.echo(f"Done. {len(valid_entries)} TWDs left.")
+        return
+
+    to_keep = []
+    for inv in invalids:
+        if not click.confirm(f"Do you want to remove '{inv.alias}'?", default=True):
+            # user wants to keep invalid TWD (weird but i'll allow it)
+            to_keep.append(inv)
+
+    # write a unison list of valid entries and the ones to keep
+    final_entries = valid_entries + to_keep
+    manager._write_all(final_entries)
+    click.echo(f"Done. {len(final_entries)} TWDs left.")

{twd_m4sc0-3.0.4 → twd_m4sc0-3.1.1}/src/twd/data.py

@@ -15,6 +15,17 @@ class Entry(BaseModel):
     name: str = Field(..., min_length=3)
     created_at: datetime = Field(default_factory=datetime.now)
 
+    def __eq__(self, other) -> bool:
+        """Compare entries based on their values"""
+        if not isinstance(other, Entry):
+            return NotImplemented
+
+        return (
+                self.alias == other.alias
+                and self.path == other.path
+                and self.name == other.name
+                )
+
     @validator("alias")
     def validate_alias(cls, v):
         if not v.replace("_", "").replace("-", "").isalnum():
@@ -26,10 +37,6 @@ class Entry(BaseModel):
     def validate_path(cls, v):
         path = Path(v).expanduser().resolve()
 
-        if not path.exists():
-            raise ValueError(f"Path does not exist: {path}")
-        if not path.is_dir():
-            raise ValueError(f"Path is not a directory: {path}")
         return path
 
     def to_csv(self) -> List[str]:
@@ -51,10 +58,21 @@ class Entry(BaseModel):
                 created_at=datetime.fromisoformat(row[3])
                 )
 
+    @classmethod
+    def from_values(cls, alias, path, name, created_at) -> "Entry":
+        """create from values"""
+        return cls(
+                alias=alias,
+                path=path,
+                name=name,
+                created_at=created_at
+                )
+
 class TwdManager:
     """twd entry manager stored in csv"""
 
     CSV_HEADERS = ["alias", "path", "name", "created_at"]
+    CSV_HEADERS_FANCY = ["Alias", "Path", "Description", "Created at"]
 
     def __init__(self, csv_path: Path):
         self.csv_path = csv_path
@@ -80,10 +98,7 @@ class TwdManager:
             next(reader) # skip headers
 
             for row in reader:
-                try:
-                    entries.append(Entry.from_csv(row))
-                except Exception as e:
-                    print(f"Warning: skipping invalid row: {e}")
+                entries.append(Entry.from_csv(row))
 
         return entries
 
@@ -121,6 +136,16 @@ class TwdManager:
 
         return None
 
+    def update(self, alias: str, entry: Entry) -> bool:
+        """update TWD by alias"""
+        if not self.exists(alias):
+            return False
+
+        # simplest form of update is remove and add
+        self.remove(alias)
+
+        self.add(entry.alias, entry.path, entry.name)
+
     def remove(self, alias: str) -> None:
         """remove entry by alias"""
         entries = self._read_all()

twd_m4sc0-3.1.1/src/twd/modals/__init__.py

@@ -0,0 +1,4 @@
+from twd.modals.confirm import ConfirmModal, EntryDeleteModal
+from twd.modals.edit import EditEntryModal
+
+__all__ = ["ConfirmModal", "EntryDeleteModal", "EditEntryModal"]

twd_m4sc0-3.1.1/src/twd/modals/confirm.py

@@ -0,0 +1,97 @@
+from textual import on
+from textual.app import ComposeResult
+from textual.screen import ModalScreen
+from textual.containers import Container, Horizontal
+from textual.widgets import Button, Label
+from typing import Union, TypeVar, Generic
+
+from twd.data import Entry
+
+T = TypeVar('T')
+
+class ConfirmModal(ModalScreen[T], Generic[T]):
+    """A confirm modal"""
+
+    DEFAULT_CSS = """
+        ConfirmModal {
+            align: center middle;
+        }
+
+        ConfirmModal > Container {
+            width: auto;
+            height: auto;
+            border: thick $background 80%;
+            background: $surface;
+        }
+
+        ConfirmModal > Container > Label {
+            width: 100%;
+            content-align-horizontal: center;
+            margin-top: 1;
+        }
+
+        ConfirmModal > Container > Horizontal {
+            width: auto;
+            height: auto;
+        }
+
+        ConfirmModal > Container > Horizontal > Button {
+            margin: 2 4;
+        }
+    """
+
+    def __init__(
+            self, 
+            message: Union[str, None] = None,
+            confirm_text: str = "Yes",
+            cancel_text: str = "No",
+            confirm_value: T | None = None,
+            cancel_value: T | None = None,
+    ):
+        """
+        message: The message to display when popping the modal
+        confirm_text: Text to show for confirmation
+        cancel_text: Text to show for cancellation
+        """
+        self.message = message
+        self.confirm_text = confirm_text
+        self.cancel_text = cancel_text
+
+        self.confirm_value = confirm_value
+        self.cancel_value = cancel_value
+
+        super().__init__()
+
+    def compose_content(self) -> ComposeResult:
+        """
+        Abstract method for presenting custom shenanigans
+        """
+        if self.message:
+            yield Label(self.message, id="content")
+        else:
+            yield Label("Are you sure?", id="content")
+
+    def compose(self) -> ComposeResult:
+        with Container():
+            yield from self.compose_content()
+            with Horizontal():
+                yield Button(self.cancel_text, id="cancel", variant="error")
+                yield Button(self.confirm_text, id="confirm", variant="success")
+
+    @on(Button.Pressed, "#cancel")
+    def cancel_pressed(self) -> None:
+        self.dismiss(self.cancel_value)
+
+    @on(Button.Pressed, "#confirm")
+    def confirm_pressed(self) -> None:
+        self.dismiss(self.confirm_value)
+
+class EntryDeleteModal(ConfirmModal):
+    """Confirmation modal with detailed entry information"""
+
+    def __init__(self, entry):
+        self.entry = entry
+        super().__init__(confirm_text="Delete", cancel_text="Cancel")
+
+    def compose_content(self) -> ComposeResult:
+        yield Label(f"Delete entry '{self.entry.name}'?", id="content")

twd_m4sc0-3.1.1/src/twd/modals/edit.py

@@ -0,0 +1,52 @@
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal
+from textual.widgets import Button, Label, Input
+
+from twd.data import Entry
+from twd.modals import ConfirmModal
+
+class EditEntryModal(ConfirmModal[Entry | None]):
+    """A modal to edit an existing entry"""
+
+    DEFAULT_CSS = """
+        Label {
+            width: 100%;
+            text-align: left;
+        }
+        Input {
+            margin: 1;
+        }
+    """
+
+    def __init__(self, entry):
+        """
+        entry: the entry to edit
+        """
+        self.entry = Entry.from_values(entry.alias, entry.path, entry.name, entry.created_at)
+
+        super().__init__(
+                confirm_text="Save",
+                cancel_text="Discard",
+                confirm_value=self.entry,
+                )
+
+    def compose_content(self) -> ComposeResult:
+        yield Label(f"Edit Entry", id="title")
+
+        yield Label("Alias")
+        yield Input(value=self.entry.alias, id="alias", disabled=True)
+        yield Label("Path")
+        yield Input(value=str(self.entry.path), id="path")
+        yield Label("Name")
+        yield Input(value=self.entry.name, id="name")
+    
+    @on(Input.Changed, "#path")
+    def on_path_change(self, event: Input.Changed) -> None:
+        self.entry.path = event.value
+    
+    @on(Input.Changed, "#name")
+    def on_name_change(self, event: Input.Changed) -> None:
+        self.entry.name = event.value
+
+        

{twd_m4sc0-3.0.4 → twd_m4sc0-3.1.1}/src/twd/tui.py

@@ -1,4 +1,5 @@
 from enum import Enum
+from datetime import datetime
 from textual import on
 from textual.app import App, ComposeResult, Binding
 from textual.containers import HorizontalGroup, VerticalScroll
@@ -7,8 +8,9 @@ from textual.widgets import Button, Digits, Footer, Header, DataTable, Label, Ru
 from textual.color import Color
 
 from twd.config import Config
-from twd.data import TwdManager
+from twd.data import TwdManager, Entry
 from twd.utils import fuzzy_search, linear_search
+from twd.modals import ConfirmModal, EntryDeleteModal, EditEntryModal
 
 class Mode(Enum):
     NORMAL = "normal"
@@ -27,8 +29,10 @@ class TWDApp(App):
             Binding("k", "cursor_up", "Up"),
 
             # modify
-            Binding("/", "enter_search_mode", "Search"),
-            Binding("escape", "enter_normal_mode", "Normal", show=False),
+            Binding("/", "slash_key", "Search"),
+            Binding("d", "d_key", "Delete"),
+            Binding("e", "e_key", "Edit"),
+            Binding("escape", "escape_key", "Normal", show=False),
             # TODO: edit
             # TODO: rename
 
@@ -40,6 +44,7 @@ class TWDApp(App):
         ]
 
     mode: Mode = reactive(Mode.NORMAL)
+    search_results = None
 
     def __init__(self, manager: TwdManager, *args, **kwargs):
         self.manager = manager
@@ -77,7 +82,7 @@ class TWDApp(App):
 
         # add headers
         table = self.query_one(DataTable)
-        table.add_columns(*self.manager.CSV_HEADERS)
+        table.add_columns(*self.manager.CSV_HEADERS_FANCY)
         
         self._populate_table()
 
@@ -93,7 +98,20 @@ class TWDApp(App):
         
         # fill data
         for entry in entries:
-            table.add_row(entry.alias, str(entry.path), entry.name, entry.created_at)
+            table.add_row(entry.alias, str(entry.path), entry.name, entry.created_at.strftime("%Y-%m-%d %H:%M:%S"))
+
+    def _current_row_entry(self) -> Entry:
+        table = self.query_one(DataTable)
+
+        # get row
+        row_key = table.coordinate_to_cell_key(table.cursor_coordinate).row_key
+        row_data = table.get_row(row_key)
+        alias = row_data[0]
+
+        # get entry
+        entry = self.manager.get(alias)
+
+        return entry
 
     def watch_mode(self, old_mode: Mode, new_mode: Mode) -> None:
         """
@@ -128,8 +146,11 @@ class TWDApp(App):
         """
         table = self.query_one(DataTable)
 
-        current_row = table.cursor_coordinate.row
-        next_row = (current_row + 1) % table.row_count
+        try:
+            current_row = table.cursor_coordinate.row
+            next_row = (current_row + 1) % table.row_count
+        except ZeroDivisionError as e:
+            return
 
         table.move_cursor(row=next_row)
 
@@ -139,12 +160,15 @@ class TWDApp(App):
         """
         table = self.query_one(DataTable)
         
-        current_row = table.cursor_coordinate.row
-        prev_row = (current_row - 1) % table.row_count
+        try:
+            current_row = table.cursor_coordinate.row
+            prev_row = (current_row - 1) % table.row_count
+        except ZeroDivisionError as e:
+            return
 
         table.move_cursor(row=prev_row)
 
-    def action_enter_search_mode(self) -> None:
+    def action_slash_key(self) -> None:
         """
         enter search mode
         """
@@ -152,14 +176,62 @@ class TWDApp(App):
             return
         self.mode = Mode.SEARCH
 
-    def action_enter_normal_mode(self) -> None:
+    def action_escape_key(self) -> None:
         """
         enter normal mode
         """
         if self.mode == Mode.NORMAL:
+            if self.search_results is not None:
+                self._populate_table()
+                self.search_results = None
             return
         self.mode = Mode.NORMAL
 
+    def action_d_key(self) -> None:
+        """
+        open confirm modal and delete entry if yes
+        """
+        if not self.mode == Mode.NORMAL:
+            return
+
+        entry = self._current_row_entry()
+
+        def check_delete(decision: bool | None) -> None:
+            """
+            open modal and return the decision
+            """
+            if not decision:
+                return
+
+            self.manager.remove(alias)
+
+            self._populate_table()
+
+            self.notify(f"Removed entry \"{entry.name}\"")
+
+        self.push_screen(EntryDeleteModal(entry), check_delete)
+
+    def action_e_key(self) -> None:
+        """
+        open edit modal and edit entry in place
+        """
+        entry = self._current_row_entry()
+
+        def save_new_entry(new_entry: Entry | None) -> None:
+            if not new_entry or entry == new_entry:
+                # user hit 'Discard'
+                # no changes so no update
+                self.notify(f"No changes")
+                return
+
+            self.notify(f"Updated TWD '{new_entry.alias}'")
+
+            self.manager.update(new_entry.alias, new_entry)
+            self.manager._write_all(self.manager._read_all())
+            self._populate_table()
+
+        self.push_screen(EditEntryModal(entry), save_new_entry)
+
     def action_exit(self) -> None:
         self.exit()
 
@@ -186,6 +258,7 @@ class TWDApp(App):
         filtered = [item[0] for item in search_result]
 
         self._populate_table(filtered)
+        self.search_results = filtered
 
     @on(Input.Submitted, "#search-input")
     def on_search_submitted(self, e: Input.Submitted) -> None:
@@ -196,6 +269,7 @@ class TWDApp(App):
             return
 
         self.mode = Mode.NORMAL
+        self._populate_table(self.search_results)
 
         self.query_one(DataTable).focus()
 
@@ -208,12 +282,7 @@ class TWDApp(App):
         table = event.data_table
         row_key = event.row_key
 
-        # get row
-        row_data = table.get_row(row_key)
-        alias = row_data[0]
-
-        # get entry
-        entry = self.manager.get(alias)
+        entry = self._current_row_entry()
 
         self.notify(f"Selected: {entry.alias} -> {entry.path}")
 

{twd_m4sc0-3.0.4 → twd_m4sc0-3.1.1}/src/twd_m4sc0.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: twd-m4sc0
-Version: 3.0.4
+Version: 3.1.1
 Summary: TWD by m4sc0
 Author: m4sc0
 License: MIT

{twd_m4sc0-3.0.4 → twd_m4sc0-3.1.1}/src/twd_m4sc0.egg-info/SOURCES.txt

@@ -7,6 +7,9 @@ src/twd/data.py
 src/twd/tui.py
 src/twd/tui.tcss
 src/twd/utils.py
+src/twd/modals/__init__.py
+src/twd/modals/confirm.py
+src/twd/modals/edit.py
 src/twd_m4sc0.egg-info/PKG-INFO
 src/twd_m4sc0.egg-info/SOURCES.txt
 src/twd_m4sc0.egg-info/dependency_links.txt

twd_m4sc0-3.0.4/src/twd/__init__.py

@@ -1,6 +0,0 @@
-from importlib.metadata import version, PackageNotFoundError
-
-try:
-    __version__ = version("twd")
-except PackageNotFoundError as e:
-    __version__ = "unknown"