Update html2print.py

Author: thseiler

html2print/html2print.py

@@ -15,7 +15,7 @@
 
 import browsers
 import requests
-from requests import Response, RequestException
+from requests import RequestException, Response
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.chrome.service import Service
@@ -41,7 +41,9 @@
 
 class ChromeDriverManager:
     def get_chrome_driver(self, path_to_cache_dir: str):
-        chrome_version = self.get_chrome_version()
+        chrome_browser_info = browsers.get("chrome")
+        chrome_version = chrome_browser_info["version"]
+        chrome_path = chrome_browser_info["path"]
 
         # If Web Driver Manager cannot detect Chrome, it returns None.
         if chrome_version is None:
@@ -52,6 +54,9 @@ def get_chrome_driver(self, path_to_cache_dir: str):
 
         chrome_major_version = chrome_version.split(".")[0]
 
+        print(  # noqa: T201
+            f"html2print: Installed Chrome path: {chrome_path}"
+        )
         print(  # noqa: T201
             f"html2print: Installed Chrome version: {chrome_version}"
         )
@@ -181,8 +186,9 @@ def send_http_get_request(url, params=None, **kwargs) -> Response:
             f"html2print: "
             f"failed to get response for URL: {url} with error: {last_error}"
         )
-        raise RequestException(f"GET request failed after 3 attempts: {url}") from last_error
-
+        raise RequestException(
+            f"GET request failed after 3 attempts: {url}"
+        ) from last_error
 
     @staticmethod
     def get_chrome_version():
@@ -329,7 +335,12 @@ def create_webdriver(
     else:
         service = Service(path_to_chrome)
 
+    path_to_chrome_browser = ""
+    path_to_chrome_browser = browsers.get("chrome")["path"]
+    print(f"html2print: Chrome available at path: {path_to_chrome_browser}")  # noqa: T201
+
     webdriver_options = Options()
+    webdriver_options.binary_location = path_to_chrome_browser
     webdriver_options.add_argument("start-maximized")
     webdriver_options.add_argument("disable-infobars")
     # Doesn't seem to be needed.
@@ -462,7 +473,7 @@ def main():
         if browsers.get("chrome") is not None:
             path_to_chrome_browser = browsers.get("chrome")["path"]
             print(  # noqa: T201
-                f"html2print: Chrome available at path:{path_to_chrome_browser}"
+                f"html2print: Chrome available at path: {path_to_chrome_browser}"
             )
         sys.exit(0)
 

tests/integration/lit.cfg

@@ -23,3 +23,6 @@ config.suffixes = ['.itest', '.c']
 config.is_windows = lit_config.isWindows
 if not lit_config.isWindows:
     config.available_features.add('PLATFORM_IS_NOT_WINDOWS')
+
+# $HOME is required for chrome under linux as it wants to store things in ~/.local
+config.environment['HOME'] = os.environ.get('HOME', '/tmp')