Thhis was exactly what I was trying to write, but wasn't able to switch to python mode on my phone :)