Add Add Attachment

Assisted-by: Codex

Author: datalogics-cgreen

src/pdfrest/client.py

@@ -82,6 +82,7 @@
     GifPdfRestPayload,
     JpegPdfRestPayload,
     OcrPdfPayload,
+    PdfAddAttachmentPayload,
     PdfCompressPayload,
     PdfDecryptPayload,
     PdfEncryptPayload,
@@ -2973,6 +2974,33 @@ def compress_pdf(
             timeout=timeout,
         )
 
+    def add_attachment_to_pdf(
+        self,
+        file: PdfRestFile | Sequence[PdfRestFile],
+        *,
+        attachment: PdfRestFile | Sequence[PdfRestFile],
+        output: str | None = None,
+        extra_query: Query | None = None,
+        extra_headers: AnyMapping | None = None,
+        extra_body: Body | None = None,
+        timeout: TimeoutTypes | None = None,
+    ) -> PdfRestFileBasedResponse:
+        """Attach an uploaded file to a PDF."""
+
+        payload: dict[str, Any] = {"files": file, "attachment": attachment}
+        if output is not None:
+            payload["output"] = output
+
+        return self._post_file_operation(
+            endpoint="/pdf-with-added-attachment",
+            payload=payload,
+            payload_model=PdfAddAttachmentPayload,
+            extra_query=extra_query,
+            extra_headers=extra_headers,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+
     def flatten_transparencies(
         self,
         file: PdfRestFile | Sequence[PdfRestFile],
@@ -4260,6 +4288,33 @@ async def compress_pdf(
             timeout=timeout,
         )
 
+    async def add_attachment_to_pdf(
+        self,
+        file: PdfRestFile | Sequence[PdfRestFile],
+        *,
+        attachment: PdfRestFile | Sequence[PdfRestFile],
+        output: str | None = None,
+        extra_query: Query | None = None,
+        extra_headers: AnyMapping | None = None,
+        extra_body: Body | None = None,
+        timeout: TimeoutTypes | None = None,
+    ) -> PdfRestFileBasedResponse:
+        """Asynchronously attach an uploaded file to a PDF."""
+
+        payload: dict[str, Any] = {"files": file, "attachment": attachment}
+        if output is not None:
+            payload["output"] = output
+
+        return await self._post_file_operation(
+            endpoint="/pdf-with-added-attachment",
+            payload=payload,
+            payload_model=PdfAddAttachmentPayload,
+            extra_query=extra_query,
+            extra_headers=extra_headers,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+
     async def flatten_transparencies(
         self,
         file: PdfRestFile | Sequence[PdfRestFile],

src/pdfrest/models/_internal.py

@@ -1142,6 +1142,46 @@ class PdfFlattenAnnotationsPayload(BaseModel):
     ] = None
 
 
+class PdfAddAttachmentPayload(BaseModel):
+    """Adapt caller options into a pdfRest-ready add-attachment request payload."""
+
+    files: Annotated[
+        list[PdfRestFile],
+        Field(
+            min_length=1,
+            max_length=1,
+            validation_alias=AliasChoices("file", "files"),
+            serialization_alias="id",
+        ),
+        BeforeValidator(_ensure_list),
+        AfterValidator(
+            _allowed_mime_types("application/pdf", error_msg="Must be a PDF file")
+        ),
+        PlainSerializer(_serialize_as_first_file_id),
+    ]
+    attachments: Annotated[
+        list[PdfRestFile],
+        Field(
+            min_length=1,
+            max_length=1,
+            validation_alias=AliasChoices(
+                "attachment",
+                "attachments",
+                "file_to_attach",
+                "files_to_attach",
+            ),
+            serialization_alias="id_to_attach",
+        ),
+        BeforeValidator(_ensure_list),
+        PlainSerializer(_serialize_as_first_file_id),
+    ]
+    output: Annotated[
+        str | None,
+        Field(serialization_alias="output", min_length=1, default=None),
+        AfterValidator(_validate_output_prefix),
+    ] = None
+
+
 class PdfRestrictPayload(BaseModel):
     """Adapt caller options into a pdfRest-ready restrict-PDF request payload."""