Switch Case Python Code