利用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 decoratordef 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函数的使用使得条件判断更加灵活。 后续步骤的参数仍然是原参数，可根据实际情况调整。 通过这种方式，我们可以用更简洁、更易维护的代码实现复杂的工作流程，并有效处理异常和回滚。
以上就是如何利用Python装饰器构建灵活的工作流程并实现状态跳转和异常处理？的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1358802.html
微信扫一扫 
支付宝扫一扫 
![[爬虫]Windows下如何安装python第三方库lxml](https://www.chuangxiangniao.com/wp-content/uploads/2025/11/175867272598906.jpg)
![[爬虫]Windows下如何安装python第三方库lxml](https://www.chuangxiangniao.com/wp-content/uploads/2025/11/175867272574191.jpg)
![[爬虫]Windows下如何安装python第三方库lxml](https://www.chuangxiangniao.com/wp-content/uploads/2025/11/175867272587234.jpg)
![[爬虫]Windows下如何安装python第三方库lxml](https://www.chuangxiangniao.com/wp-content/uploads/2025/11/175867272666334.jpg)
![[WPF自定义控件库]排序、筛选以及高亮](https://www.chuangxiangniao.com/wp-content/uploads/2025/11/175867275465956.jpg)
![[WPF自定义控件库]排序、筛选以及高亮](https://www.chuangxiangniao.com/wp-content/uploads/2025/11/175867275531393.jpg)
![[WPF自定义控件库]排序、筛选以及高亮](https://www.chuangxiangniao.com/wp-content/uploads/2025/11/175680265669916-44.png)
