That's a clever way of leveraging existing tooling! I hadn't heard of astroid before, I've always used the built in ast module