Python 封装 浏览器驱动 工具类

# coding=utf-8

from selenium import webdriver
import os, sys

from selenium.common.exceptions import WebDriverException

driver_base = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
sys.path.append(driver_base)

driver = None


def init_driver(browser=None):
    global driver
    try:
        if browser is None or browser == 'Chrome':
            option = webdriver.ChromeOptions()
            # 增加Chrome启动时属性，去掉告警提示
            # option.add_argument("disable-infobars")
            # 开启无痕模式
            option.add_argument('--incognito')
            path = os.path.join(driver_base, 'config\\chromedriver.exe')
            driver = webdriver.Chrome(executable_path=path, options=option)
        elif browser == 'Firefox':
            path = os.path.join(driver_base, 'config\\geckodriver.exe')
            driver = webdriver.Firefox(executable_path=path)
        return driver
    except WebDriverException as e:
        raise e