changing pdf printing to use built-in selenium print to pdf functionality

Author: nils-drechsel

visokio-omniprint/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "visokio-omniprint"
-version = "1.1.7"
+version = "1.2.0"
 description = "Visokio Omniscope PDF printing library"
 readme = "README.md"
 requires-python = ">=3.7"
@@ -25,7 +25,7 @@ classifiers = [
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
 dependencies = [
-  "selenium",
+  "selenium>=4.26.0",
   "webdriver-manager",
   "get-chrome-driver>=1.3.19",
   "tinify",

visokio-omniprint/visokio_omniprint/DriverBase.py

@@ -73,21 +73,6 @@ def get_driver(self, is_docker):
             return self.get_driver_host(options)
 
 
-    def send_devtools(self, driver, cmd, params):
-        """
-        This function sends a command to the Chrome DevTools API and returns the result.
-        """
-        resource = "/session/%s/chromium/send_command_and_get_result" % driver.session_id
-        url = driver.command_executor._url + resource
-        body = json.dumps({"cmd": cmd, "params": params})
-        response = driver.command_executor._request("POST", url, body)
-
-        if not response:
-            raise Exception(response.get("value"))
-
-        return response.get("value")
-
-
     def get_driver_for_url(self, url, timeout, is_docker):
         driver = self.get_driver(is_docker)
         driver.get(url)  # Navigate to the specified URL

visokio-omniprint/visokio_omniprint/Pdf.py

@@ -1,5 +1,6 @@
 import json
 import base64
+from selenium.webdriver.common.print_page_options import PrintOptions
 
 from .DriverBase import DriverBase
 
@@ -9,17 +10,13 @@ def create_pdf(self, url, timeout, is_docker):
 
         driver = self.get_driver_for_url(url, timeout, is_docker)
 
-        print_options = {
-            "landscape": False,
-            "displayHeaderFooter": False,
-            "printBackground": True,
-            "preferCSSPageSize": True,
-        }
+        print_options = PrintOptions()
+        print_options.orientation = "portrait"
 
-        result = self.send_devtools(driver, "Page.printToPDF", print_options)
+        result = driver.print_page(print_options)
         driver.get("about:blank") # Navigate away to ensure page is unloaded properly.
         driver.quit()
-        return base64.b64decode(result["data"])
+        return base64.b64decode(result)
 
 
     def write_pdf_to_path(self, path, pdf):