UI 自动化高亮元素与截图小工具

巴黎的灯光下

做UI自动化时，写了2个方法,方便清晰辨识定位的元素and 保留定位证据，与大家分享下。

高亮元素

1. def highlight(func):
   
2. 
   
3.     def apply_style(element):
   
4.                 # 实现的方式很简单，就是定位到元素后，执行js样式
   
5.         driver.execute_script("arguments[0].style.border='6px solid red'", element)
   
6. 
   
7.     @wraps(func)
   
8.     def wrapper(self, *args, **kwargs):
   
9.         element = func(self, *args, **kwargs)
   
10.         apply_style(element)
    
11.         return element
    
12. 
    
13.     return wrapper
    

复制代码

截图定位的元素

1. def screenshot(func):
   
2. 
   
3.     def screen_shot(screen_name):
   
4.         driver.save_screenshot(screen_name)
   
5. 
   
6.     @wraps(func)
   
7.     def wrapper(self, *args, **kwargs):
   
8.         element = func(self, *args, **kwargs)
   
9. 
   
10.         screen_shot(str(args[-1]) + '.jpg') # 默认是以定位元素属性为文件名
    
11.         return element
    
12. 
    
13. 
    
14.     return wrapper
    

复制代码

使用

1. class Action(object):
   
2.     def __init__(self, driver):
   
3.         self.driver = driver
   
4. 
   
5.     @screenshot
   
6.     @highlight
   
7.     def find_element(self, *loc):
   
8.         try:
   
9.             WebDriverWait(self.driver, 15).until(lambda driver: driver.find_element(*loc).is_displayed())
   
10.             return self.driver.find_element(*loc)
    
11.         except Exception as e:
    
12.             return e
    

复制代码

测试(体现百度价值的时候到了.)

1. driver = webdriver.Chrome()
   
2. 
   
3. driver.get('http://www.baidu.com')
   
4. 
   
5. action = Action(driver)
   
6. 
   
7. action.find_element(By.ID, 'su')
   
8. time.sleep(3)
   
9. driver.quit()
   

复制代码

效果图(su.jpg)

如果没封装一层定位方法，可以使用如下方式，稍微和上面有些不同，该方法会在定位前后截图，可以用来对比。

1. def highlight(element, element_name=None, debug=True):
   
2.         '''
   
3.         debug 参数用来开关截图功能
   
4.         '''
   
5.     # 高亮元素
   
6.     def apply_style():
   
7.         driver.execute_script("arguments[0].style.border='6px solid red'", element)
   
8. 
   
9.     # 截图
   
10.     def screen_shot(screen_name):
    
11.         driver.save_screenshot(screen_name)
    
12. 
    
13.     if debug:
    
14.         try:
    
15.             screen_shot(str(element_name) + '_before.jpg')
    
16.             apply_style()
    
17.             screen_shot(str(element_name) + '_after.jpg')
    
18.         except Exception as e:
    
19.             return e
    
20. 
    
21.     apply_style()
    
22. 
    
23. #  使用也很简单
    
24. element = driver.find_element(By.ID, 'kw')
    
25. highlight(element, 'kw')

复制代码

还有可以改进的地方，就这些，希望对你有帮助。O(∩_∩)O

草帽路飞UU

谢谢分享，特别是高亮元素那里

乐哈哈yoyo

优秀就来自于每一天的改进和创新，值得鼓励和肯定

八戒你干嘛

实际测试中应该是要能高亮定位失败的元素才真的有用吧，既然元素已经定位到高亮有何意义呢