定位元素操作

# -*- coding:utf-8 -*-
# date = 2019/5/13
# name = "yedeng"
# project_name = python
# explain : 從yaml文件中讀取返回webelement

import yaml
from selenium.webdriver.remote.webdriver import WebDriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.remote.webelement import WebElement
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import TimeoutException

from utils.FindWebElemet import FindWebElement
from utils.FindBy import FindBy
from utils.log import Log
from utils.ReadIni import Readini

# yaml文件位置
# Readin 讀取配置文件中的等待時間
'''
   初始化添加對應的yaml文件位置，然後調用getAllValue方法，調用 getelement方法獲取到對應的webelement
   改進方法:保證yaml的文件名和test_case的文件名或者類名一致
'''


class ReadyamlToElement:
    # 獲取當前腳本所在文件夾路徑

    def __init__(self, driver: WebDriver, yamlpath):
        """
        實例化readyaml類
        :param driver:
        :param yamlpath: 對應的yaml文件名
        """
        self.__yamlPath = yamlpath
        self.__driver = driver

    def __getAllValue(self):
        """
           將指定yaml文件轉化爲dic字典，key爲
           :return  dic字典
         """
        '''
        curPath = os.path.dirname(os.path.realpath(__file__))
        # 獲取yaml文件路徑
        yamlPath = os.path.join(curPath, "cfgyaml.yaml")
        print(type(yamlPath))
        '''
        with open(self.__yamlPath, 'r', encoding='utf-8') as f:
            cfg = f.read()

        dic_yaml = yaml.load(cfg, Loader=yaml.FullLoader)  # 用load方法轉字典
        return dic_yaml

    '''
       通過key獲取到對應的webelement，改進方法，將其全部遍歷後，生成key:element的字典，從裏面取對應的數據
    '''

    def getelement(self, key) -> WebElement:
        """
        通過yaml文件的key獲取到對應的Webelement
        :param key: yaml文件的key
        :return: 對應的WebElement對象
        :rtype keyelement: selenium.webdriver.remote.webelement
        """
        dic = ReadyamlToElement(self.__driver, self.__yamlPath).__getAllValue()  # 接受readyaml().getAllValue()返回的dict
        # print(dic)
        if key in dic:
            list_value = dic.get(key)
            # list_Type_value = list_value[0].split(":")[1]    #獲取type的值
            # list_Value_value = list_value[1].split(":")[1]   #獲取value的值
            list_Type_value = list_value.get("Type")  # 獲取type的值
            list_Value_value = list_value.get("Value")  # 獲取value的值
            keyelement = FindWebElement(self.__driver, list_Type_value,
                                        list_Value_value).find_element(self.__yamlPath, key)  # 用於返回webelement對象
            Log.info(self.__yamlPath + "中獲取到該element,通過該key:" + key)
            return keyelement
        else:
            Log.error(self.__yamlPath + "中不存在該key:" + key + ":關閉程序")
            # self.__driver.quit()

    def getelements(self, key) -> list:
        """
        通過yaml文件的key獲取到對應的Webelement
        :param key: yaml文件的key
        :return: 對應的WebElement對象 類型爲WebElement
        :rtype keyelements: list
        """
        dic = ReadyamlToElement(self.__driver, self.__yamlPath).__getAllValue()  # 接受readyaml().getAllValue()返回的dict
        # print(dic)
        if key in dic:
            list_value = dic.get(key)
            # list_Type_value = list_value[0].split(":")[1]    #獲取type的值
            # list_Value_value = list_value[1].split(":")[1]   #獲取value的值
            list_Type_value = list_value.get("Type")  # 獲取type的值
            list_Value_value = list_value.get("Value")  # 獲取value的值
            keyelementlist = FindWebElement(self.__driver, list_Type_value,
                                            list_Value_value).find_elements(self.__yamlPath, key)  # 用於返回webelement對象
            Log.info(self.__yamlPath + "中獲取到該element,通過該key:" + key)
            return keyelementlist
        else:
            Log.error(self.__yamlPath + "中不存在該key:" + key + ":關閉程序")
            # self.__driver.quit()

    def getelementsintnum(self, key, intnum=0) -> WebElement:
        """
        通過yaml文件的key獲取一組WebElemnts,通過數組下標獲取對應的WebElemnt
        :param key: yaml文件的key
        :param intnum 數組下標
        :type key : String
        :type intnum : int
        :return: 對應的WebElement對象
        :rtype keyelements[intnum]: selenium.webdriver.remote.webelement
        """
        dic = ReadyamlToElement(self.__driver, self.__yamlPath).__getAllValue()  # 接受readyaml().getAllValue()返回的dict
        # print(dic)
        list_keyelements = []
        if key in dic:
            list_value = dic.get(key)
            # list_Type_value = list_value[0].split(":")[1]    #獲取type的值
            # list_Value_value = list_value[1].split(":")[1]   #獲取value的值
            list_Type_value = list_value.get("Type")  # 獲取type的值
            list_Value_value = list_value.get("Value")  # 獲取value的值
            list_keyelements = FindWebElement(self.__driver, list_Type_value,
                                              list_Value_value).find_elements(self.__yamlPath, key)  # 用於返回webelement對象
            Log.info(self.__yamlPath + "中獲取到該element,通過該key:" + key)
        else:
            Log.error(self.__yamlPath + "中不存在該key:" + key + ":關閉程序")
            # self.__driver.quit()
            # 判斷輸入的值是否正確，正確則正確找到，否則就報空
        if -len(list_keyelements) <= intnum < len(list_keyelements):
            return list_keyelements[intnum]
        else:
            Log.error("當前輸入的intnum大於" + str(len(list_keyelements)) + "有問題")
            return None

    def getelementsstringtext(self, key, strtext: str, attributename=None) -> WebElement:
            """
            通過yaml文件的key獲取一組WebElemnts,通過指定元素的值獲取到對應的WebElemnt
            :param key: yaml文件的key
            :param strtext 對應元素的值
            :type key : str
            :type strtext : str
            :param attributename 元素的屬性值，Name of the attribute/property to retrieve
            :type attributename : str
            :return: 對應的WebElement對象
            :rtype element: selenium.webdriver.remote.webelement
            """
            dic = ReadyamlToElement(self.__driver, self.__yamlPath).__getAllValue()  # 接受readyaml().getAllValue()返回的dict
            # print(dic)
            list_keyelements = []
            if key in dic:
                list_value = dic.get(key)
                # list_Type_value = list_value[0].split(":")[1]    #獲取type的值
                # list_Value_value = list_value[1].split(":")[1]   #獲取value的值
                list_Type_value = list_value.get("Type")  # 獲取type的值
                list_Value_value = list_value.get("Value")  # 獲取value的值
                list_keyelements = FindWebElement(self.__driver, list_Type_value,
                                                  list_Value_value).find_elements(self.__yamlPath, key)  # 用於返回webelement對象
                Log.info(self.__yamlPath + "中獲取到該element,通過該key:" + key)
            else:
                Log.error(self.__yamlPath + "中不存在該key:" + key + ":關閉程序")
                # self.__driver.quit()
                # 判斷輸入的值是否正確，正確則正確找到，否則就報空
            booleanfind = False
            element = None
            # 當沒有設備屬性值時用元素的text值
            if attributename is None:
                for webelement in list_keyelements:
                    if webelement.text == strtext:
                        element = webelement
                        booleanfind = True
                        break
            # 當設備屬性值時用元素的指定的屬性值的內容
            else:
                for webelement in list_keyelements:
                    if webelement.get_attribute(attributename) == strtext:
                        element = webelement
                        booleanfind = True
                        break
        
            if booleanfind:
                return element
            else:
                Log.error(self.__yamlPath + "中未找到對應的元素值，請確認" + strtext + "的正確")
                return element

    def waitgetelement(self, key):
        """
        通過yaml文件的key獲取到對應的Webelement,加入了顯示等待，WeblEment.is_displayed()爲真時返回該該webElement
        :param key:  yaml文件的key
        :return: 對應的WebElement對象
        """
        try:
            inttime = Readini().getvalue("webdriver", "implicit_waitTime")
            WebDriverWait(self, int(inttime)).until(lambda x: x.getelement(key).is_displayed())
            return self.getelement(key)
        except Exception as e:
            Log.error(self.__yamlPath + "中該元素不存在:" + key + "導致錯誤:" + e.__str__())
            # self.__driver.quit()

    def waitgetelements(self, key):
        """
        通過yaml文件的key獲取到對應的Webelement,加入了顯示等待，WeblEment.is_displayed()爲真時返回該該webElement
        :param key:  yaml文件的key
        :return: 對應的WebElement對象
        """
        try:
            inttime = Readini().getvalue("webdriver", "implicit_waitTime")
            WebDriverWait(self, int(inttime)).until(lambda x: x.getelement(key).is_displayed())
            return self.getelements(key)
        except Exception as e:
            Log.error(self.__yamlPath + "中該元素不存在:" + key + "導致錯誤:" + e.__str__())
            # self.__driver.quit()

    def waitgetelement_clickable(self, key) -> WebElement:
        """
        通過yaml文件的key獲取到對應的Webelement,當元素可以點擊時返回webelement
        :param key: yaml文件的key
        :return: 對應的WebElement對象
        :rtype keyelement: selenium.webdriver.remote.webelement
        """
        dic = ReadyamlToElement(self.__driver, self.__yamlPath).__getAllValue()  # 接受readyaml().getAllValue()返回的dict
        if key in dic:
            list_value = dic.get(key)
            list_Type_value = list_value.get("Type")  # 獲取type的值
            list_Value_value = list_value.get("Value")  # 獲取value的值
            try:
                # 顯式等待時間
                inttime = Readini().getvalue("webdriver", "implicit_waitTime")
                # 配合expected_conditions當元素可以點擊時返回element
                elemnet = WebDriverWait(self.__driver, int(inttime)).until(EC.element_to_be_clickable((
                    FindBy(list_Type_value).find_by(), list_Value_value)), (self.__yamlPath + "中No such element : 請檢查該‘" + key +
                                                                            "’對應下的Value值是否能在頁面上定位成功"))
                return elemnet
            except TimeoutException as e:
                Log.error(e.__str__())

    def getlocator(self, key) -> tuple:
        """
        返回location 元組類型的值，主要給EC模塊入參使用
        :param key : yaml文件的key
        :return: tup_locator 元組類型的值如(By.Id,"value")
        :rtype tup_locator : tuple
        """
        dic = ReadyamlToElement(self.__driver, self.__yamlPath).__getAllValue()  # 接受readyaml().getAllValue()返回的dict
        if key in dic:
            list_value = dic.get(key)
            # list_Type_value = list_value[0].split(":")[1]    #獲取type的值
            # list_Value_value = list_value[1].split(":")[1]   #獲取value的值
            list_Type_value = list_value.get("Type")  # 獲取type的值
            list_Value_value = list_value.get("Value")  # 獲取value的值
            tup_locator = (FindBy(list_Type_value).find_by(), list_Value_value)
            if len(tup_locator) is None:
                Log.error(self.__yamlPath + "中請檢查該‘" + key + "’的值是否正確")
            else:
                return tup_locator