基础类封装-查找页面元素类库封装

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @Time    : 2019/10/14 23:37
# @Author  : Tang Yiwei
# @Email   : 892398433@qq.com
# @File    : ObjectsMap.py
# @Software: PyCharm

import traceback
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.common.exceptions import NoSuchElementException,TimeoutException

def findElement(driver, selector, second=30):
    """获取单个页面元素对象"""
    wait = WebDriverWait(driver, second, 0.1)
    try:
        if '=>' not in selector:
            element = wait.until(lambda x: x.find_element(by=By.ID, value=selector))
            return element
        locationType = selector.split('=>')[0]
        locatorExpression = selector.split('=>')[1]
        if locationType.lower() in ['i', 'id']:
            element = wait.until(lambda x: x.find_element(by=By.ID, value=locatorExpression))
        elif locationType.lower() in ['n', 'name']:
            element = wait.until(lambda x: x.find_element(by=By.NAME, value=locatorExpression))
        elif locationType.lower() in ['c', 'class']:
            element = wait.until(lambda x: x.find_element(by=By.CLASS_NAME, value=locatorExpression))
        elif locationType.lower() in ['l', 'link']:
            element = wait.until(lambda x: x.find_element(by=By.LINK_TEXT, value=locatorExpression))
        elif locationType.lower() in ['p', 'partial_link']:
            element = wait.until(lambda x: x.find_element(by=By.PARTIAL_LINK_TEXT, value=locatorExpression))
        elif locationType.lower() in ['t', 'tag_name']:
            element = wait.until(lambda x: x.find_element(by=By.TAG_NAME, value=locatorExpression))
        elif locationType.lower() in ['x', 'xpath']:
            element = wait.until(lambda x: x.find_element(by=By.XPATH, value=locatorExpression))
        elif locationType.lower() in ['s', 'css']:
            element = wait.until(lambda x: x.find_element(by=By.CSS_SELECTOR, value=locatorExpression))
        return element
    except (NoSuchElementException, TimeoutException):
        traceback.print_exc()
    except Exception as e:
        raise e


def findElements(driver, selector, second=30):
    """获取多个相同页面元素对象，以list返回"""
    wait = WebDriverWait(driver, second, 0.1)
    try:
        if '=>' not in selector:
            elements = wait.until(lambda x: x.find_elements(by=By.ID, value=selector))
            return elements
        locationType = selector.split('=>')[0]
        locatorExpression = selector.split('=>')[1]
        if locationType.lower() in ['i', 'id']:
            elements = wait.until(lambda x: x.find_elements(by=By.ID, value=locatorExpression))
        elif locationType.lower() in ['n', 'name']:
            elements = wait.until(lambda x: x.find_elements(by=By.NAME, value=locatorExpression))
        elif locationType.lower() in ['c', 'class']:
            elements = wait.until(lambda x: x.find_elements(by=By.CLASS_NAME, value=locatorExpression))
        elif locationType.lower() in ['l', 'link']:
            elements = wait.until(lambda x: x.find_elements(by=By.LINK_TEXT, value=locatorExpression))
        elif locationType.lower() in ['p', 'partial_link']:
            elements = wait.until(lambda x: x.find_elements(by=By.PARTIAL_LINK_TEXT, value=locatorExpression))
        elif locationType.lower() in ['t', 'tag_name']:
            elements = wait.until(lambda x: x.find_elements(by=By.TAG_NAME, value=locatorExpression))
        elif locationType.lower() in ['x', 'xpath']:
            elements = wait.until(lambda x: x.find_elements(by=By.XPATH, value=locatorExpression))
        elif locationType.lower() in ['s', 'css']:
            elements = wait.until(lambda x: x.find_elements(by=By.CSS_SELECTOR, value=locatorExpression))
        return elements
    except (NoSuchElementException, TimeoutException):
        traceback.print_exc()
    except Exception as e:
        raise e


if __name__ == "__main__":
    from selenium import webdriver
    from config.VarConfig import chromeDriverFilePath
    driver = webdriver.Chrome(executable_path=chromeDriverFilePath)
    driver.get("http://www.baidu.com")
    searchBox = findElement(driver,"id=>kw")
    print(searchBox.tag_name)
    aList = findElements(driver,"t=>a")
    print(len(aList))
    driver.quit()