Add proper bates numbers support to ALExhibitDocument

Passes along all of the bates parameters through all of the exhibit
document classes (ALExhibitDocument, ALExhibitList, ALExhibit).

Author: BryceStevenWilley

docassemble/AssemblyLine/al_document.py

@@ -2768,9 +2768,14 @@ def as_pdf(
         self,
         *,
         refresh: bool = False,
-        prefix: str = "",
         pdfa: bool = False,
         add_page_numbers: bool = True,
+        page_number_prefix: str = "",
+        page_number_digits: int = 5,
+        page_number_area = None,
+        page_number_font_size: float = 10,
+        page_number_offset_horizontal: float = 15,
+        page_number_offset_vertical: float = 15,
         add_cover_page: bool = True,
         filename: Optional[str] = None,
         append_matching_suffix: bool = True,
@@ -2812,7 +2817,15 @@ def as_pdf(
             )
 
         if add_page_numbers:
-            concatenated_pages.bates_number(prefix=prefix, start=self.start_page)
+            concatenated_pages.bates_number(
+                start=self.start_page,
+                prefix=page_number_prefix,
+                digits=page_number_digits,
+                area=page_number_area,
+                font_size=page_number_font_size,
+                offset_horizontal=page_number_offset_horizontal,
+                offset_vertical=page_number_offset_vertical
+            )
 
         setattr(self._cache, safe_key, concatenated_pages)
         return getattr(self._cache, safe_key)
@@ -2946,6 +2959,12 @@ def as_pdf(
         filename="file.pdf",
         pdfa: bool = False,
         add_page_numbers: bool = False,
+        page_number_prefix: str = "",
+        page_number_digits: int = 5,
+        page_number_area = None,
+        page_number_font_size: float = 10,
+        page_number_offset_horizontal: float = 15,
+        page_number_offset_vertical: float = 15,
         toc_pages: int = 0,
         append_matching_suffix: bool = True,
     ) -> DAFile:
@@ -2967,12 +2986,19 @@ def as_pdf(
                 exhibit.cover_page
         if self.include_table_of_contents and toc_pages != 1:
             self._update_page_numbers(toc_guess_pages=toc_pages)
+        if not page_number_prefix and self.bates_prefix:
+            page_number_prefix = self.bates_prefix
         return pdf_concatenate(
             [
                 exhibit.as_pdf(
                     add_cover_page=self.include_exhibit_cover_pages,
                     add_page_numbers=add_page_numbers,
-                    prefix=self.bates_prefix,
+                    page_number_prefix=page_number_prefix,
+                    page_number_digits=page_number_digits,
+                    page_number_area=page_number_area,
+                    page_number_font_size=page_number_font_size,
+                    page_number_offset_horizontal=page_number_offset_horizontal,
+                    page_number_offset_vertical=page_number_offset_vertical
                 )
                 for exhibit in self
             ],
@@ -3105,12 +3131,19 @@ class ALExhibitDocument(ALDocument):
     has_addendum: bool
     auto_labeler: Callable
     auto_ocr: bool
+    # Deprecated
     bates_prefix: str
     maximum_size: int
     maximum_size_per_doc: int
     suffix_to_append: str
     exhibits: ALExhibitList
     table_of_contents: DAFile
+    page_number_prefix: str
+    page_number_digits: int
+    page_number_area: str | None
+    page_number_font_size: float
+    page_number_offset_horizontal: float
+    page_number_offset_vertical: float
 
     def init(self, *pargs, **kwargs) -> None:
         """Standard DAObject init method.
@@ -3125,8 +3158,6 @@ def init(self, *pargs, **kwargs) -> None:
             self.exhibits.auto_labeler = self.auto_labeler
         if hasattr(self, "auto_ocr"):
             self.exhibits.auto_ocr = self.auto_ocr
-        if hasattr(self, "bates_prefix"):
-            self.exhibits.bates_prefix = self.bates_prefix
         if hasattr(self, "include_exhibit_cover_pages"):
             self.exhibits.include_exhibit_cover_pages = self.include_exhibit_cover_pages
         else:
@@ -3144,13 +3175,30 @@ def init(self, *pargs, **kwargs) -> None:
         else:
             self.include_table_of_contents = True
             self.exhibits.include_table_of_contents = True
+        if hasattr(self, "bates_prefix"):
+            self.exhibits.bates_prefix = self.bates_prefix
         if not hasattr(self, "add_page_numbers"):
             self.add_page_numbers = False
+        self._set_default_attributes()
         self.has_addendum = False
         if not hasattr(self, "suffix_to_append"):
             # When the key is "preview", append it to the file name
             self.suffix_to_append = "preview"
 
+    def _set_default_attributes(self) -> None:
+        if not hasattr(self, "page_number_prefix"):
+            self.page_number_prefix = ""
+        if not hasattr(self, "page_number_digits"):
+            self.page_number_digits = 5
+        if not hasattr(self, "page_number_area"):
+            self.page_number_area = None
+        if not hasattr(self, "page_number_font_size"):
+            self.page_number_font_size = 10
+        if not hasattr(self, "page_number_offset_horizontal"):
+            self.page_number_offset_horizontal = 15
+        if not hasattr(self, "page_number_offset_vertical"):
+            self.page_number_offset_vertical = 15
+
     def has_overflow(self) -> bool:
         """
         Check if there is any overflow in the document.
@@ -3225,19 +3273,34 @@ def as_pdf(
             filename = base_name(self.filename) + ".pdf"
 
         if len(self.exhibits):
+            self._set_default_attributes()
             if self.include_table_of_contents:
                 toc_pages = self.table_of_contents.num_pages()
                 return pdf_concatenate(
                     self.table_of_contents,
                     self.exhibits.as_pdf(
-                        add_page_numbers=self.add_page_numbers, toc_pages=toc_pages
+                        add_page_numbers=self.add_page_numbers,
+                        page_number_prefix=self.page_number_prefix,
+                        page_number_digits=self.page_number_digits,
+                        page_number_area=self.page_number_area,
+                        page_number_font_size=self.page_number_font_size,
+                        page_number_offset_horizontal=self.page_number_offset_horizontal,
+                        page_number_offset_vertical=self.page_number_offset_vertical,
+                        toc_pages=toc_pages,
+                        pdfa=pdfa
                     ),
                     filename=filename,
                     pdfa=pdfa,
                 )
             else:
                 return self.exhibits.as_pdf(
                     add_page_numbers=self.add_page_numbers,
+                    page_number_prefix=self.page_number_prefix,
+                    page_number_digits=self.page_number_digits,
+                    page_number_area=self.page_number_area,
+                    page_number_font_size=self.page_number_font_size,
+                    page_number_offset_horizontal=self.page_number_offset_horizontal,
+                    page_number_offset_vertical=self.page_number_offset_vertical,
                     filename=filename,
                     pdfa=pdfa,
                 )