Pythonista is great but it hasn't been actively developed in a couple years, and I think pieces of it might be broken. But as long as you stay away from the UI-building aspect of it, it should be enough for what you need.