PO模式之浏览器的封装

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

from common import config_value
import os

current_path = os.path.dirname(file)
dri_path = os.path.join(current_path, config_value.config.chrome_path)

class Browser:
def init(self, driver_path=dri_path, driver_name=config_value.config.driver_name):
self.__driver_name = driver_name
self.__driver_path = driver_path

def get_driver(self):
    if str(self.__driver_name).lower() == 'chrome':
        return self.__get_chrome_driver()
    elif str(self.__driver_name).lower() == 'firefox':
        return self.__get_firefox_driver()

def __get_chrome_driver(self):
    chrome_options = Options()
    chrome_options.add_argument('--disable-gpu')  # 谷歌文档提到需要加上这个属性来规避bug
    chrome_options.add_argument('lang=zh_CN.UTF-8')  # 设置默