Add Tab.print_to_pdf

Author: stephanlensky

CHANGELOG.md

@@ -11,7 +11,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
-- Added `tab.screenshot_b64` and `element.screenshot_b64` methods to return screenshot as base64 string @falmar
+- Added `Tab.screenshot_b64` and `Element.screenshot_b64` methods to return screenshot as base64 string @falmar
+- Added `Tab.print_to_pdf` to print the current page to a PDF file @stephanlensky
 
 ### Changed
 

zendriver/core/tab.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import asyncio
+import base64
 import datetime
 import logging
 import pathlib
@@ -1381,14 +1382,33 @@ async def save_screenshot(
 
         data = await self.screenshot_b64(format=format, full_page=full_page)
 
-        import base64
-
         data_bytes = base64.b64decode(data)
         if not path:
             raise RuntimeError("invalid filename or path: '%s'" % filename)
         path.write_bytes(data_bytes)
         return str(path)
 
+    async def print_to_pdf(self, filename: PathLike, **kwargs: Any) -> pathlib.Path:
+        """
+        Prints the current page to a PDF file and saves it to the specified path.
+
+        :param filename: The path where the PDF will be saved.
+        :param kwargs: Additional options for printing to be passed to :py:obj:`cdp.page.print_to_pdf`.
+        :return: The path to the saved PDF file.
+        :rtype: pathlib.Path
+        """
+        filename = pathlib.Path(filename)
+        if filename.is_dir():
+            raise ValueError(
+                f"filename {filename} must be a file path, not a directory"
+            )
+
+        data, _ = await self.send(cdp.page.print_to_pdf(**kwargs))
+
+        data_bytes = base64.b64decode(data)
+        filename.write_bytes(data_bytes)
+        return filename
+
     async def set_download_path(self, path: PathLike):
         """
         sets the download path and allows downloads