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

第如何提取Playwright录制文件中的元素定位信息我这里主要用到了ast的parsewalkdump三个方法，parse方法将脚本转化为抽象语法树，walk用来遍历整个抽象语法树（类似xml遍历），dump用于调试和将无法解析的内容直接输出，其实这里可以使用unparse或通过linenocol_offsetend_linenoend_col_offset这四个属性定位到源码，但是这些都需要较高的python版本才有，我现在项目中使用的python3.7版本中还没有这些，只能先用dump代替（当然，也是我比较偷懒）。代码如下：

importast

frominspectimportsignature

fromplaywright.sync_apiimportPage

import_num=0

dup_num=0

result=[]

withopen('playwright录制脚本','rb')asf:

root=ast.parse(f.read())

parse_methods=set()

foriindir(Page):

ifnoti.startswith('_'):

attr=getattr(Page,i)

ifcallable(attr):

sign=signature(attr).parameters

if'selector'insign:

parse_methods.add(i)

fornodeinast.walk(root):

ifisinstance(node,ast.Call):

ifisinstance(node.func,ast.Attribute)and\

node.func.attrinparse_methods:

iflen(node.args)==0:

continue

selector=node.args[0]

ifisinstance(selector,ast.Str):

selector=selector.s

elifisinstance(selector,ast.Name):

selector=f'变量:{selector.id}'

else:

selector=f'无法解析:{ast.dump(selector,False)}'

frame_locator=None

ifisinstance(node.func.value,ast.Call)and\

isinstance(node.func.value.func,ast.Attribute)and\

node.func.value.func.attr=='frame':

iflen(node.func.value.args)0:

frame_locator=str(node.func.value.args[0])

else:

frame_node=node.func.value.keywords[0].value

ifisinstance(frame_node,ast.Str):

frame_locator=frame_node.s

elifisinstance(frame_node,ast.Name):

frame_locator=f'变量:{frame_node.id}'

else:

frame_locator=f'无法解析:{ast.dump(frame_node,False)}'

foriinresult:

ifselector==i[0]andframe_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