上一篇讲了数据分离，这一篇说一下元素定位方法的封装，我目前做的比较简陋

直接上代码，然后一行一行的给大家解释吧

from practice.Desired_Caps.Desired_caps import DesiredCaps
from selenium.webdriver.common.by import By


class GetElement(DesiredCaps):

    # 封装find_element方法
    def get_element(self, type_, value):

        __get_type = type_.lower()
        if __get_type == 'id':
            return self.driver.find_element(By.ID, value)
        elif __get_type == 'class_name':
            return self.driver.find_element(By.CLASS_NAME, value)
        elif __get_type == 'xpath':
            return self.driver.find_element(By.XPATH, value)

首先我导入了上一篇中封装好的驱动器对象，然后倒入了定位使用的By方法

然后我继承了DesiredCaps类，为什么这里我要选择继承呢，因为我需要使用里面定义的driver

然后我把ID,CLASS_NAME,XPATH三种方法进行了封装

设置type_和value两个参数，对用户使用的定位方法进行判断

设置私有变量(__两个下划线表示该变量是私有变量，只能在该函数中调用)，为了避免其他同事使用的时候写入的变量不做大小写区分，这里统一转换为小写

判断

当传入的type为'id'时，则使用self.driver.find_element(By.ID, value)方法，

当传入的type为'class_name'时，则使用self.driver.find_element(By.CLASS_NAME, value)方法

当传入的type为'xpath'时，则使用self.driver.find_element(By.XPATH, value)方法

这里用return的原因就是，我在元素定位的时候需要使用这个方法，所以必须return

这里封装之后有一个好处就是不用再去一直写driver.find_element了，看着密密麻麻的就很头大

其实我感觉封装了和之前差不多，哈哈哈，不过呢这么用确实方便的很多，下一篇元素定位操作的时候就会看到效果啦。

这里只是给大家举个例子，其他的定位方法也可以参考这样进行封装，里面的参数也可以换成一个*args，这样更方便点。