如何通過by定位元素輔助ReadyamlToElement類使用

如何通過by定位元素輔助ReadyamlToElement類使用？這篇文章運(yùn)用了實(shí)例代碼展示，代碼非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下。

# -*- coding:utf-8 -*-
# date = 2019/5/13
# name = "yedeng"
# project_name = python
# explain : WebElement元素定位，給GetelementFromYaml使用

import inspect, os
from utils.log import Log
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.remote import webelement


def get_current_function_name():
    return inspect.stack()[1][3]


'''
驗(yàn)證輸入的文件是否存在，有待改進(jìn)
'''


class Getyamlfile:
    def file_name(self, file_dir):
        list_dirs = os.listdir(file_dir)       # 遍歷指定目錄下的所有文件名
        yamlfile = ((os.path.split(__file__)[-1].split(".")[0]) + ".yaml")   # 當(dāng)前的py文件名+.yaml
        # print("現(xiàn)在" + (sys._getframe().f_code.co_filename).split('.')[0])
        if yamlfile in list_dirs:
            return yamlfile
        else:
            return None



"""
  通過key獲取到對應(yīng)的elemet
  改進(jìn)設(shè)想：將yaml文件先全部轉(zhuǎn)化為對應(yīng)的webelement類型
  key:WebElement  預(yù)計會出現(xiàn)的問題：1.頁面未加載完，2.通過點(diǎn)擊頁面元素才出現(xiàn)的元素 3.frame標(biāo)簽  這些可能會定位不到
  待驗(yàn)證
"""


class FindWebElement:
    def __init__(self, driver, Type, Value):
        """
        :type driver: selenium.webdriver.remote.webdriver.WebDriver
        實(shí)例化findWebElement
        :param driver:
        :param Type: yaml文件的Type值
        :param Value: yaml文件的value值
        """
        self.__driver = driver
        self.__Type = Type
        self.__Value = Value
    # 定義查找元素

    def find_element(self, yamlpath, key) -> webelement:
        """
        通過'id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath'查找相應(yīng)的元素
        :return: WebElemt對應(yīng)
        """
        """定位元素"""
        by = self.__Type
        value = self.__Value
        element = None
        if by in ['id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath']:
                # noinspection PyBroadException
            try:
                if by == 'id':
                    element = self.__driver.find_element_by_id(value)
                elif by == 'name':
                    element = self.__driver.find_element_by_name(value)
                elif by == 'class':
                    element = self.__driver.find_element_by_class_name(value)
                elif by == 'tag':
                    element = self.__driver.find_element_by_tag_name(value)
                elif by == 'link':
                    element = self.__driver.find_element_by_link_text(value)
                elif by == 'plink':
                    element = self.__driver.find_element_by_partial_link_text(value)
                elif by == 'css':
                    element = self.__driver.find_element_by_css_selector(value)
                elif by == 'xpath':
                    element = self.__driver.find_element_by_xpath(value)
                else:
                    Log.error("沒有找到對應(yīng)元素Type類型，請使用'id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath'")
                # Log.info('元素定位成功。定位方式：%s，使用的值%s：' % (by, value))
                return element
            except NoSuchElementException as e:
                # print("no suceelement")
                Log.error("yaml文件為:" + yamlpath + ";頁面中沒有定位key值為'" + key + "'對應(yīng)的元素;" + e.__str__() + ";請確保定位準(zhǔn)確。")
                # self.get_img()  # 調(diào)用截圖
        else:
            # print("輸入的元素定位方式錯誤")
            Log.error("輸入的元素定位方式錯誤,請使用'id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath'")

    def find_elements(self, yamlpath, key) -> list:
        """
        通過'id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath'查找相應(yīng)的元素
        :return: 當(dāng)list_elements只有1個時直接返回selenium.webdriver.remote.webelement類，否則返回list
        :rtype list_elements : list of WebElement
        """
        """定位元素"""
        by = self.__Type
        value = self.__Value
        list_elements = None
        if by in ['id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath']:
                # noinspection PyBroadException
            try:
                if by == 'id':
                    list_elements = self.__driver.find_elements_by_id(value)
                elif by == 'name':
                    list_elements = self.__driver.find_elements_by_name(value)
                elif by == 'class':
                    list_elements = self.__driver.find_elements_by_class_name(value)
                elif by == 'tag':
                    list_elements = self.__driver.find_elements_by_tag_name(value)
                elif by == 'link':
                    list_elements = self.__driver.find_elements_by_link_text(value)
                elif by == 'plink':
                    list_elements = self.__driver.find_elements_by_partial_link_text(value)
                elif by == 'css':
                    list_elements = self.__driver.find_elements_by_css_selector(value)
                elif by == 'xpath':
                    list_elements = self.__driver.find_elements_by_xpath(value)
                else:
                    Log.error("沒有找到對應(yīng)元素Type類型，請使用'id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath'")
                # Log.info('元素定位成功。定位方式：%s，使用的值%s：' % (by, value))
                return list_elements
            except NoSuchElementException as e:
                # print("no suceelement")
                Log.error("yaml文件為:" + yamlpath + ";頁面中沒有定位key值為'" + key + "'對應(yīng)的元素;" + e.__str__() + ";請確保定位準(zhǔn)確。")
                # self.get_img()  # 調(diào)用截圖
        else:
            # print("輸入的元素定位方式錯誤")
            Log.error("輸入的元素定位方式錯誤,請使用'id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath'")

上述就是by定位元素輔助ReadyamlToElement類使用的詳細(xì)內(nèi)容，詳細(xì)使用情況還需要大家自己動手實(shí)驗(yàn)使用過才能領(lǐng)會。如果想了解更多，歡迎關(guān)注億速云行業(yè)資訊頻道！