From 0e26e6122e02ca893f7f9c788ccb18c53c0932cb Mon Sep 17 00:00:00 2001 From: jdholtz Date: Wed, 23 Aug 2023 23:30:40 -0500 Subject: [PATCH] Add better checking for version_main to add headless flag Gets rid of the logger warning message when version_main is None --- undetected_chromedriver/__init__.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/undetected_chromedriver/__init__.py b/undetected_chromedriver/__init__.py index 96f108a..d0988ed 100644 --- a/undetected_chromedriver/__init__.py +++ b/undetected_chromedriver/__init__.py @@ -396,16 +396,10 @@ class Chrome(selenium.webdriver.chrome.webdriver.WebDriver): options.arguments.extend(["--no-sandbox", "--test-type"]) if headless or options.headless: - #workaround until a better checking is found - try: - if self.patcher.version_main < 108: - options.add_argument("--headless=chrome") - elif self.patcher.version_main >= 108: - options.add_argument("--headless=new") - except: - logger.warning("could not detect version_main." - "therefore, we are assuming it is chrome 108 or higher") + if not self.patcher.version_main or self.patcher.version_main >= 108: options.add_argument("--headless=new") + else: + options.add_argument("--headless=chrome") options.add_argument("--window-size=1920,1080") options.add_argument("--start-maximized")