Python 根据状态切换

def step1():
    print("Step 1: Initializing...")
    # 返回下一个要执行的步骤
    return "step2"

def step2():
    print("Step 2: Processing...")
    # 条件判断跳转
    if some_condition():
        return "step3"
    else:
        return "step4"

def step3():
    print("Step 3: Handling special case...")
    return "step5"

def step4():
    print("Step 4: Normal operation...")
    return "step5"

def step5():
    print("Step 5: Finalizing...")
    return None  # 无后续步骤

def some_condition():
    # 假设条件为真
    return True

# 定义步骤映射
steps = {
    "step1": step1,
    "step2": step2,
    "step3": step3,
    "step4": step4,
    "step5": step5
}

# 执行步骤链
current_step = "step1"

while current_step:
    current_step = steps[current_step]()  # 执行当前步骤并获取下一步骤