1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 # @Time : 2019/10/14 23:37
4 # @Author : Tang Yiwei
5 # @Email : 892398433@qq.com
6 # @File : ObjectsMap.py
7 # @Software: PyCharm
8
9 import traceback
10 from selenium.webdriver.support.ui import WebDriverWait
11 from selenium.webdriver.common.by import By
12 from selenium.common.exceptions import NoSuchElementException,TimeoutException
13
14 def findElement(driver, selector, second=30):
15 """获取单个页面元素对象"""
16 wait = WebDriverWait(driver, second, 0.1)
17 try:
18 if '=>' not in selector:
19 element = wait.until(lambda x: x.find_element(by=By.ID, value=selector))
20 return element
21 locationType = selector.split('=>')[0]
22 locatorExpression = selector.split('=>')[1]
23 if locationType.lower() in ['i', 'id']:
24 element = wait.until(lambda x: x.find_element(by=By.ID, value=locatorExpression))
25 elif locationType.lower() in ['n', 'name']:
26 element = wait.until(lambda x: x.find_element(by=By.NAME, value=locatorExpression))
27 elif locationType.lower() in ['c', 'class']:
28 element = wait.until(lambda x: x.find_element(by=By.CLASS_NAME, value=locatorExpression))
29 elif locationType.lower() in ['l', 'link']:
30 element = wait.until(lambda x: x.find_element(by=By.LINK_TEXT, value=locatorExpression))
31 elif locationType.lower() in ['p', 'partial_link']:
32 element = wait.until(lambda x: x.find_element(by=By.PARTIAL_LINK_TEXT, value=locatorExpression))
33 elif locationType.lower() in ['t', 'tag_name']:
34 element = wait.until(lambda x: x.find_element(by=By.TAG_NAME, value=locatorExpression))
35 elif locationType.lower() in ['x', 'xpath']:
36 element = wait.until(lambda x: x.find_element(by=By.XPATH, value=locatorExpression))
37 elif locationType.lower() in ['s', 'css']:
38 element = wait.until(lambda x: x.find_element(by=By.CSS_SELECTOR, value=locatorExpression))
39 return element
40 except (NoSuchElementException, TimeoutException):
41 traceback.print_exc()
42 except Exception as e:
43 raise e
44
45
46 def findElements(driver, selector, second=30):
47 """获取多个相同页面元素对象，以list返回"""
48 wait = WebDriverWait(driver, second, 0.1)
49 try:
50 if '=>' not in selector:
51 elements = wait.until(lambda x: x.find_elements(by=By.ID, value=selector))
52 return elements
53 locationType = selector.split('=>')[0]
54 locatorExpression = selector.split('=>')[1]
55 if locationType.lower() in ['i', 'id']:
56 elements = wait.until(lambda x: x.find_elements(by=By.ID, value=locatorExpression))
57 elif locationType.lower() in ['n', 'name']:
58 elements = wait.until(lambda x: x.find_elements(by=By.NAME, value=locatorExpression))
59 elif locationType.lower() in ['c', 'class']:
60 elements = wait.until(lambda x: x.find_elements(by=By.CLASS_NAME, value=locatorExpression))
61 elif locationType.lower() in ['l', 'link']:
62 elements = wait.until(lambda x: x.find_elements(by=By.LINK_TEXT, value=locatorExpression))
63 elif locationType.lower() in ['p', 'partial_link']:
64 elements = wait.until(lambda x: x.find_elements(by=By.PARTIAL_LINK_TEXT, value=locatorExpression))
65 elif locationType.lower() in ['t', 'tag_name']:
66 elements = wait.until(lambda x: x.find_elements(by=By.TAG_NAME, value=locatorExpression))
67 elif locationType.lower() in ['x', 'xpath']:
68 elements = wait.until(lambda x: x.find_elements(by=By.XPATH, value=locatorExpression))
69 elif locationType.lower() in ['s', 'css']:
70 elements = wait.until(lambda x: x.find_elements(by=By.CSS_SELECTOR, value=locatorExpression))
71 return elements
72 except (NoSuchElementException, TimeoutException):
73 traceback.print_exc()
74 except Exception as e:
75 raise e
76
77
78 if __name__ == "__main__":
79 from selenium import webdriver
80 from config.VarConfig import chromeDriverFilePath
81 driver = webdriver.Chrome(executable_path=chromeDriverFilePath)
82 driver.get("http://www.baidu.com")
83 searchBox = findElement(driver,"id=>kw")
84 print(searchBox.tag_name)
85 aList = findElements(driver,"t=>a")
86 print(len(aList))
87 driver.quit()
