重新封装WebDriver中的方法

一、继承WebDriver

思路就是继承selenium 的WebDriver（Chrome）库，然后重新封装里面的查找元素的函数，增加点log输出以及等待元素出现的逻辑，以find_element_by_id为例，先上源码

class Webdriver(Chrome):
    logger = get_logger('selenium')
    logger.setLevel(logging.INFO)

    def __init__(self, executable_path="chromedriver", port=0,
                 options=None, service_args=None,
                 desired_capabilities=None, service_log_path=None,
                 chrome_options=None):
        super(Webdriver, self).__init__(chrome_options=chrome_options, executable_path=executable_path,
                                        port=port, options=options, service_args=service_args,
                                        service_log_path=service_log_path, desired_capabilities=desired_capabilities)

    def find_element_by_id(self, el):
        try:
            self.logger.info("---------->正在定位{}".format(el))
            return WebDriverWait(self, 10).until(EC.visibility_of_element_located((By.ID, el)))
        except Exception:
            self.logger.error("{}定位失败".format(el))
            raise

二、重点

在继承WebDriver时，要对父类进行初始化，查看源码，得到__init__函数内初始化的变量以及赋值，所以在继承时初始化使用这些变量初始化，于是就有了如上的代码。完成继承后就可以重新封装想要扩展功能的函数，也可以新加别的功能的函数，同时也可以使用父类的函数，很是方便。

三、 tips

WebDriverWait函数通过查看源码得知第一个driver参数是WebDriver 的类型，自己新定义的类是继承自WebDriver 的，所以也是WebDriver类型的，因此传参的时候，传self即可，表示自身实例化对象。