Update __init__.py
This commit is contained in:
parent
f3c040297a
commit
6e0f49797d
|
@ -21,6 +21,7 @@ import logging
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import zipfile
|
import zipfile
|
||||||
|
from distutils.version import LooseVersion
|
||||||
from urllib.request import urlopen, urlretrieve
|
from urllib.request import urlopen, urlretrieve
|
||||||
|
|
||||||
from selenium.webdriver import Chrome as _Chrome
|
from selenium.webdriver import Chrome as _Chrome
|
||||||
|
@ -29,8 +30,9 @@ from selenium.webdriver import ChromeOptions as _ChromeOptions
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
TARGET_VERSION = 84
|
|
||||||
__IS_PATCHED__ = 0
|
__IS_PATCHED__ = 0
|
||||||
|
TARGET_VERSION = 0
|
||||||
|
|
||||||
|
|
||||||
class Chrome:
|
class Chrome:
|
||||||
|
@ -106,9 +108,15 @@ class ChromeDriverManager(object):
|
||||||
def __init__(self, executable_path=None, target_version=None, *args, **kwargs):
|
def __init__(self, executable_path=None, target_version=None, *args, **kwargs):
|
||||||
|
|
||||||
_platform = sys.platform
|
_platform = sys.platform
|
||||||
|
|
||||||
|
if TARGET_VERSION: # user override using global
|
||||||
self.target_version = TARGET_VERSION
|
self.target_version = TARGET_VERSION
|
||||||
if target_version:
|
if target_version:
|
||||||
self.target_version = target_version
|
self.target_version = target_version # user override
|
||||||
|
if not self.target_version:
|
||||||
|
# if target_version still not set, fetch the current major release version
|
||||||
|
self.target_version = self.get_release_version_number().version[0] # only major version int
|
||||||
|
|
||||||
self._base = base_ = "chromedriver{}"
|
self._base = base_ = "chromedriver{}"
|
||||||
|
|
||||||
exe_name = self._base
|
exe_name = self._base
|
||||||
|
@ -172,7 +180,7 @@ class ChromeDriverManager(object):
|
||||||
if not self.target_version
|
if not self.target_version
|
||||||
else f"LATEST_RELEASE_{self.target_version}"
|
else f"LATEST_RELEASE_{self.target_version}"
|
||||||
)
|
)
|
||||||
return urlopen(self.__class__.DL_BASE + path).read().decode()
|
return LooseVersion(urlopen(self.__class__.DL_BASE + path).read().decode())
|
||||||
|
|
||||||
|
|
||||||
def fetch_chromedriver(self):
|
def fetch_chromedriver(self):
|
||||||
|
@ -183,7 +191,7 @@ class ChromeDriverManager(object):
|
||||||
"""
|
"""
|
||||||
base_ = self._base
|
base_ = self._base
|
||||||
zip_name = base_.format(".zip")
|
zip_name = base_.format(".zip")
|
||||||
ver = self.get_release_version_number()
|
ver = self.get_release_version_number().vstring
|
||||||
if os.path.exists(self.executable_path):
|
if os.path.exists(self.executable_path):
|
||||||
return self.executable_path
|
return self.executable_path
|
||||||
urlretrieve(
|
urlretrieve(
|
||||||
|
|
Loading…
Reference in New Issue