如何提取Playwright錄制文件中的元素定位信息

引言

最近組內推行Playwright進行自動化測試,由我封裝瞭一個簡單的框架供大傢使用。本意是想讓大傢使用Page Object模式來編寫自動化代碼的,可是不知道是因為大傢嫌麻煩,還是Playwright自帶的錄制工具太好用,基本上都是先錄制,再把錄制好的腳本放到框架中執行,結果就是Page Object的優勢完全沒有享受到,另外我在頁面操作中加入瞭自動等待、重試等優化機制也是,還得在腳本裡面一個個處理。由此,我分析大傢不願意用Page Object的一個原因是自己編寫元素定位比較麻煩,而錄制下來的腳本裡面已經包含瞭定位信息,那麼我是不是可以將錄制腳本中的定位信息解析出來,在轉換成我框架裡的頁面基類,就可以直接使用瞭。

方法研究

首先我想到使用讀取python文件配合正則表達式的方式進行解析,但是仔細一想,如果一個表達式有多行的話就很難處理瞭,還需要分析語法進行拼接。後來查詢到可以使用python自帶的ast庫將python文件轉化為抽象語法樹進行解析,一下子就豁然開朗瞭。

元素定位是元素交互的前提條件,也是自動化測試最重要的步驟之一。元素” 存在 “頁面”中 ,要定位元素,就是調用“頁面”的方法:

  • querySelector(engine=body) # 選擇單個元素
  • querySelectorAll(engine=body) # 選擇多個元素
  • waitForSelector(engine=body) # 選擇單個元素,並且自動等待到元素可見、可操作

提取代碼

我這裡主要用到瞭ast的 parse walk dump 三個方法, parse 方法將腳本轉化為抽象語法樹, walk 用來遍歷整個抽象語法樹(類似xml遍歷), dump 用於調試和將無法解析的內容直接輸出,其實這裡可以使用 unparse 或通過 lineno col_offset end_lineno end_col_offset 這四個屬性定位到源碼,但是這些都需要較高的python版本才有,我現在項目中使用的python 3.7版本中還沒有這些,隻能先用 dump 代替(當然,也是我比較偷懶)。代碼如下:

import ast
from inspect import signature

from playwright.sync_api import Page

import_num = 0
dup_num = 0
result = []
with open('<playwright錄制腳本>', 'rb') as f:
    root = ast.parse(f.read())
parse_methods = set()
for i in dir(Page):
    if not i.startswith('_'):
        attr = getattr(Page, i)
        if callable(attr):
            sign = signature(attr).parameters
            if 'selector' in sign:
                parse_methods.add(i)
for node in ast.walk(root):
    if isinstance(node, ast.Call):
        if isinstance(node.func, ast.Attribute) and \
                node.func.attr in parse_methods:
            if len(node.args) == 0:
                continue
            selector = node.args[0]
            if isinstance(selector, ast.Str):
                selector = selector.s
            elif isinstance(selector, ast.Name):
                selector = f'變量: {selector.id}'
            else:
                selector = f'無法解析: {ast.dump(selector, False)}'
            frame_locator = None
            if isinstance(node.func.value, ast.Call) and \
                    isinstance(node.func.value.func, ast.Attribute) and \
                    node.func.value.func.attr == 'frame':
                if len(node.func.value.args) > 0:
                    frame_locator = str(node.func.value.args[0])
                else:
                    frame_node = node.func.value.keywords[0].value
                    if isinstance(frame_node, ast.Str):
                        frame_locator = frame_node.s
                    elif isinstance(frame_node, ast.Name):
                        frame_locator = f'變量: {frame_node.id}'
                    else:
                        frame_locator = f'無法解析: {ast.dump(frame_node, False)}'
            for i in result:
                if selector == i[0] and frame_locator == i[1]:
                    dup_num += 1
                    break
            else:
        result.append((selector, frame_locator))
                print(f"{selector}, {frame_locator} {node.func.attr == 'query_selector_all'}")
                import_num += 1
print(f'成功解析{import_num}條定位信息, 跳過{dup_num}條重復信息')

代碼解析

代碼中很多地方用到瞭 isinstance 方法來判斷節點是什麼類型,因為需要解析的內容都是 page.click(‘…’)或 page.frame(‘…’).click(‘…’)形式,所以大多數時候需要提取 Call對象的 func的 attr屬性即為所需要的定位信息。另外使用瞭 inspect 庫來提取可能需要解析的方法,如 click fill等,直接從Playwright中提取,就不需要一個一個去找瞭。再加上一些重復判斷,即為上述代碼。

希望有瞭將錄制腳本轉換Page Object的方法之後,可以將Page Object的模式在組內推廣起來吧。

到此這篇關於如何提取Playwright錄制文件中的元素定位信息的文章就介紹到這瞭,更多相關提取Playwright元素定位內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: