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。