Fix text to remove user input

src/maildirclean/cli.py

@@ -2,9 +2,23 @@ import argparse
 import sys
 from pathlib import Path
 
+from maildirclean.filedir import delete_all_from_selected_email
+
 from .maildir import MailDir, TopSender, parse_maildir
 
 
+class InputError(Exception):
+    """Exception raised for custom error scenarios.
+
+    Attributes:
+        message -- explanation of the error
+    """
+
+    def __init__(self):
+        self.message = "Please enter a valid number or 'q' to quit"
+        super().__init__(self.message)
+
+
 def parse_arguments() -> argparse.Namespace:
     """Parse command line arguments
 
@@ -24,6 +38,15 @@ def parse_arguments() -> argparse.Namespace:
         default=5,
         help="Number of top senders to display (default: 5)",
     )
+
+    parser.add_argument(
+        "--no-confirm",
+        dest="confirm",
+        action="store_false",
+        default=True,
+        help="Skip confirmation prompts",
+    )
+
     parser.add_argument(
         "--verbose", "-v", action="store_true", help="Enable verbose output"
     )
@@ -78,13 +101,18 @@ def run_loop(args: argparse.Namespace, maildir_path: str | Path):
         )
 
         print(output)
-        if not user_input_loop(top_senders, maildir):
+        if not user_input_loop(top_senders, maildir, args.confirm):
             break
 
 
-def user_input_loop(top_senders: list[TopSender], maildir: MailDir) -> bool:
+def user_input_loop(top_senders: list[TopSender], maildir: MailDir, confirm) -> bool:
     user_input = input("> ").strip()
-    while handle_user_input(user_input, top_senders, maildir):
+
+    while True:
+        try:
+            return handle_user_input(user_input, top_senders, maildir, confirm)
+        except InputError as ex:
+            print(ex.message)
             user_input = input("> ").strip()
 
 
@@ -146,15 +174,14 @@ def parse_selections(user_input, max_selection):
     return selections
 
 
-def handle_user_input(user_input, top_senders, maildir):
+def handle_user_input(user_input, top_senders, maildir, confirm) -> bool:
     if user_input.lower() == "q":
         return False
 
     try:
         selections = parse_selections(user_input, len(top_senders))
-        for selection in selections:
-            selected_sender = top_senders[selection - 1]
-            print(f"Selected {selected_sender.email}")
+        selected_senders = [top_senders[selection - 1] for selection in selections]
+        delete_all_from_selected_email(selected_senders, maildir, confirm)
         return True
     except ValueError:
-        print("Please enter a valid number or 'q' to quit")
+        raise InputError

src/maildirclean/filedir.py

@@ -5,9 +5,19 @@ import os
 from maildirclean.maildir import MailDir, TopSender
 
 
-def delete_all_from_selected_email(selected_senders: list[TopSender], maildir: MailDir):
+def delete_all_from_selected_email(
+    selected_senders: list[TopSender], maildir: MailDir, confirm=True
+):
     for sender in selected_senders:
-        delete_files(maildir.get_paths_for_email(sender.email))
+        emails = maildir.get_paths_for_email(sender.email)
+        if confirm:
+            print(
+                f"Deleting {len(emails)} addressed from sender {sender.email}\nConfirm (y)"
+            )
+            if input("> ").strip() not in ["y", "Y", "yes", "Yes"]:
+                print(f"Cancelling delete")
+                continue
+        delete_files(emails)
         maildir.remove_email(sender.email)
 
 

src/maildirclean/maildir.py

@@ -121,7 +121,7 @@ class MailDir:
         return senders
 
     def get_paths_for_email(self, email: str) -> list[str]:
-        return self._df.loc[self._df == email, "path"].to_list()
+        return self._df.loc[self._df.email == email, "path"].to_list()
 
     def remove_email(self, email: str):
         self._df.drop(self._df[self._df["email"] == email].index, inplace=True)

tests/test_filedir.py

@@ -1,5 +1,20 @@
 from fixtures import *
-from maildirclean.filedir import delete_files
+from maildirclean.filedir import delete_all_from_selected_email, delete_files
+from maildirclean.maildir import TopSender, parse_maildir
+
+
+def test_deleted_from_selected_email(sample_email_dir):
+    maildir = parse_maildir(sample_email_dir)
+    delete_all_from_selected_email(
+        [TopSender("test@something.org", [""], 2)], maildir, confirm=False
+    )
+
+    assert len(maildir._df) == 1
+    assert maildir._df.iloc[0].email == "not_a_test@something.org"
+
+    files = list(Path(sample_email_dir).glob("*"))
+    assert len(files) == 1
+    assert files[0] == Path(sample_email_dir) / "1"
 
 
 def test_delete_files(sample_email_dir):