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 : WaitUtil.py
7 # @Software: PyCharm
8
9 from selenium.webdriver.support import expected_conditions as EC
10 from util.ObjectMap import *
11 from selenium.common.exceptions import NoSuchElementException,TimeoutException
12
13 class WaitUtil():
14
15 def __init__(self,driver):
16 self.driver = driver
17 self.wait = WebDriverWait(self.driver,30)
18
19 def presence_of_element_located(self,selector):
20 """显示的等待页面元素出现在DOM中,但不一定可见,存在则返回该元素的页面对象,定位器传入元祖类型"""
21 try:
22 if '=>' not in selector:
23 element = self.wait.until(EC.presence_of_element_located(findElement(self.driver, selector)))
24 return element
25 else:
26 locationType = selector.split('=>')[0]
27 locatorExpression = selector.split('=>')[1]
28 if locationType.lower() in ['i', 'id']:
29 element = self.wait.until(EC.presence_of_element_located((By.ID, locatorExpression)))
30 elif locationType.lower() in ['n', 'name']:
31 element = self.wait.until(EC.presence_of_element_located((By.NAME, locatorExpression)))
32 elif locationType.lower() in ['c', 'class']:
33 element = self.wait.until(EC.presence_of_element_located((By.CLASS_NAME, locatorExpression)))
34 elif locationType.lower() in ['l', 'link']:
35 element = self.wait.until(EC.presence_of_element_located((By.LINK_TEXT, locatorExpression)))
36 elif locationType.lower() in ['p', 'partial_link']:
37 element = self.wait.until(EC.presence_of_element_located((By.PARTIAL_LINK_TEXT, locatorExpression)))
38 elif locationType.lower() in ['t', 'tag_name']:
39 element = self.wait.until(EC.presence_of_element_located((By.TAG_NAME, locatorExpression)))
40 elif locationType.lower() in ['x', 'xpath']:
41 element = self.wait.until(EC.presence_of_element_located((By.XPATH, locatorExpression)))
42 elif locationType.lower() in ['s', 'css']:
43 element = self.wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, locatorExpression)))
44 return element
45 except TimeoutException as e:
46 print(traceback.print_exc())
47 except NoSuchElementException as e:
48 print(traceback.print_exc())
49 except Exception as e:
50 raise e
51
52 def frame_to_be_available_and_switch_to_it(self,selector):
53 """检查Frame是否存在,存在则切换进frame控件中，定位器传入元祖类型"""
54 try:
55 if '=>' not in selector:
56 self.wait.until(EC.frame_to_be_available_and_switch_to_it((By.ID,selector)))
57 else:
58 locationType = selector.split('=>')[0]
59 locatorExpression = selector.split('=>')[1]
60 if locationType.lower() in ['i', 'id']:
61 self.wait.until(EC.frame_to_be_available_and_switch_to_it((By.ID, locatorExpression)))
62 elif locationType.lower() in ['n', 'name']:
63 self.wait.until(EC.frame_to_be_available_and_switch_to_it((By.NAME, locatorExpression)))
64 elif locationType.lower() in ['c', 'class']:
65 self.wait.until(EC.frame_to_be_available_and_switch_to_it((By.CLASS_NAME, locatorExpression)))
66 elif locationType.lower() in ['l', 'link']:
67 self.wait.until(EC.frame_to_be_available_and_switch_to_it((By.LINK_TEXT, locatorExpression)))
68 elif locationType.lower() in ['p', 'partial_link']:
69 self.wait.until(EC.frame_to_be_available_and_switch_to_it((By.PARTIAL_LINK_TEXT, locatorExpression)))
70 elif locationType.lower() in ['t', 'tag_name']:
71 self.wait.until(EC.frame_to_be_available_and_switch_to_it((By.TAG_NAME, locatorExpression)))
72 elif locationType.lower() in ['x', 'xpath']:
73 self.wait.until(EC.frame_to_be_available_and_switch_to_it((By.XPATH, locatorExpression)))
74 elif locationType.lower() in ['s', 'css']:
75 self.wait.until(EC.frame_to_be_available_and_switch_to_it((By.CSS_SELECTOR, locatorExpression)))
76 except TimeoutException as e:
77 print(traceback.print_exc())
78 except NoSuchElementException as e:
79 print(traceback.print_exc())
80 except Exception as e:
81 raise e
82
83 def visibility_of_element_located(self,selector):
84 """显示等待页面元素的出现在DOM中,并且可见,存在则返回该页面元素对象，定位器传入元祖类型"""
85 try:
86 if '=>' not in selector:
87 element = self.wait.until(EC.visibility_of_element_located((By.ID,selector)))
88 return element
89 else:
90 locationType = selector.split('=>')[0]
91 locatorExpression = selector.split('=>')[1]
92 if locationType.lower() in ['i', 'id']:
93 element = self.wait.until(EC.visibility_of_element_located((By.ID, locatorExpression)))
94 elif locationType.lower() in ['n', 'name']:
95 element = self.wait.until(EC.visibility_of_element_located((By.NAME, locatorExpression)))
96 elif locationType.lower() in ['c', 'class']:
97 element = self.wait.until(EC.visibility_of_element_located((By.CLASS_NAME, locatorExpression)))
98 elif locationType.lower() in ['l', 'link']:
99 element = self.wait.until(EC.visibility_of_element_located((By.LINK_TEXT, locatorExpression)))
100 elif locationType.lower() in ['p', 'partial_link']:
101 element = self.wait.until(EC.visibility_of_element_located((By.PARTIAL_LINK_TEXT, locatorExpression)))
102 elif locationType.lower() in ['t', 'tag_name']:
103 element = self.wait.until(EC.visibility_of_element_located((By.TAG_NAME, locatorExpression)))
104 elif locationType.lower() in ['x', 'xpath']:
105 element = self.wait.until(EC.visibility_of_element_located((By.XPATH, locatorExpression)))
106 elif locationType.lower() in ['s', 'css']:
107 element = self.wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, locatorExpression)))
108 return element
109 except TimeoutException as e:
110 print(traceback.print_exc())
111 except NoSuchElementException as e:
112 print(traceback.print_exc())
113 except Exception as e:
114 raise e
115
116 def alert_is_present(self):
117 """判断弹框是否弹出"""
118 try:
119 alertText = self.wait.until(EC.alert_is_present()).text
120 return alertText
121 except Exception as e:
122 raise e
123
124 def element_located_selection_state_to_be(self, selector, state):
125 """
126 判断一个元素的状态是否是给定的选择状态，第一个传入的参数是一个定位器，定位器是一个元组（by,xpath）;
127 第二个元素表示状态，True表示选中状态，False表示未选中状态；相等返回True,不相等返回False;
128 """
129 try:
130 if '=>' not in selector:
131 element = self.wait.until(EC.element_located_selection_state_to_be((By.ID,selector),state))
132 return element
133 else:
134 locationType = selector.split('=>')[0]
135 locatorExpression = selector.split('=>')[1]
136 if locationType.lower() in ['i', 'id']:
137 element = self.wait.until(EC.element_located_selection_state_to_be((By.ID, locatorExpression),state))
138 elif locationType.lower() in ['n', 'name']:
139 element = self.wait.until(EC.element_located_selection_state_to_be((By.NAME, locatorExpression),state))
140 elif locationType.lower() in ['c', 'class']:
141 element = self.wait.until(EC.element_located_selection_state_to_be((By.CLASS_NAME, locatorExpression),state))
142 elif locationType.lower() in ['l', 'link']:
143 element = self.wait.until(EC.element_located_selection_state_to_be((By.LINK_TEXT, locatorExpression),state))
144 elif locationType.lower() in ['p', 'partial_link']:
145 element = self.wait.until(EC.element_located_selection_state_to_be((By.PARTIAL_LINK_TEXT, locatorExpression),state))
146 elif locationType.lower() in ['t', 'tag_name']:
147 element = self.wait.until(EC.element_located_selection_state_to_be((By.TAG_NAME, locatorExpression),state))
148 elif locationType.lower() in ['x', 'xpath']:
149 element = self.wait.until(EC.element_located_selection_state_to_be((By.XPATH, locatorExpression),state))
150 elif locationType.lower() in ['s', 'css']:
151 element = self.wait.until(EC.element_located_selection_state_to_be((By.CSS_SELECTOR, locatorExpression),state))
152 return element
153 except TimeoutException as e:
154 print(traceback.print_exc())
155 except NoSuchElementException as e:
156 print(traceback.print_exc())
157 except Exception as e:
158 raise e
159
160 def element_selection_state_to_be(self,selector,state):
161 """
162 判断给定的元素是否被选中，第一个参数是WebDriver对象;
163 第二个参数是期望的元素状态，相等返回True,否则返回False;
164 """
165 try:
166 self.wait.until(EC.element_selection_state_to_be(findElement(self.driver,selector),state))
167 except TimeoutException as e:
168 print(traceback.print_exc())
169 except NoSuchElementException as e:
170 print(traceback.print_exc())
171 except Exception as e:
172 raise e
173
174 def element_to_be_selected(self,selector):
175 """
176 期望元素处于选中状态，参数为一个WebDriver实例对象
177 """
178 try:
179 element = self.wait.until(EC.element_to_be_selected(findElement(self.driver,selector)))
180 return element
181 except TimeoutException as e:
182 print(traceback.print_exc())
183 except NoSuchElementException as e:
184 print(traceback.print_exc())
185 except Exception as e:
186 raise e
187
188 def element_to_be_clickable(self,selector):
189 """
190 判断元素是否可见并且能被点击，条件满足，返回该元素的页面对象，否则返回False,传入元祖类型定位器
191 """
192 try:
193 if '=>' not in selector:
194 element = self.wait.until(EC.element_to_be_clickable((By.ID, selector)))
195 return element
196 else:
197 locationType = selector.split('=>')[0]
198 locatorExpression = selector.split('=>')[1]
199 if locationType.lower() in ['i', 'id']:
200 element = self.wait.until(EC.element_to_be_clickable((By.ID, locatorExpression)))
201 elif locationType.lower() in ['n', 'name']:
202 element = self.wait.until(EC.element_to_be_clickable((By.NAME, locatorExpression)))
203 elif locationType.lower() in ['c', 'class']:
204 element = self.wait.until(EC.element_to_be_clickable((By.CLASS_NAME, locatorExpression)))
205 elif locationType.lower() in ['l', 'link']:
206 element = self.wait.until(EC.element_to_be_clickable((By.LINK_TEXT, locatorExpression)))
207 elif locationType.lower() in ['p', 'partial_link']:
208 element = self.wait.until(EC.element_to_be_clickable((By.PARTIAL_LINK_TEXT, locatorExpression)))
209 elif locationType.lower() in ['t', 'tag_name']:
210 element = self.wait.until(EC.element_to_be_clickable((By.TAG_NAME, locatorExpression)))
211 elif locationType.lower() in ['x', 'xpath']:
212 element = self.wait.until(EC.element_to_be_clickable((By.XPATH, locatorExpression)))
213 elif locationType.lower() in ['s', 'css']:
214 element = self.wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR, locatorExpression)))
215 return element
216 except TimeoutException as e:
217 print(traceback.print_exc())
218 except NoSuchElementException as e:
219 print(traceback.print_exc())
220 except Exception as e:
221 raise e
222
223 def invisibility_of_element_located(self,selector):
224 """
225 希望某个元素不可见或者不存在于DOM中，满足条件返回True,否则返回定位到的元素对象，定位器传入元祖类型
226 """
227 try:
228 if '=>' not in selector:
229 element = self.wait.until(EC.invisibility_of_element_located((By.ID, selector)))
230 return element
231 else:
232 locationType = selector.split('=>')[0]
233 locatorExpression = selector.split('=>')[1]
234 if locationType.lower() in ['i', 'id']:
235 element = self.wait.until(EC.invisibility_of_element_located((By.ID, locatorExpression)))
236 elif locationType.lower() in ['n', 'name']:
237 element = self.wait.until(EC.invisibility_of_element_located((By.NAME, locatorExpression)))
238 elif locationType.lower() in ['c', 'class']:
239 element = self.wait.until(EC.invisibility_of_element_located((By.CLASS_NAME, locatorExpression)))
240 elif locationType.lower() in ['l', 'link']:
241 element = self.wait.until(EC.invisibility_of_element_located((By.LINK_TEXT, locatorExpression)))
242 elif locationType.lower() in ['p', 'partial_link']:
243 element = self.wait.until(EC.invisibility_of_element_located((By.PARTIAL_LINK_TEXT, locatorExpression)))
244 elif locationType.lower() in ['t', 'tag_name']:
245 element = self.wait.until(EC.invisibility_of_element_located((By.TAG_NAME, locatorExpression)))
246 elif locationType.lower() in ['x', 'xpath']:
247 element = self.wait.until(EC.invisibility_of_element_located((By.XPATH, locatorExpression)))
248 elif locationType.lower() in ['s', 'css']:
249 element = self.wait.until(EC.invisibility_of_element_located((By.CSS_SELECTOR, locatorExpression)))
250 return element
251 except TimeoutException as e:
252 print(traceback.print_exc())
253 except NoSuchElementException as e:
254 print(traceback.print_exc())
255 except Exception as e:
256 raise e
257
258 def visibility_of(self,selector):
259 """
260 希望某个元素出现在页面的DOM中，并且可见，如果满足条件返回该元素的页面元素对象，传入WebElement对象
261 """
262 try:
263 element = self.wait.until(EC.visibility_of((findElement(self.driver,selector))))
264 return element
265 except TimeoutException as e:
266 print(traceback.print_exc())
267 except NoSuchElementException as e:
268 print(traceback.print_exc())
269 except Exception as e:
270 raise e
271
272 def visibility_of_any_elements_located(self,selector):
273 """
274 判断页面上至少有一个元素可见，返回满足条件的所有页面元素对象,传入元祖类型的定位器
275 """
276 try:
277 if '=>' not in selector:
278 element = self.wait.until(EC.visibility_of_any_elements_located((By.ID, selector)))
279 return element
280 else:
281 locationType = selector.split('=>')[0]
282 locatorExpression = selector.split('=>')[1]
283 if locationType.lower() in ['i', 'id']:
284 element = self.wait.until(EC.visibility_of_any_elements_located((By.ID, locatorExpression)))
285 elif locationType.lower() in ['n', 'name']:
286 element = self.wait.until(EC.visibility_of_any_elements_located((By.NAME, locatorExpression)))
287 elif locationType.lower() in ['c', 'class']:
288 element = self.wait.until(EC.visibility_of_any_elements_located((By.CLASS_NAME, locatorExpression)))
289 elif locationType.lower() in ['l', 'link']:
290 element = self.wait.until(EC.visibility_of_any_elements_located((By.LINK_TEXT, locatorExpression)))
291 elif locationType.lower() in ['p', 'partial_link']:
292 element = self.wait.until(EC.visibility_of_any_elements_located((By.PARTIAL_LINK_TEXT, locatorExpression)))
293 elif locationType.lower() in ['t', 'tag_name']:
294 element = self.wait.until(EC.visibility_of_any_elements_located((By.TAG_NAME, locatorExpression)))
295 elif locationType.lower() in ['x', 'xpath']:
296 element = self.wait.until(EC.visibility_of_any_elements_located((By.XPATH, locatorExpression)))
297 elif locationType.lower() in ['s', 'css']:
298 element = self.wait.until(EC.visibility_of_any_elements_located((By.CSS_SELECTOR, locatorExpression)))
299 return element
300 except TimeoutException as e:
301 print(traceback.print_exc())
302 except NoSuchElementException as e:
303 print(traceback.print_exc())
304 except Exception as e:
305 raise e
306
307 def presence_of_all_elements_located(self,selector):
308 """
309 判断页面上至少有一个元素出现，如果满足条件返回所有满足定位表达式的页面元素，传入元祖类型定位器
310 """
311 try:
312 if '=>' not in selector:
313 element = self.wait.until(EC.presence_of_all_elements_located((By.ID, selector)))
314 return element
315 else:
316 locationType = selector.split('=>')[0]
317 locatorExpression = selector.split('=>')[1]
318 if locationType.lower() in ['i', 'id']:
319 element = self.wait.until(EC.presence_of_all_elements_located((By.ID, locatorExpression)))
320 elif locationType.lower() in ['n', 'name']:
321 element = self.wait.until(EC.presence_of_all_elements_located((By.NAME, locatorExpression)))
322 elif locationType.lower() in ['c', 'class']:
323 element = self.wait.until(EC.presence_of_all_elements_located((By.CLASS_NAME, locatorExpression)))
324 elif locationType.lower() in ['l', 'link']:
325 element = self.wait.until(EC.presence_of_all_elements_located((By.LINK_TEXT, locatorExpression)))
326 elif locationType.lower() in ['p', 'partial_link']:
327 element = self.wait.until(EC.presence_of_all_elements_located((By.PARTIAL_LINK_TEXT, locatorExpression)))
328 elif locationType.lower() in ['t', 'tag_name']:
329 element = self.wait.until(EC.presence_of_all_elements_located((By.TAG_NAME, locatorExpression)))
330 elif locationType.lower() in ['x', 'xpath']:
331 element = self.wait.until(EC.presence_of_all_elements_located((By.XPATH, locatorExpression)))
332 elif locationType.lower() in ['s', 'css']:
333 element = self.wait.until(EC.presence_of_all_elements_located((By.CSS_SELECTOR, locatorExpression)))
334 return element
335 except TimeoutException as e:
336 print(traceback.print_exc())
337 except NoSuchElementException as e:
338 print(traceback.print_exc())
339 except Exception as e:
340 raise e
341
342 def staleness_of(self,selector):
343 """
344 判断一个元素是否仍在DOM中，如果在规定的时间内已经移出，返回True,否则返回False，传入WebElement对象
345 """
346 try:
347 element = self.wait.until(EC.staleness_of((findElement(self.driver,selector))))
348 return element
349 except TimeoutException as e:
350 print(traceback.print_exc())
351 except NoSuchElementException as e:
352 print(traceback.print_exc())
353 except Exception as e:
354 raise e
355
356 def text_to_be_present_in_element(self,selector,text):
357 """
358 判断文本内容text是否出现在某个元素中，判断的是元素的text,定位器是一个元祖
359 """
360 try:
361 if '=>' not in selector:
362 element = self.wait.until(EC.text_to_be_present_in_element((By.ID, selector),text))
363 return element
364 else:
365 locationType = selector.split('=>')[0]
366 locatorExpression = selector.split('=>')[1]
367 if locationType.lower() in ['i', 'id']:
368 element = self.wait.until(EC.text_to_be_present_in_element((By.ID, locatorExpression),text))
369 elif locationType.lower() in ['n', 'name']:
370 element = self.wait.until(EC.text_to_be_present_in_element((By.NAME, locatorExpression),text))
371 elif locationType.lower() in ['c', 'class']:
372 element = self.wait.until(EC.text_to_be_present_in_element((By.CLASS_NAME, locatorExpression),text))
373 elif locationType.lower() in ['l', 'link']:
374 element = self.wait.until(EC.text_to_be_present_in_element((By.LINK_TEXT, locatorExpression),text))
375 elif locationType.lower() in ['p', 'partial_link']:
376 element = self.wait.until(EC.text_to_be_present_in_element((By.PARTIAL_LINK_TEXT, locatorExpression),text))
377 elif locationType.lower() in ['t', 'tag_name']:
378 element = self.wait.until(EC.text_to_be_present_in_element((By.TAG_NAME, locatorExpression),text))
379 elif locationType.lower() in ['x', 'xpath']:
380 element = self.wait.until(EC.text_to_be_present_in_element((By.XPATH, locatorExpression),text))
381 elif locationType.lower() in ['s', 'css']:
382 element = self.wait.until(EC.text_to_be_present_in_element((By.CSS_SELECTOR, locatorExpression),text))
383 return element
384 except TimeoutException as e:
385 print(traceback.print_exc())
386 except NoSuchElementException as e:
387 print(traceback.print_exc())
388 except Exception as e:
389 raise e
390
391 def text_to_be_present_in_element_value(self,selector,text):
392 """
393 判断text是否出现在元素的value属性值中，定位器是一个元祖
394 """
395 try:
396 if '=>' not in selector:
397 element = self.wait.until(EC.text_to_be_present_in_element_value((By.ID, selector),text))
398 return element
399 else:
400 locationType = selector.split('=>')[0]
401 locatorExpression = selector.split('=>')[1]
402 if locationType.lower() in ['i', 'id']:
403 element = self.wait.until(EC.text_to_be_present_in_element_value((By.ID, locatorExpression),text))
404 elif locationType.lower() in ['n', 'name']:
405 element = self.wait.until(EC.text_to_be_present_in_element_value((By.NAME, locatorExpression),text))
406 elif locationType.lower() in ['c', 'class']:
407 element = self.wait.until(EC.text_to_be_present_in_element_value((By.CLASS_NAME, locatorExpression),text))
408 elif locationType.lower() in ['l', 'link']:
409 element = self.wait.until(EC.text_to_be_present_in_element_value((By.LINK_TEXT, locatorExpression),text))
410 elif locationType.lower() in ['p', 'partial_link']:
411 element = self.wait.until(EC.text_to_be_present_in_element_value((By.PARTIAL_LINK_TEXT, locatorExpression),text))
412 elif locationType.lower() in ['t', 'tag_name']:
413 element = self.wait.until(EC.text_to_be_present_in_element_value((By.TAG_NAME, locatorExpression),text))
414 elif locationType.lower() in ['x', 'xpath']:
415 element = self.wait.until(EC.text_to_be_present_in_element_value((By.XPATH, locatorExpression),text))
416 elif locationType.lower() in ['s', 'css']:
417 element = self.wait.until(EC.text_to_be_present_in_element_value((By.CSS_SELECTOR, locatorExpression),text))
418 return element
419 except TimeoutException as e:
420 print(traceback.print_exc())
421 except NoSuchElementException as e:
422 print(traceback.print_exc())
423 except Exception as e:
424 raise e
425
426 def new_window_is_opened(self,handles):
427 """
428 判断窗口是否打开
429 :param handles:窗口句柄， 可传self.driver.window_handles,用于click后打开新窗口后的判断
430 :return:True or False
431 """
432 return self.wait.until(EC.new_window_is_opened(handles))
433
434
435 def title_contains(self,partial_title):
436 """
437 判断页面title标签内容是否包含partial_title，只需要部分匹配即可，包含返回True,否则返回False
438 """
439 try:
440 return self.wait.until(EC.title_contains(partial_title))
441 except TimeoutException as e:
442 print(traceback.print_exc())
443 except NoSuchElementException as e:
444 print(traceback.print_exc())
445 except Exception as e:
446 raise e
447
448 def title_is(self,title_text):
449 """
450 判断页面title内容是否和传入的title_text内容完全匹配，匹配返回True,否则返回False
451 """
452 try:
453 return self.wait.until(EC.title_is(title_text))
454 except:
455 return False
456
457
458
459
460 if __name__ == "__main__":
461 pass
