I love this. I made one myself a few months ago at https://github.com/samzhang111/pit.

Someone else posted another version of this recently where referenced modules were imported automatically, but I can't remember the repo...

There's a handful of great ideas that I might include in pype, such as the -p argument.

I'm not aware of any way of automatically importing referenced modules, and manually parsing the user code for modules seems a bit daunting. I'd love to see a solution, if you can find the repo.