Python编程新境界，代码逻辑分离指南！

本文链接：https://blog.csdn.net/Saki_Python/article/details/135952621

在 Python 编程中，适当的代码逻辑分离可以帮助降低复杂度、提高可读性，减少大量的 if-else 结构。本文将深入探讨如何使用不同方法来改进代码结构，降低对 if-else 结构的依赖。

1. 使用字典替代if-else

通过字典映射，将不同的操作与对应的函数关联起来，减少大量的if-else结构。

def action1():
    return "Action 1"

def action2():
    return "Action 2"

def action3():
    return "Action 3"

options = {
    '1': action1,
    '2': action2,
    '3': action3
}

choice = input("Enter choice (1, 2, 3): ")

if choice in options:
    result = options[choice]()
    print(result)
else:
    print("Invalid choice")

2. 使用策略模式

通过创建不同的策略类，将不同的行为封装在类内部，提高可维护性和灵活性。

class Action1:
    def execute(self):
        return "Action 1"

class Action2:
    def execute(self):
        return "Action 2"

class Action3:
    def execute(self):
        return "Action 3"

class Context:
    def __init__(self, strategy):
        self.strategy = strategy

    def execute_action(self):
        return self.strategy.execute()

# 在需要执行的地方选择特定的策略
choice = input("Enter choice (1, 2, 3): ")

if choice == '1':
    context = Context(Action1())
elif choice == '2':
    context = Context(Action2())
elif choice == '3':
    context = Context(Action3())
else:
    print("Invalid choice")

if choice in ('1', '2', '3'):
    result = context.execute_action()
    print(result)

3. 使用多态

利用 Python 的多态特性，将不同类对象统一调用相同的方法，从而消除冗长的 if-else 结构。

class BaseAction:
    def execute(self):
        pass

class Action1(BaseAction):
    def execute(self):
        return "Action 1"

class Action2(BaseAction):
    def execute(self):
        return "Action 2"

class Action3(BaseAction):
    def execute(self):
        return "Action 3"

# 统一调用执行方法
def perform_action(action):
    return action.execute()

choice = input("Enter choice (1, 2, 3): ")

if choice == '1':
    result = perform_action(Action1())
elif choice == '2':
    result = perform_action(Action2())
elif choice == '3':
    result = perform_action(Action3())
else:
    result = "Invalid choice"

print(result)

4. 使用装饰器

装饰器能够为函数添加额外的功能，使代码结构更为清晰，避免深层嵌套的 if-else 结构。

def choice_validator(func):
    def inner(*args, **kwargs):
        choice = args[0]
        if choice in ('1', '2', '3'):
            return func(*args, **kwargs)
        else:
            return "Invalid choice"
    return inner

@choice_validator
def perform_action(choice):
    actions = {
        '1': "Action 1",
        '2': "Action 2",
        '3': "Action 3"
    }
    return actions[choice]

choice = input("Enter choice (1, 2, 3): ")
result = perform_action(choice)
print(result)