史上最强大的selenium webdriver的封装

# coding:utf-8

import logging
import unittest
from selenium import webdriver


class cached_class_property(object):  
    def __init__(self, func):
        self.func = func

    def __get__(self, obj, cls):
        if obj is None:
            return self
        value = self.func(obj)
        setattr(cls, self.func.__name__, value)
        return value


class DriverWrapper():
    """
    1、基于selenium的二次封装，
    2、支持无限实例化此类，仍然保持使用同一个浏览器窗口。
    3、支持使用自定义的方法名字，同时直接性的支持使用所有此类中没有定义的方法，但在官方类中有的api方法，比如直接支持DriverWrapper().execute_script(script, *args)这种写法。
    4、其余想自定义名称的方法可以在这个类下面接着写或者继承这个类再添加其他更多方法
    """

    def __init__(self, driver_name):
        """
        :param driver_name: 浏览器名字数字或者字母
        """
        self.driver_name = driver_name

    @cached_class_property
    def driver(self):
        driver_var = None
        if self.driver_name in [‘chrome‘, 1]:
            driver_var = webdriver.Chrome()
        if self.driver_name in [‘firefox‘, 2]:
            driver_var = webdriver.Chrome()
        return driver_var

    @cached_class_property
    def logger(self):
        logger_var = logging.getLogger(self.__class__.__name__)
        logger_var.setLevel(logging.DEBUG)
        stream_handler = logging.StreamHandler()
        stream_handler.setFormatter(logging.Formatter(‘%(asctime)s - %(name)s - %(filename)s - %(lineno)d - %(levelname)s - %(message)s‘, "%Y-%m-%d %H:%M:%S"))
        logger_var.addHandler(stream_handler)
        return logger_var

    def open(self, url):
        self.driver.get(url)

    def find_element_by_css_selector(self, css_str):  # 使用自定义的方法覆盖了原方法，比如先打印出一段话
        self.logger.debug(‘要查找的元素的css选择器是 --> ‘ + css_str)
        self.driver.find_element_by_css_selector(css_str)

    def __getattr__(self, item):  # 想把其他的webdriver的操作方法直接添加进来，不一个一个的再写一个方法然后调用driver属性的方法，不想一直搞冗余的代码，可以这么做。python先使用__getattribute__，查不到才会调用__getsttr__方法，利用这个特性，来实现这个添加driver的属性到自己类里面
        return getattr(self.driver, item)


class _Test(unittest.TestCase):
    def test(self):
        driver_wrapper = DriverWrapper(1)
        driver_wrapper.open(‘https://www.baidu.com‘)  # 有人不喜欢用get，可以叫open什么的
        driver_wrapper.find_element_by_css_selector(‘#kw‘)  # 当类中存在方法，优先使用了自己类里面的方法，所以每次使用css选择器查找元素时候会打印一个日志
        driver_wrapper.find_element_by_id(‘kw‘)  # 当类中不存在此方法，使用Chrome类的方法

        driver_wrapper2 = DriverWrapper(1)  # 重新实例化一次这个类，仍然可以接着使用之前的浏览器，不会重新弹一个浏览器窗口
        driver_wrapper2.open(‘https://www.sina.com‘)
        driver_wrapper.logger.info(‘运行完了，现在想关闭浏览器‘)
        driver_wrapper2.driver.close()  # 这样做也可以，但不算是动态添加属性了，这是直接使用的该实例的driver属性的方法，driver属性是Chrome的一个实例。


if __name__ == ‘__main__‘:
    unittest.main()