How To Write A Switch Case In Python