Make PDFBaker options its own (data)class

Author: dannyadair

src/pdfbaker/__main__.py

@@ -7,7 +7,7 @@
 import click
 
 from pdfbaker import __version__
-from pdfbaker.baker import PDFBaker
+from pdfbaker.baker import PDFBaker, PDFBakerOptions
 from pdfbaker.errors import PDFBakerError
 
 logger = logging.getLogger(__name__)
@@ -36,6 +36,7 @@ def cli() -> None:
 @click.option(
     "--debug", is_flag=True, help="Debug mode (implies --verbose and --keep-build)"
 )
+# pylint: disable=too-many-arguments,too-many-positional-arguments
 def bake(
     config_file: Path,
     quiet: bool,
@@ -50,13 +51,13 @@ def bake(
         keep_build = True
 
     try:
-        baker = PDFBaker(
-            config_file,
+        options = PDFBakerOptions(
             quiet=quiet,
             verbose=verbose,
             trace=trace,
             keep_build=keep_build,
         )
+        baker = PDFBaker(config_file, options=options)
         baker.bake()
         return 0
     except PDFBakerError as exc:

src/pdfbaker/baker.py

@@ -7,6 +7,7 @@
 """
 
 import logging
+from dataclasses import dataclass
 from pathlib import Path
 from typing import Any
 
@@ -15,7 +16,7 @@
 from .errors import ConfigurationError
 from .logging import TRACE, LoggingMixin
 
-__all__ = ["PDFBaker"]
+__all__ = ["PDFBaker", "PDFBakerOptions"]
 
 
 DEFAULT_CONFIG = {
@@ -28,6 +29,23 @@
 }
 
 
+@dataclass
+class PDFBakerOptions:
+    """Options for controlling PDFBaker behavior.
+
+    Attributes:
+        quiet: Show errors only
+        verbose: Show debug information
+        trace: Show trace information (even more detailed than debug)
+        keep_build: Keep build artifacts after processing
+    """
+
+    quiet: bool = False
+    verbose: bool = False
+    trace: bool = False
+    keep_build: bool = False
+
+
 class PDFBaker(LoggingMixin):
     """Main class for PDF document generation."""
 
@@ -53,31 +71,27 @@ def __init__(
     def __init__(
         self,
         config_file: Path,
-        quiet: bool = False,
-        verbose: bool = False,
-        trace: bool = False,
-        keep_build: bool = False,
+        options: PDFBakerOptions | None = None,
     ) -> None:
         """Initialize PDFBaker with config file path. Set logging level.
 
         Args:
-            config_file: Path to config file, document directory is its parent
-            quiet: Show errors only
-            verbose: Show debug information
-            trace: Show trace information (even more detailed than debug)
-            keep_build: Keep build artifacts
+            config_file: Path to config file
+            options: Optional options for logging and build behavior
         """
         super().__init__()
+        options = options or PDFBakerOptions()
+
         logging.basicConfig(level=logging.INFO, format="%(levelname)s: %(message)s")
-        if quiet:
+        if options.quiet:
             logging.getLogger().setLevel(logging.ERROR)
-        elif trace:
+        elif options.trace:
             logging.getLogger().setLevel(TRACE)
-        elif verbose:
+        elif options.verbose:
             logging.getLogger().setLevel(logging.DEBUG)
         else:
             logging.getLogger().setLevel(logging.INFO)
-        self.keep_build = keep_build
+        self.keep_build = options.keep_build
         self.config = self.Configuration(
             baker=self,
             base_config=DEFAULT_CONFIG,

src/pdfbaker/logging.py

@@ -4,7 +4,8 @@
 from typing import Any
 
 TRACE = 5
-logging.addLevelName(TRACE, 'TRACE')
+logging.addLevelName(TRACE, "TRACE")
+
 
 class LoggingMixin:
     """Mixin providing consistent logging functionality across pdfbaker classes."""

tests/test_baker.py

@@ -3,7 +3,7 @@
 import shutil
 from pathlib import Path
 
-from pdfbaker.baker import PDFBaker
+from pdfbaker.baker import PDFBaker, PDFBakerOptions
 
 
 def test_examples() -> None:
@@ -31,5 +31,6 @@ def test_examples() -> None:
         f.write(content)
 
     # Run baker
-    baker = PDFBaker(test_config, quiet=True, keep_build=True)
+    options = PDFBakerOptions(quiet=True, keep_build=True)
+    baker = PDFBaker(test_config, options=options)
     baker.bake()