python - selenium windows下自动根据chrome版本下载最新驱动
前言
在编写自动化测试框架时，chrome浏览器总是会自动更新版本，时不时会导致编写脚本时，chromedriver不可用。所以编写以下脚本完成自动更新检测，在每次启动时，更新或者替换chormedriver，只针对于winodws系统。
代码分解，完整代码放在最后。
1、从windows注册表中获取chrome版本号
# logger是自己封装的日志模块，可以用print代替!!!
import winreg # 和注册表交互
def get_chrome_version():
"""
通过windows注册表获取chrome版本号
:return: 版本号（str）
"""
try:
# 从注册表中获得版本号
key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r'Software\Google\Chrome\BLBeacon')
_v, _type = winreg.QueryValueEx(key, 'version')
logger.info('Chrome浏览器版本号为: {}'.format(_v))
return _v # 返回版本号
except WindowsError as e:
logger.error(f"检查Chrome失败: {e}")
2、从chromedriver驱动中获取当前驱动版本号
import subprocess # 用于执行cmd命令
def get_driver_version(abs_path):
"""
获取driver版本号
:param abs_path: chromedriver.exe的绝对路径
:return: 版本号（str）
"""
cmd = r'{} --version'.format(abs_path) # 拼接成cmd命令
try:
# 执行cmd命令并接收命令回显
out, err = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate()
out = out.decode('utf-8')
_v = out.split(' ')[1] # 拆分回显字符串，获取版本号
logger.info('chromedriver驱动版本号为:{}'.format(_v))
return _v
except IndexError as e:
logger.error(f'检查driver驱动失败:{e}')
return 0
3、下载并解压chromedriver。
import urllib.request # 发送http请求
import urllib.parse # 拼接url
import zipfile # 操作.zip文件
def get_driver_version(abs_path):
"""
获取driver版本号
:param abs_path: chromedriver.exe的绝对路径
:return: 版本号（str）
"""
cmd = r'{} --version'.format(abs_path) # 拼接成cmd命令
try:
# 执行cmd命令并接收命令回显
out, err = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate()
out = out.decode('utf-8')
_v = out.split(' ')[1] # 拆分回显字符串，获取版本号
logger.info('chromedriver驱动版本号为:{}'.format(_v))
return _v
except IndexError as e:
logger.error(f'检查driver驱动失败:{e}')
return 0
4、主方法运行版本检测并下载
import urllib.request # 发送http请求
import urllib.parse # 拼接url
def check_version_match(driver_path=''):
if platform.system() == "Windows":
chrome_version = get_chrome_version()
# 拼接镜像下载路径
url = f"https://cdn.npmmirror.com/binaries/chromedriver/{chrome_version}/chromedriver_win32.zip"
if os.path.exists(driver_path):
logger.info(f'Chrome版本与chromedriver版本比较')
driver_version = get_driver_version(driver_path)
if str(chrome_version) != str(driver_version):
# 下载文件的保存路径，与chromedriver同级
down_load_driver(url, driver_path) # call下载文件的方法
else:
logger.info("版本号一致，忽略下载~")
return driver_path
else:
logger.error(f"不存在的路径：【{driver_path}】")
sys.exit()
5、完整代码
import os
import platform
import winreg # 和注册表交互
import subprocess # 用于执行cmd命令
import urllib.request # 发送http请求
import urllib.parse # 拼接url
import zipfile # 操作.zip文件
from config.log_config import logger
import sys
def get_chrome_version():
"""
通过windows注册表获取chrome版本号
:return: 版本号（str）
"""
try:
# 从注册表中获得版本号
key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r'Software\Google\Chrome\BLBeacon')
_v, _type = winreg.QueryValueEx(key, 'version')
logger.info('Chrome浏览器版本号为: {}'.format(_v))
return _v # 返回版本号
except WindowsError as e:
logger.error(f"检查Chrome失败: {e}")
def get_driver_version(abs_path):
"""
获取driver版本号
:param abs_path: chromedriver.exe的绝对路径
:return: 版本号（str）
"""
cmd = r'{} --version'.format(abs_path) # 拼接成cmd命令
try:
# 执行cmd命令并接收命令回显
out, err = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate()
out = out.decode('utf-8')
_v = out.split(' ')[1] # 拆分回显字符串，获取版本号
logger.info('chromedriver驱动版本号为:{}'.format(_v))
return _v
except IndexError as e:
logger.error(f'检查driver驱动失败:{e}')
return 0
def down_load_driver(url, original_driver_path):
"""
下载driver驱动
:param url: 下载地址
:param original_driver_path: 原始driver驱动存放地址
:return:
"""
# 访问淘宝镜像首页
logger.info(f'下载url为： {url}')
# 指定下载的文件名和保存位置
save_d = os.path.dirname(original_driver_path)
logger.info(f'driver文件保存位置为： {save_d}')
# 开始下载，并显示下载进度(progressFunc)
urllib.request.urlretrieve(url, os.path.join(save_d, 'chromedriver_win32.zip'))
# 压缩包路径
z_file_path = os.path.join(save_d, 'chromedriver_win32.zip')
# 下载完成后解压
zFile = zipfile.ZipFile(z_file_path, 'r')
for fileM in zFile.namelist():
zFile.extract(fileM, save_d)
zFile.close()
logger.warning("----------------------【driver文件下载完成】---------------------")
# 删除压缩包
os.remove(z_file_path)
logger.warning("----------------------【删除[chromedriver_win32.zip]压缩包】---------------------")
def check_version_match(driver_path=''):
if platform.system() == "Windows":
chrome_version = get_chrome_version()
# 拼接镜像下载路径
url = f"https://cdn.npmmirror.com/binaries/chromedriver/{chrome_version}/chromedriver_win32.zip"
if os.path.exists(driver_path):
logger.info(f'Chrome版本与chromedriver版本比较')
driver_version = get_driver_version(driver_path)
if str(chrome_version) != str(driver_version):
# 下载文件的保存路径，与chromedriver同级
down_load_driver(url, driver_path) # call下载文件的方法
else:
logger.info("版本号一致，忽略下载~")
return driver_path
else:
logger.error(f"不存在的路径：【{driver_path}】")
sys.exit()
