Mirrors
/
undetected-chromedriver
mirror of https://github.com/ultrafunkamsterdam/undetected-chromedriver.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added flag to keep the webdriver file if it exists, to speed up startup

This commit is contained in:
Kam S 2021-06-14 11:38:58 -04:00
parent e598e1ca1b
commit 46b7c2846a
2 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
undetected_chromedriver/patcher.py
View File

@ -43,7 +43,7 @@ class Patcher(object):
d = "~/.undetected_chromedriver" d = "~/.undetected_chromedriver"
data_path = os.path.abspath(os.path.expanduser(d)) data_path = os.path.abspath(os.path.expanduser(d))
def __init__(self, executable_path=None, force=False, version_main: int = 0): def __init__(self, executable_path=None, force=False, version_main: int = 0, keep_webdriver_file=False):
""" """
Args: Args:
@ -56,6 +56,7 @@ class Patcher(object):
""" """
self.force = force self.force = force
self.keep_webdriver_file = keep_webdriver_file
if not executable_path: if not executable_path:
executable_path = os.path.join(self.data_path, self.exe_name) executable_path = os.path.join(self.data_path, self.exe_name)
@ -98,10 +99,14 @@ class Patcher(object):
except FileNotFoundError: except FileNotFoundError:
pass pass
# If the binary exists and keep_webdriver_file was set to True,
# then no there is need to download it again
if not (os.path.exists(self.executable_path) and self.keep_webdriver_file):
release = self.fetch_release_number() release = self.fetch_release_number()
self.version_main = release.version[0] self.version_main = release.version[0]
self.version_full = release self.version_full = release
self.unzip_package(self.fetch_package()) self.unzip_package(self.fetch_package())
# i.patch() # i.patch()
return self.patch() return self.patch()

3
undetected_chromedriver/v2.py
View File

@ -82,6 +82,7 @@ class Chrome(selenium.webdriver.Chrome):
delay=5, delay=5,
version_main=None, version_main=None,
patcher_force_close=False, patcher_force_close=False,
keep_webdriver_file=False,
): ):
""" """
Creates a new instance of the chrome driver. Creates a new instance of the chrome driver.
@ -148,7 +149,7 @@ class Chrome(selenium.webdriver.Chrome):
setting it is not recommended, unless you know the implications and think setting it is not recommended, unless you know the implications and think
you might need it. you might need it.
""" """
patcher = Patcher(executable_path=executable_path, force=patcher_force_close, version_main=version_main) patcher = Patcher(executable_path=executable_path, force=patcher_force_close, version_main=version_main, keep_webdriver_file=keep_webdriver_file)
patcher.auto() patcher.auto()
if not options: if not options: