通用技术 UI 自动化高亮元素与截图小工具

xie_0723 · 发布于 2017年3月20日 · 最后由 xie_0723 回复于 2017年3月22日 · 1198 次阅读
本帖已被设为精华帖!

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

  • 高亮元素

def highlight(func):

    def apply_style(element):
                # 实现的方式很简单,就是定位到元素后,执行js样式
        driver.execute_script("arguments[0].style.border='6px solid red'", element)

    @wraps(func)
    def wrapper(self, *args, **kwargs):
        element = func(self, *args, **kwargs)
        apply_style(element)
        return element

    return wrapper


  • 截图定位的元素

def screenshot(func):

    def screen_shot(screen_name):
        driver.save_screenshot(screen_name)

    @wraps(func)
    def wrapper(self, *args, **kwargs):
        element = func(self, *args, **kwargs)

        screen_shot(str(args[-1]) + '.jpg') # 默认是以定位元素属性为文件名
        return element


    return wrapper

  • 使用
class Action(object):
    def __init__(self, driver):
        self.driver = driver

    @screenshot
    @highlight
    def find_element(self, *loc):
        try:
            WebDriverWait(self.driver, 15).until(lambda driver: driver.find_element(*loc).is_displayed())
            return self.driver.find_element(*loc)
        except Exception as e:
            return e

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

driver = webdriver.Chrome()

driver.get('http://www.baidu.com')

action = Action(driver)

action.find_element(By.ID, 'su')
time.sleep(3)
driver.quit()

  • 效果图(su.jpg) 效果图(su.jpg)

  • 如果没封装一层定位方法,可以使用如下方式,稍微和上面有些不同,该方法会在定位前后截图,可以用来对比。
def highlight(element, element_name=None, debug=True):
        '''
        debug 参数用来开关截图功能
        '''
    # 高亮元素
    def apply_style():
        driver.execute_script("arguments[0].style.border='6px solid red'", element)

    # 截图
    def screen_shot(screen_name):
        driver.save_screenshot(screen_name)

    if debug:
        try:
            screen_shot(str(element_name) + '_before.jpg')
            apply_style()
            screen_shot(str(element_name) + '_after.jpg')
        except Exception as e:
            return e

    apply_style()

#  使用也很简单
element = driver.find_element(By.ID, 'kw')
highlight(element, 'kw')

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

如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
共收到 13 条回复
104 seveniruby 将本帖设为了精华贴 3月21日 00:53
6504

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

1439

@lose 😊

@seveniruby 谢谢思寒加精。✨

104
1439xie_0723 回复

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

9489

这个东西很赞,正好需要

6493

高亮很赞

827

不错啊,谢谢分享

1439

@lbxoqy 不客气.😀

96

学到了,谢谢

14443

Traceback (most recent call last):
File "E:\test\test02.py", line 55, in
action.find_element(By.ID, 'su')
NameError: name 'By' is not defined
@xie_0723 您好!按照您提供的代码出现了异常,请问如何解决?

15001
14443hshj1989 回复

from selenium.webdriver.common.by import By

14443

@bao_han 好的,谢谢。

1439

@xlth1947 不客气😀

需要 登录 后方可回复, 如果你还没有账号请点击这里 注册