如何使另一个函数中的函数同时退出两个函数？

Question

我正在制作一个程序，它会询问用户一些问题，并使用不同的功能来对用户输入的内容做出反应。我需要这样做，如果用户键入“取消”，它不仅会结束他们当前所在的函数，甚至会结束调用先前函数的函数。

def func1(x):
    func2(x+5)
    print 5
def func2(x):
    func3(x+5)
    print 5
def func3(x):
    print x
    return

func1(20)
print 10

# 30
# 5
# 5
# 10

所以目前这将打印 30，退出 func3，打印 5，退出 func2，打印 5，退出 func1。我需要一种方法让它在 func3 打印 30 后停止。有什么想法吗？

所以想要的输出应该是。

#30
#10

编辑：这是一个极其简化的版本，他们不会在此特定代码中键入取消。应该是raw_input。

Answer 1

引发exception 会像您描述的那样退出函数堆栈。所以你可以拥有最外层的函数try-expect所有的内部函数。如果任何内部函数引发exception，它将冒泡（即它不会执行堆栈中的任何后续代码），直到它到达最外部函数的except。

def func1(x):
    try:
      func2(x+5)
      print 5
    except:
      pass  

def func2(x):
    func3(x+5)
    print 5

def func3(x):
    print x
    raise("Foo")  # user's input should drive this.
    return

func1(20)
print 10

它以正常方式到达func3。但是func3 表示它不希望正常执行流到func2，后者将其级联到func1。

如果您不想更改函数的内容，可以使用以下装饰器。它还具有 cmets 中提到的更正。

class UserAbort(Exception):
    pass

def swallow(f):
    def wrapper(*args):
        try:
            f(*args)
        except UserAbort:
            pass
    return wrapper

@swallow
def func1(x):
    func2(x+5)
    print 5

def func2(x):
    func3(x+5)
    print 5

def func3(x):
    print x
    raise UserAbort  # user's input should drive this.
    return

func1(20)
print 10