Update driver.py

uNickz · Dec 21, 2023 · a1eadbd · a1eadbd
1 parent f310faf
commit a1eadbd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/WebSite2PDF/driver/driver.py b/WebSite2PDF/driver/driver.py
@@ -1,4 +1,5 @@
 import requests
+import selenium
 import logging
 import json
 import os
@@ -26,7 +27,10 @@ def start_client(self, options: Optional[Options] = None, service: Optional[Serv
             raise ClientAlreadyStarted
 
         if install_driver:
-            self.driver = webdriver.Chrome(executable_path = ChromeDriverManager().install(), options = options, service = service)
+            if tuple(map(int, selenium.__version__.split("."))) >= (4, 10, 0):
+                self.driver = webdriver.Chrome(options = options, service = Service(executable_path = ChromeDriverManager().install()))
+            else:
+                self.driver = webdriver.Chrome(executable_path = ChromeDriverManager().install(), options = options, service = service)
         else:
             self.driver = webdriver.Chrome(options = options, service = service)