Python代码逻辑分离，从减少if...else开始

大模型研究院

于 2023-08-29 14:52:03 发布

阅读量187

点赞数

文章标签： python 开发语言团队开发 github 职场和发展

本文链接：https://blog.csdn.net/l01011_/article/details/132561306

版权

今天带大家学习如何利用对象（类）的特性来解决冗长的if…else条件判断

前言

if…else…条件语句我相信学习python的童鞋们都喜欢用，例如下一面这一段

def func(param):
    if param == "cat":
        print("这是一只猫咪")
    elif param == "dag":
        print("这是一直狗狗")
    elif param == "pig":
        print("这是一只小猪")
    ...
    else:
      print("不知道是什么")

写很长的一段if语句来判断不同的情况

这段代码的使用场景是满足用户可以完成在不同场景进行对应的操作，对应后端的代码，你肯定能想到最简单的实现方式就是上面的if … else 语句。但是随着你的场景不断的增加，你的if else 代码也跟着越来越长了。这带来的一个问题就是系统的可扩展性差，后续的维护会变得举步维艰。

对象特性

对象的三大特性：封装、继承、多态

封装:根据职责将属性和⽅法封装到⼀个抽象的类中; 增强代码的安全性

继承:实现代码的重⽤，相同的代码不需要重复的编写; 增强代码的可重用性

多态:不同的对象调⽤相同的⽅法，产⽣不同的执⾏结果; 增加代码的灵活度,增强代码的可扩展性

定义基础类

首先，我们将定义一个基础类，它将作为不同条件的父类。我们将在这个类中定义一个通用的方法，用于执行操作。

class Animal():
    def execute(self):
        pass

创建子类

接下来，我们将创建不同的类来表示不同的条件，每个类都会根据条件来执行execute()方法来做某些事。

class Cat(Animal):
    def execute(self):
        print("say miao~miao~!")

class Dog(Animal):
    def execute(self):
        print("say wang~wang~!")

class Pig(Animal):
    def execute(self):
        print("say hang~hang~!")

在这里我们会用到对象的其中一个特性：继承，我们继承的基类Animal，并且对execute()方法进行了重写.

构建实例

我们可以用一个字典或者工厂方法来实现通过一个支付类型来找到对应的具体支付实例对象。

dict_name = {
     "cat": Cat(),
     "dog": Dog(),
     "pig": Pig(),
}

替换if else

创建一个函数，利用dict_name字段的关键字条件调用不的类执行execute()方法.

def func(anmial):
  dict_name.get(animal).execute()

# 执行函数
func("cat")
# 输出：say miao~miao~!

func("dog")
# 输出：say wang~wang~!

func("pig")
# 输出：say hang~hang~!