Python selenium封装元素定位FindElement工具类

# coding=utf-8
from config.setting_base import SettingBase
from util.read_ini import ReadIni
from selenium.webdriver.common.by import By
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as ES
from contextlib import contextmanager


class FindElement(object):
    def __init__(self, driver, file_name=None, node=None):
        self.driver = driver
        self.read_ini = ReadIni(file_name=file_name, node=node)
        self.locator = ()# 配置文件中定位元素的方式，如：id,name,xpath
    def get_by_key(self, key):
        data = self.read_ini.get_value(key)
        by_key = data.split('>')[0].strip()
        return by_key

    # 配置文件的value
    def get_value(self, key):
        data = self.read_ini.get_value(key)
        value = data.split('>')[1].strip()
        return valuedef selector_to_locator(self, key):
        selector_by = self.get_by_key(key)
        selector_value = self.get_value(key)
        if selector_by == 'id':
            locator = (By.ID, selector_value)
        elif selector_by == 'name':
            locator = (By.NAME, selector_value)
        elif selector_by == 'class_name':
            locator = (By.CLASS_NAME, selector_value)
        elif selector_by == 'link_text':
            locator = (By.PARTIAL_LINK_TEXT, selector_value)
        elif selector_by == 'tag_name':
            locator = (By.TAG_NAME, selector_value)
        elif selector_by == 'xpath':
            locator = (By.XPATH, selector_value)
        elif selector_by == 'css_selector':
            locator = (By.CSS_SELECTOR, selector_value)
        else:
            raise NameError("Please enter a valid selector of targeting elements.")
        return locator

    @contextmanager
    def find_base(self, key):
        if not isinstance(key, tuple):
            self.locator = self.selector_to_locator(key)
        else:
            self.locator = key
        if isinstance(self.locator, tuple):
            try:
                yield
            except:
                print(f"定位失败：定位方式->{self.locator[0]}, value值->{self.locator[1]}")
                return False

    def find(self, key, timeout=None, value=None):
        """
        页面查找单个元素
        :param key: 元素的元组(By, value)
        :param timeout: 超时时间
        :param value: 需要查找的值
        """
        with self.find_base(key):
            if timeout is None:
                timeout = SettingBase.UI_WAIT_TIME
            ele = self.__find_value(self.locator, timeout, value)
            return ele

    def not_find(self, key, timeout=None, value=None):
        """
        页面查找没有此元素
        :param key: 元素的元组(By, value)
        :param timeout: 超时时间
        :param value:需要查找的值
        """
        with self.find_base(key):
            if timeout is None:
                timeout = SettingBase.UI_WAIT_TIME
            self.__not_find_value(self.locator, timeout, value)

    def __find_value(self, locator, timeout, value):
        if value is None:
            # presence_of_element_located 不关心元素是否可见，只要元素存在在页面中即可
            ele = WebDriverWait(self.driver, timeout, SettingBase.POLL_FREQUENCY).until(
                 ES.presence_of_element_located(locator))
        else:
            # 判断元素中是否存在指定的文本，返回布尔值
            ele = WebDriverWait(self.driver, timeout, SettingBase.POLL_FREQUENCY).until(
                ES.text_to_be_present_in_element(locator, value))
        return ele

    def __not_find_value(self, locator, timeout, value):
        if value is None:
            # presence_of_element_located 不关心元素是否可见，只要元素存在在页面中即可
            bools = WebDriverWait(self.driver, timeout, SettingBase.POLL_FREQUENCY).until_not(
                ES.presence_of_element_located(locator))
        else:
            # 判断元素中是否存在指定的文本，返回布尔值
            bools = WebDriverWait(self.driver, timeout, SettingBase.POLL_FREQUENCY).until_not(
                ES.text_to_be_present_in_element(locator, value))
        return bools

    def finds(self, key, timeout=None):
        """
        页面查找多个元素
        :param key: 元素的元组(By, value)
        :param timeout: 超时时间
        :return: list or False
        """
        with self.find_base(key):
            if timeout is None:
                timeout = SettingBase.UI_WAIT_TIME
            # presence_of_all_elements_located 等待所有locator元素都加载出来，返回list
            eles = WebDriverWait(self.driver, timeout, SettingBase.POLL_FREQUENCY).until(
                ES.presence_of_all_elements_located(self.locator))
            return eles

    def find_title(self, key, timeout=None):
        """
        title中包含元素
        :param key: 元素的元组(By, value)
        :param timeout: 超时时间
        :return: True when the title matches, False otherwise
        """
        with self.find_base(key):
            if timeout is None:
                timeout = SettingBase.UI_WAIT_TIME
            # 判断元素中是否存在包含title，返回布尔值
            ele = WebDriverWait(self.driver, timeout, SettingBase.POLL_FREQUENCY).until(
                ES.title_contains(self.locator))
            return ele