Python替代goto的惯用方案：分步任务终止后执行收尾代码

问题描述 #

我有一个按步骤执行多项任务的函数，希望在每步结束后选择继续执行或跳过后续步骤，且所有任务完成后必须执行一段收尾代码。我知道可以使用try+finally实现，但这并非错误处理场景，不想抛出伪异常。目前我通过滥用无限while循环搭配break语句实现类似goto的效果，想咨询是否存在更符合Python风格的实现方式？

现有代码示例：

# ugly goto hack
while True:
  do_work()

  if condition:
    break

  do_more_work()
  
  if condition:
    break

  do_even_more_work()

  if condition:
    break

  #....

  break

do_some_final_work()

解决方案1：任务列表迭代法 #

将所有步骤任务存入列表，按顺序循环执行，每步后检查终止条件，满足则跳出循环。这种方式逻辑直观，代码简洁，完全贴合Python的迭代思维。

# 按执行顺序排列任务函数
tasks = [do_work, do_more_work, do_even_more_work]

for task in tasks:
    task()
    if condition:  # 每步执行后检查是否终止后续流程
        break

do_some_final_work()  # 确保收尾代码必执行

如果各步骤的终止条件不同，可将任务与对应条件封装成元组：

tasks = [
    (do_work, lambda: condition1),
    (do_more_work, lambda: condition2),
    (do_even_more_work, lambda: condition3),
]

for task, check in tasks:
    task()
    if check():
        break

do_some_final_work()

解决方案2：生成器控制执行流 #

用生成器逐个产出任务，通过循环调用控制执行节奏，适合步骤逻辑复杂、需要动态生成任务的场景。

def task_generator():
    yield do_work
    yield do_more_work
    yield do_even_more_work

for task in task_generator():
    task()
    if condition:
        break

do_some_final_work()

解决方案3：上下文管理器（可选） #

若希望将收尾逻辑与任务执行绑定，可自定义上下文管理器。这里的try+finally并非用于错误处理，而是利用上下文管理器的特性确保收尾代码必然执行，属于正常流程管理，不算伪异常滥用。

from contextlib import contextmanager

@contextmanager
def task_session():
    try:
        yield
    finally:
        do_some_final_work()

with task_session():
    do_work()
    if condition:
        return
    do_more_work()
    if condition:
        return
    do_even_more_work()
    # ...后续步骤

内容的提问来源于stack exchange，提问作者user23538168