Update __init__.py
This commit is contained in:
parent
cebddec58e
commit
9f727e1f46
|
@ -138,18 +138,17 @@ class ChromeDriverManager(object):
|
|||
installed = False
|
||||
selenium_patched = False
|
||||
|
||||
|
||||
def __init__(self, executable_path=None, target_version=None, *args, **kwargs):
|
||||
self.executable_path = executable_path or "chromedriver.exe"
|
||||
self.platform = sys.platform
|
||||
self.target_version = target_version
|
||||
|
||||
|
||||
def patch_selenium_webdriver(self_):
|
||||
"""
|
||||
Patches existing webdriver path on <executable_path> OR if executable_path is None, will download
|
||||
and patch a new webdriver binary for chrome <version_int> (automatically finds latest release of main version)
|
||||
Patches selenium package Chrome, ChromeOptions classes for current session
|
||||
|
||||
:param str executable_path: OPTIONAL path to existing chromedriver executable to patch
|
||||
:param int version_int: OPTIONAL target chrome main version. default 79
|
||||
:return:
|
||||
"""
|
||||
import selenium.webdriver.chrome.service
|
||||
|
@ -161,6 +160,7 @@ class ChromeDriverManager(object):
|
|||
)
|
||||
self_.__class__.selenium_patched = True
|
||||
|
||||
|
||||
def install(self, patch_selenium=True):
|
||||
"""
|
||||
Initialize the patch
|
||||
|
@ -185,6 +185,7 @@ class ChromeDriverManager(object):
|
|||
if patch_selenium:
|
||||
self.patch_selenium_webdriver()
|
||||
|
||||
|
||||
def get_release_version_number(self):
|
||||
"""
|
||||
Gets the latest major version available, or the latest major version of self.target_version if set explicitly.
|
||||
|
@ -198,6 +199,7 @@ class ChromeDriverManager(object):
|
|||
)
|
||||
return urlopen(_DL_BASE + path).read().decode()
|
||||
|
||||
|
||||
def fetch_chromedriver(self):
|
||||
"""
|
||||
Downloads ChromeDriver from source and unpacks the executable
|
||||
|
@ -217,6 +219,7 @@ class ChromeDriverManager(object):
|
|||
os.remove(zip_name)
|
||||
return exe_name
|
||||
|
||||
|
||||
def patch_binary(self):
|
||||
"""
|
||||
Patches the ChromeDriver binary
|
||||
|
|
Loading…
Reference in New Issue