import os

import re

import winreg

import zipfile

import requests

base_url = 'http://npm.taobao.org/mirrors/chromedriver/'

version_re = re.compile(r'^[1-9]\d*\.\d*.\d*')  # 匹配前3位版本号的正则表达式

def getChromeVersion():

    """通过注册表查询chrome版本"""

    try:

        key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, 'Software\\Google\\Chrome\\BLBeacon')

        value, t = winreg.QueryValueEx(key, 'version')

        return version_re.findall(value)[0]  # 返回前3位版本号

    except WindowsError as e:

        # 没有安装chrome浏览器

        return "1.1.1"

def getChromeDriverVersion():

    """查询Chromedriver版本"""

    outstd2 = os.popen('cd D:\Python\chromedriver && chromedriver --version').read()

    try:

        version = outstd2.split(' ')[1]

        version = ".".join(version.split(".")[:-1])

        return version

    except Exception as e:

        return "0.0.0"

def getLatestChromeDriver(version):

    # 获取该chrome版本的最新driver版本号

    url = f"{base_url}LATEST_RELEASE_{version}"

    latest_version = requests.get(url).text

    print(f"与当前chrome匹配的最新chromedriver版本为: {latest_version}")

    # 下载chromedriver

    print("开始下载chromedriver...")

    download_url = f"{base_url}{latest_version}/chromedriver_win32.zip"

    file = requests.get(download_url)

    with open("D:\Python\chromedriver\chromedriver.zip", 'wb') as zip_file:  # 保存文件到脚本所在目录

        zip_file.write(file.content)

    print("下载完成.")

    # 解压

    f = zipfile.ZipFile("D:\Python\chromedriver\chromedriver.zip", 'r')

    for file in f.namelist():

        f.extract(file,'D:\Python\chromedriver')

    print("解压完成.")

def checkChromeDriverUpdate():

    chrome_version = getChromeVersion()

    print(f'当前chrome版本: {chrome_version}')

    driver_version = getChromeDriverVersion()

    print(f'当前chromedriver版本: {driver_version}')

    if chrome_version == driver_version:

        print("版本兼容，无需更新.")

        return

    print("chromedriver版本与chrome浏览器不兼容，更新中>>>")

    try:

        getLatestChromeDriver(chrome_version)

        print("chromedriver更新成功!")

    except requests.exceptions.Timeout:

        print("chromedriver下载失败，请检查网络后重试！")

    except Exception as e:

        print(f"chromedriver未知原因更新失败: {e}")

if __name__ == "__main__":

    checkChromeDriverUpdate()

#安装依赖

pip install webdriver_manager.chrome

#代码

from webdriver_manager.chrome import ChromeDriverManager

browser = webdriver.Chrome(ChromeDriverManager().install())