Python編程進階代碼邏輯分離指南

更新時間：2023年12月14日 11:07:49 作者：濤哥聊Python

在?Python?編程中,適當?shù)拇a邏輯分離可以幫助降低復雜度、提高可讀性,減少大量的?if-else?結構,本文將深入探討如何使用不同方法來改進代碼結構,降低對?if-else?結構的依賴

1. 使用字典替代if-else

通過字典映射，將不同的操作與對應的函數(shù)關聯(lián)起來，減少大量的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. 使用策略模式

通過創(chuàng)建不同的策略類，將不同的行為封裝在類內(nèi)部，提高可維護性和靈活性。

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()
# 在需要執(zhí)行的地方選擇特定的策略
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. 使用多態(tài)

利用 Python 的多態(tài)特性，將不同類對象統(tǒng)一調(diào)用相同的方法，從而消除冗長的 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"
# 統(tǒng)一調(diào)用執(zhí)行方法
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. 使用裝飾器

裝飾器能夠為函數(shù)添加額外的功能，使代碼結構更為清晰，避免深層嵌套的 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)