Python: Python: switch statement (a replacement of..)