New shared function add_parser_verbosity_group().

Author: samsrabin

relink.py

@@ -12,7 +12,12 @@
 import time
 from pathlib import Path
 
-from shared import DEFAULT_INPUTDATA_ROOT, DEFAULT_STAGING_ROOT, get_log_level
+from shared import (
+    DEFAULT_INPUTDATA_ROOT,
+    DEFAULT_STAGING_ROOT,
+    get_log_level,
+    add_parser_verbosity_group,
+)
 
 # Set up logger
 logger = logging.getLogger(__name__)
@@ -175,7 +180,11 @@ def find_owned_files_scandir(item, user_uid, inputdata_root=DEFAULT_INPUTDATA_RO
 
 
 def replace_files_with_symlinks(
-    item_to_process, target_dir, username, inputdata_root=DEFAULT_INPUTDATA_ROOT, dry_run=False
+    item_to_process,
+    target_dir,
+    username,
+    inputdata_root=DEFAULT_INPUTDATA_ROOT,
+    dry_run=False,
 ):
     """
     Finds files owned by a specific user in a source directory tree,
@@ -357,17 +366,8 @@ def parse_arguments():
         help=argparse.SUPPRESS,
     )
 
-    # Verbosity options (mutually exclusive)
-    verbosity_group = parser.add_mutually_exclusive_group()
-    verbosity_group.add_argument(
-        "-v", "--verbose", action="store_true", help="Enable verbose output"
-    )
-    verbosity_group.add_argument(
-        "-q",
-        "--quiet",
-        action="store_true",
-        help="Quiet mode (show only warnings and errors)",
-    )
+    # Add verbosity options
+    add_parser_verbosity_group(parser)
 
     parser.add_argument(
         "--dry-run",

rimport

@@ -89,17 +89,8 @@ def build_parser() -> argparse.ArgumentParser:
         help="Check whether file(s) is/are already published.",
     )
 
-    # Verbosity options (mutually exclusive)
-    verbosity_group = parser.add_mutually_exclusive_group()
-    verbosity_group.add_argument(
-        "-v", "--verbose", action="store_true", help="Enable verbose output (DEBUG level)"
-    )
-    verbosity_group.add_argument(
-        "-q",
-        "--quiet",
-        action="store_true",
-        help="Quiet mode (show only warnings and errors)",
-    )
+    # Add verbosity options
+    shared.add_parser_verbosity_group(parser)
 
     # Provide -help to mirror legacy behavior (in addition to -h and --help)
     parser.add_argument(

shared.py

@@ -3,6 +3,7 @@
 """
 
 import logging
+from argparse import ArgumentParser
 
 DEFAULT_INPUTDATA_ROOT = "/glade/campaign/cesm/cesmdata/cseg/inputdata/"
 DEFAULT_STAGING_ROOT = (
@@ -28,3 +29,27 @@ def get_log_level(quiet: bool = False, verbose: bool = False) -> int:
     if verbose:
         return logging.DEBUG
     return logging.INFO
+
+
+def add_parser_verbosity_group(parser: ArgumentParser):
+    """Add mutually exclusive verbosity options to an argument parser.
+
+    Adds -v/--verbose and -q/--quiet flags as a mutually exclusive group.
+
+    Args:
+        parser: ArgumentParser instance to add the verbosity group to.
+
+    Returns:
+        The mutually exclusive argument group that was created.
+    """
+    verbosity_group = parser.add_mutually_exclusive_group()
+    verbosity_group.add_argument(
+        "-v", "--verbose", action="store_true", help="Enable verbose output (DEBUG level)"
+    )
+    verbosity_group.add_argument(
+        "-q",
+        "--quiet",
+        action="store_true",
+        help="Quiet mode (show only warnings and errors)",
+    )
+    return verbosity_group