htmlcmp 1.2.2__tar.gz → 1.2.4__tar.gz

{htmlcmp-1.2.2 → htmlcmp-1.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: htmlcmp
-Version: 1.2.2
+Version: 1.2.4
 Summary: Compare HTML files by rendered output
 Author: Andreas Stefl
 Maintainer-email: Andreas Stefl <stefl.andreas@gmail.com>

{htmlcmp-1.2.2 → htmlcmp-1.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "htmlcmp"
-version = "1.2.2"
+version = "1.2.4"
 description = "Compare HTML files by rendered output"
 classifiers = []
 authors = [

{htmlcmp-1.2.2 → htmlcmp-1.2.4}/src/htmlcmp/html_render_diff.py

@@ -1,7 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-import shutil
 import sys
 import argparse
 import io
@@ -36,8 +35,13 @@ def screenshot(browser: webdriver.Remote, url: str) -> Image.Image:
 
     target_find_by = By.TAG_NAME
     target = "body"
+    loaded_page_settling_time = 0
 
-    web_driver_wait = WebDriverWait(browser, 5)
+    # TODO for pdf2htmlex the second screenshot sometimes fades in from white... not sure why, but a sleep solves it
+    if "poppler" in url:
+        loaded_page_settling_time = 0.3
+
+    web_driver_wait = WebDriverWait(browser, 10)
     web_driver_wait.until(
         expected_conditions.presence_of_element_located((target_find_by, target))
     )
@@ -45,6 +49,8 @@ def screenshot(browser: webdriver.Remote, url: str) -> Image.Image:
         lambda driver: driver.execute_script("return document.readyState") == "complete"
     )
 
+    time.sleep(loaded_page_settling_time)
+
     png = browser.get_screenshot_as_png()
     return Image.open(io.BytesIO(png))
 
@@ -74,15 +80,22 @@ def get_browser(
 
 
 def html_render_diff(
-    a: str | Path, b: str | Path, browser: webdriver.Remote
+    a: str | Path,
+    b: str | Path,
+    browser: webdriver.Remote,
+    browser_b: webdriver.Remote = None,
 ) -> tuple[Image.Image, tuple[Image.Image, Image.Image]]:
     if not isinstance(a, (str, Path)) or not isinstance(b, (str, Path)):
         raise TypeError("Both a and b must be str or Path instances")
     if not isinstance(browser, webdriver.Remote):
         raise TypeError(f"Expected webdriver.Remote, got {type(browser)}")
+    if browser_b is None:
+        browser_b = browser
+    elif not isinstance(browser_b, webdriver.Remote):
+        raise TypeError(f"Expected webdriver.Remote, got {type(browser_b)}")
 
     image_a = screenshot(browser, to_url(a))
-    image_b = screenshot(browser, to_url(b))
+    image_b = screenshot(browser_b, to_url(b))
 
     image_a = image_a.convert("RGB")
     image_b = image_b.convert("RGB")

{htmlcmp-1.2.2 → htmlcmp-1.2.4}/src/htmlcmp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: htmlcmp
-Version: 1.2.2
+Version: 1.2.4
 Summary: Compare HTML files by rendered output
 Author: Andreas Stefl
 Maintainer-email: Andreas Stefl <stefl.andreas@gmail.com>