利用Python装饰器构建灵活、可维护的工作流程
本文介绍如何利用Python装饰器构建类似有限状态机的灵活工作流程，从而避免传统if-else语句带来的代码冗余和难以维护的问题。 尤其在处理多步骤流程、异常情况和回滚重试时，装饰器能提供更优雅的解决方案。
考虑一个四步工作流程，每步都可能抛出异常或返回不同值，影响后续步骤执行。我们希望通过装饰器定义每个步骤的条件跳转和重试机制，实现类似有限状态机的效果：
@fsm(condition=lambda x: isinstance(x, Exception), method='step2')
@fsm(condition=lambda x: x is True, method='step3')
@fsm(condition=lambda x: x is False, method='step4')
def step1(arg):
return arg
@fsm(condition=lambda x: isinstance(x, Exception), method='step2', retry=3)
@fsm(condition=lambda x: x is True, method='step3')
@fsm(condition=lambda x: x is False, method='step4')
def step2(arg):
return arg
@fsm(condition=lambda x: isinstance(x, Exception), method='step1', retry=3)
@fsm(condition=lambda x: x is True, method='step3')
@fsm(condition=lambda x: x is False, method='step4')
def step3(arg):
return arg
@fsm(condition=lambda x: isinstance(x, Exception), retry=3)
def step4(arg):
return arg
# 启动流程
step1('arg')
为此，我们需要两个装饰器：fsm 用于状态跳转，fsmerror 用于异常重试。
立即学习“Python免费学习笔记（深入）”；
fsm 装饰器接收 condition (判断条件，这里使用lambda函数更灵活) 和 method (跳转目标函数名) 作为参数。
fsmerror 装饰器处理异常，接收 retry (重试次数) 和 method (失败跳转目标函数名)。
下面是一个可能的实现：
def fsmerror(retry, method):
def decorator(func):
def wrapper(*args, **kwargs):
retries_left = retry
while retries_left > 0:
try:
return func(*args, **kwargs)
except Exception as e:
retries_left -= 1
print(f"Error: {e}, {retries_left} retries left.")
# 重试次数用尽，跳转到指定函数
next_step = globals()[method]
return next_step(*args, **kwargs)
return wrapper
return decorator
def fsm(condition, method, retry=0): # 添加retry参数到fsm
def decorator(func):
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
if condition(result):
next_step = globals()[method]
return next_step(*args, **kwargs)
return result
return wrapper
return decorator
# 模拟异常和返回值
def error_func(arg):
raise Exception("模拟异常")
# ... (step1, step2, step3, step4 函数定义同上) ...
# 启动流程
step1('arg')
此代码改进了异常处理，使用 try-except 块更清晰地处理异常，并使用 retries_left 变量跟踪剩余重试次数。 fsm 装饰器也添加了 retry 参数，允许在状态跳转中也设置重试机制。 lambda函数的使用使得条件判断更加灵活。 后续步骤的参数仍然是原参数，可根据实际情况调整。 通过这种方式，我们可以用更简洁、更易维护的代码实现复杂的工作流程，并有效处理异常和回滚。
