pyton 策略模式优化 if else