AST选择器

使用CSS选择器语法查询AST元素

"通过CSS选择器语法查询AST元素 🌲。💅 "

安装和使用

安装

pip install ast-selector

使用

from ast_selector import AstSelector

tree = load_python_code_as_ast_tree()
query = "FunctionDef Raise $FunctionDef"
# Query all functions that raises at least an exception

functions_raising_exceptions = AstSelector(query, tree).all()

使用案例

返回整数的函数

from ast_selector import AstSelector

tree = load_python_code_as_ast_tree()
query = "FunctionDef.returns[id=int] $FunctionDef"
# Query all functions that return ints e.g. def sum() -> int

function_element = AstSelector(query, tree).first()

许可

MIT

致谢

在全职工作的同时，保持开源项目的开发非常困难。我从心底感谢上帝给予的灵感和精力。