經過by來定位元素，輔助ReadyamlToElement類使用

# -*- 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]


'''
驗證輸入的文件是否存在，有待改進
'''


class Getyamlfile:
    def file_name(self, file_dir):
        list_dirs = os.listdir(file_dir)       # 遍歷指定目錄下的全部文件名
        yamlfile = ((os.path.split(__file__)[-1].split(".")[0]) + ".yaml")   # 當前的py文件名+.yaml
        # print("如今" + (sys._getframe().f_code.co_filename).split('.')[0])
        if yamlfile in list_dirs:
            return yamlfile
        else:
            return None



"""
  經過key獲取到對應的elemet
  改進設想：將yaml文件先所有轉化爲對應的webelement類型
  key:WebElement  預計會出現的問題：1.頁面未加載完，2.經過點擊頁面元素纔出現的元素 3.frame標籤  這些可能會定位不到
  待驗證
"""


class FindWebElement:
    def __init__(self, driver, Type, Value):
        """
        :type driver: selenium.webdriver.remote.webdriver.WebDriver
        實例化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'查找相應的元素
        :return: WebElemt對應
        """
        """定位元素"""
        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("沒有找到對應元素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 + "'對應的元素;" + e.__str__() + ";請確保定位準確。")
                # self.get_img()  # 調用截圖
        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'查找相應的元素
        :return: 當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("沒有找到對應元素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 + "'對應的元素;" + e.__str__() + ";請確保定位準確。")
                # self.get_img()  # 調用截圖
        else:
            # print("輸入的元素定位方式錯誤")
            Log.error("輸入的元素定位方式錯誤,請使用'id', 'name', 'class', 'tag', 'link', 'plink', 'css', 'xpath'")