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

15
undetected_chromedriver/patcher.py
View File

@ -43,7 +43,7 @@ class Patcher(object):
d = "~/.undetected_chromedriver"
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:
@ -56,6 +56,7 @@ class Patcher(object):
"""
self.force = force
self.keep_webdriver_file = keep_webdriver_file
if not executable_path:
executable_path = os.path.join(self.data_path, self.exe_name)
@ -98,10 +99,14 @@ class Patcher(object):
except FileNotFoundError:
pass
release = self.fetch_release_number()
self.version_main = release.version[0]
self.version_full = release
self.unzip_package(self.fetch_package())
# 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()
self.version_main = release.version[0]
self.version_full = release
self.unzip_package(self.fetch_package())
# i.patch()
return self.patch()

3
undetected_chromedriver/v2.py
View File

@ -82,6 +82,7 @@ class Chrome(selenium.webdriver.Chrome):
delay=5,
version_main=None,
patcher_force_close=False,
keep_webdriver_file=False,
):
"""
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
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()
if not options: