相信看了工厂模式的简介，一定能懂这样的设计思路了吧

    我把这段代码比作什么工厂好呢，就叫它： Driver工厂
    已配置环境的driver、没有配置的（准备用 .exe）、使用selenium gird的工厂

    当然使用gird的driver工厂，你需要传入的是master-node-url
    1610704328(1).png

    1. import abc
    2. from abc import ABC
    3. from selenium import webdriver
    4. from conf.ErrorConfig import GETDRIVERERROR, BROWEREXPECTION
    5. from selenium.webdriver.chrome.webdriver import WebDriver
    7. from conf.PathConfig import DRIVER_DIR
    10. class Driver(ABC):
    11.     driver = None
    13.     @abc.abstractmethod
    14.     def product_driver(self, *args, **kwargs):
    15.         pass
    17.     @classmethod
    18.     def get_driver(cls, *args, **kwargs) -> WebDriver:
    19.         if cls.driver is None:
    20.             try:
    21.                 cls.driver: WebDriver = cls().product_driver(*args, **kwargs)
    22.                 cls.driver.maximize_window()
    23.             except:
    24.                 raise GETDRIVERERROR
    26.             return cls.driver
    27.         else:
    28.             return cls.driver
    31. class Local(Driver):
    32.     """
    33.     访问本地环境变量
    34.     """
    35.     def product_driver(self, browertype='chrome', debug=False):
    36.         if debug:
    37.             options = webdriver.ChromeOptions()
    38.             options.debugger_address = "127.0.0.1:9222"
    39.             driver = webdriver.Chrome(options=options)
    40.         else:
    41.             if browertype == 'firefox':
    42.                 driver = webdriver.Firefox()
    43.             elif browertype == 'chrome':
    44.                 driver = webdriver.Chrome()
    45.             else:
    46.                 raise BROWEREXPECTION
    47.         return driver
    50. class test(Driver):
    51.     """
    52.     未配置chromedriver环境变量
    53.     """
    54.     def product_driver(self, *args, **kwargs):
    55.         driver = webdriver.Chrome(executable_path=DRIVER_DIR + 'chromedriver.exe')
    56.         return driver
    59. class Grid(Driver):
    61.     def product_driver(self, selenium_grid_url, brower='chrome'):
    62.         """
    63.         selenium gird
    64.         :param selenium_grid_url: 访问gird server-node地址 
    65.         :param brower: 
    66.         :return: 
    67.         """
    68.         des = webdriver.DesiredCapabilities
    69.         if brower == 'firefox':
    70.             capablities = des.FIREFOX.copy()
    71.         elif brower == 'chrome':
    72.             capablities = des.CHROME.copy()
    73.         else:
    74.             raise
    75.         driver = webdriver.Remote(command_executor=selenium_grid_url,
    76.                                   desired_capabilities=capablities)
    77.         return driver
    80. class DriverFactory:
    81.     """
    82.     DriverFacory FOR chrome Firefox..
    83.     """
    85.     def __init__(self):
    86.         pass
    88.     def __new__(cls, *args, **kwargs):
    89.         if not hasattr(DriverFactory, "_instance"):
    90.             DriverFactory._instance = object.__new__(cls)
    91.         return DriverFactory._instance
    93.     @staticmethod
    94.     def chooseBrower(env: str = 'test'):
    95.         """
    96.         choose env to product driver
    97.         :param env:
    98.         :param
    99.         :return:
    100.         """
    101.         map_ = {"local": Local,
    102.                 'test': test,
    103.                 "grid": Grid}
    104.         if map_.get(env, None) is None:
    105.             raise BROWEREXPECTION(env)
    106.         return map_[env.lower()]
    109. if __name__ == "__main__":
    110.     dri = DriverFactory().chooseBrower("test").get_driver()
    111.     dri.get('https://www.baidu.com')

    看到结束
    白嫖不能忍，请发一次文