Question: How to walk/visit the AST?

I'm trying to extract all the imports from a python file. To do this I'm using the rustpython_parser::parse function to get an AST and then iterating over all the statements in the body to find Stmt::Import and Stmt::ImportFrom. This works for imports defined at the root of the file, however misses any imports defined inside e.g. function/class definitions etc. So I think what I really want to do is walk the AST and visit all the nodes. Is there some way to achieve this with one of the rustpython packages?

从 rustpython_parser::parse 入口点和 body 中的 AST 语句开始。检查 RustPython 软件包中是否已有 AST visitor 或 traversal API，并确认它是否能够访问嵌套在函数和类定义中的 import。