This functionality is available in C++, make it available in Python directly to operate on transform modules.
walk
Context
-print-ir-after-all