selenium4 自动下载驱动

包依赖：

pip install webdriver-manager==4.0.1
pip install selenium==4.1.0

# -*- coding: utf-8 -*-
import os
from selenium import webdriver


def get_file_path(fileName):
    """
    获取 文件的路径
    :param fileName: 文件名称（带后缀）
    :return: file_path 文件路径
    """
    # 获取当前文件所在目录
    current_dir = os.path.dirname(os.path.abspath(__file__))
    # 获取指定目录下所有文件的路径
    for root, dirs, files in os.walk(current_dir):
        for file in files:
            if file == fileName:
                # 拼接文件路径
                file_path = os.path.join(root, file)
                # 获取文件的绝对路径
                absolute_path = os.path.abspath(file_path)
                # 打印文件路径
                file_path = absolute_path.replace('\\', '\\\\')
                return file_path
            
            

def browser_Firefox():
    """
    选择Firefox火狐浏览器执行自动化
    :return:
    """

    # 识别驱动执行
    # firefoxPtah = get_file_path('geckodriver.exe')
    # driver = webdriver.Firefox(executable_path=firefoxPtah)

    # 无需驱动执行
    from webdriver_manager.firefox import GeckoDriverManager  # firefox
    from selenium.webdriver.firefox.service import Service as FirefoxService
    driver = webdriver.Firefox(service=FirefoxService(GeckoDriverManager().install()))

    driver.maximize_window()  # 浏览器最大化
    print(f"打开{driver}浏览器")
    
    yield driver  # 把driver带出 def函数  并退出def函数
    

def browser_Ie():
    """
    选择ie浏览器执行自动化
    :return:
    """

    # 识别驱动执行
    # iePath = get_file_path("IEDriverServer.exe")  # ie的驱动还没下过  忘记名字了
    # driver = webdriver.Ie(executable_path=iePath.replace('\\', '\\\\'))

    # 无需驱动执行
    from webdriver_manager.microsoft import IEDriverManager  # ie
    from selenium.webdriver.ie.service import Service as IEService
    driver = webdriver.Ie(service=IEService(IEDriverManager().install()))

    driver.maximize_window()  # 浏览器最大化
    print(f"打开{driver}浏览器")
    
    yield driver  # 把driver带出 def函数  并退出def函数


def browser_Edge():
    """
    选择Edge浏览器执行自动化
    :return:
    """

    # 识别驱动执行
    # edgePath = get_file_path("msedgedriver.exe")
    # driver = webdriver.Edge(executable_path=edgePath.replace('\\', '\\\\'))

    # 无需驱动执行
    from webdriver_manager.microsoft import EdgeChromiumDriverManager  # edge
    from selenium.webdriver.edge.service import Service as EdgeService
    driver = webdriver.Edge(service=EdgeService(EdgeChromiumDriverManager().install()))

    driver.maximize_window()  # 浏览器最大化
    print(f"打开{driver}浏览器")
    
    yield driver  # 把driver带出 def函数  并退出def函数


def browser_Chrome():
    """
    选择Chrome谷歌浏览器执行自动化
    :return:
    """

    # 识别驱动执行
    # chromePath = get_file_path("chromedriver.exe")
    # driver = webdriver.Chrome(executable_path=chromePath.replace('\\', '\\\\'))

    # 无需驱动执行
    from webdriver_manager.chrome import ChromeDriverManager  # Chrome
    from selenium.webdriver.chrome.service import Service as ChromeService
    driver = webdriver.Chrome(service=ChromeService(ChromeDriverManager().install()))


    driver.maximize_window()  # 浏览器最大化
    print(f"打开{driver}浏览器")
    
    yield driver  # 把driver带出 def函数  并退出def函数