Is there a nice way to do this in python?