move ask_to_proceed() to interaction_helper.py

Author: ashariyar

pdfalyzer/__init__.py

@@ -14,11 +14,12 @@
 from pdfalyzer.decorators.pdf_file import PdfFile
 from pdfalyzer.output.pdfalyzer_presenter import PdfalyzerPresenter
 from pdfalyzer.pdfalyzer import Pdfalyzer
-from pdfalyzer.util.argument_parser import ask_to_proceed, parser
+from pdfalyzer.util.argument_parser import parser
 from pdfalyzer.util.cli_tools_argument_parsers import (MAX_QUALITY, parse_combine_pdfs_args,
      parse_pdf_page_extraction_args, parse_text_extraction_args)
 from pdfalyzer.util.exceptions import PdfParserError
 from pdfalyzer.util.helpers.filesystem_helper import file_size_in_mb, set_max_open_files
+from pdfalyzer.util.helpers.interaction_helper import ask_to_proceed
 from pdfalyzer.util.logging import log  # Trigger log setup
 from pdfalyzer.util.output_section import OutputSection
 from pdfalyzer.util.pdf_parser_manager import PdfParserManager

pdfalyzer/util/argument_parser.py

@@ -3,7 +3,6 @@
 """
 import sys
 from argparse import ArgumentParser
-from typing import Type
 
 from rich_argparse_plus import RichHelpFormatterPlus
 from rich.prompt import Confirm
@@ -137,16 +136,3 @@
 # Make sure the selection section is at the top
 parser._action_groups = parser._action_groups[:2] + [parser._action_groups[-1]] + parser._action_groups[2:-1]
 is_pdfalyze_script = parser.prog.startswith(PDFALYZE)  # startswith() bc on Windows we end up with 'pdfalyze.cmd'?
-
-
-#############
-#  Helpers  #
-#############
-
-def ask_to_proceed(msg: str | Text | None = None) -> None:
-    """Exit if user doesn't confirm they want to proceed."""
-    msg = msg if isinstance(msg, Text) else Text(msg or "Proceed anyway?")
-
-    if not Confirm.ask(msg):
-        log_console.print('Exiting...', style='dim')
-        sys.exit()

pdfalyzer/util/cli_tools_argument_parsers.py

@@ -20,7 +20,7 @@
 
 from pdfalyzer.util.helpers.filesystem_helper import (do_all_files_exist, extract_page_number, is_pdf,
      with_pdf_extension)
-from pdfalyzer.util.argument_parser import ask_to_proceed
+from pdfalyzer.util.helpers.interaction_helper import ask_to_proceed
 from pdfalyzer.util.page_range import PageRangeArgumentValidator
 
 MAX_QUALITY = 10

pdfalyzer/util/helpers/interaction_helper.py

@@ -0,0 +1,15 @@
+"""Helpers for dealing with user interaction."""
+import sys
+
+from rich.prompt import Confirm
+from rich.text import Text
+from yaralyzer.util.logging import log_console
+
+
+def ask_to_proceed(msg: str | Text | None = None) -> None:
+    """Exit if user doesn't confirm they want to proceed."""
+    msg = msg if isinstance(msg, Text) else Text(msg or "Proceed anyway?")
+
+    if not Confirm.ask(msg):
+        log_console.print('Exiting...', style='dim')
+        sys.exit()