Add bates number params to AL Document

docassemble/AssemblyLine/al_document.py

@@ -1582,6 +1582,22 @@ def init(self, *pargs, **kwargs) -> None:
             self.auto_gather = False
         if "gathered" not in kwargs:
             self.gathered = True
+        if "add_page_numbers" not in kwargs:
+            self.add_page_numbers = False
+        if "page_number_prefix" not in kwargs:
+            self.page_number_prefix = ""
+        if "page_number_start" not in kwargs:
+            self.page_number_start = 1
+        if "page_number_digits" not in kwargs:
+            self.page_number_digits = 5
+        if "page_number_area" not in kwargs:
+            self.page_number_area = None
+        if "page_number_font_size" not in kwargs:
+            self.page_number_font_size = None
+        if "page_number_horizontal_offset" not in kwargs:
+            self.page_number_horizontal_offset = None
+        if "page_number_vertical_offset" not in kwargs:
+            self.page_number_vertical_offset = None
         self.initializeAttribute("cache", DALazyAttribute)
         self.always_enabled = hasattr(self, "enabled") and self.enabled
         # Pre-cache some DALazyTemplates we set up to aid translation that won't
@@ -1649,6 +1665,16 @@ def as_pdf(
                 filename=f"{base_name(self.filename)}{append_suffix}.pdf",
                 pdfa=pdfa,
             )
+        if self.add_page_numbers:
+            pdf.bates_number(
+                prefix=self.page_number_prefix,
+                start=self.page_number_start,
+                digits=self.page_number_digits,
+                area=self.page_number_area,
+                font_size=self.page_number_font_size,
+                horizontal_offset=self.page_number_horizontal_offset,
+                vertical_offset=self.page_number_vertical_offset,
+            )
         pdf.title = self.title
         setattr(self.cache, safe_key, pdf)