有没有办法判断我是否在 Python 中使用递归？

Question

我正在编写一个函数来遍历用户的文件系统并创建一棵代表该目录的树（该树实际上是 Tkinter 中的 TreeView 小部件，但在功能上是一棵树）。

我能想到的最好的方法是递归。然而，我在函数中的一种情况要求我知道它是否是“原始”函数调用，在这种情况下文件没有父节点，或者它是否是“递归”函数调用，即已被调用的函数。由函数本身创建，以便我可以为这些文件提供适当的父节点。

Python 有没有办法询问一个函数，“嘿，你是递归的吗？”或者“嘿，你是从哪里打来的？”

Answer 1

与其他所有语言几乎相同 - 在您的情况下，您传递对父级的引用并检查它是否为 None。如果是这样，您将创建一个适当的父节点。

Answer 2

我在函数中的一个案例要求我知道它是否是“原始”函数调用，在这种情况下文件没有父节点

这似乎是特定函数需要太多工作的奇怪情况。你必须建造这棵树——为什么你需要知道它附着在哪里？为什么不直接构建您负责的节点并返回它们呢？

def make_tree(path):
    return [
        make_tree(os.path.join(path, element))
        for element in get_elements(path)]

当你收到它时，再次走动树吗？

如果你真的想集成它，只需传递父级：

def make_tree(path, parent_node = None):
    new_node = Node(...)
    for ....:
        make_tree(path+..., new_node)

    if parent_node is not None:
        parent_node.add(new_node)
    else:
        .....

Answer 3

在调用递归的同时包含对父级或某些级别信息的引用应该是简单且常见的。

另一种方法（不过我不喜欢）是使用 pythons inspect 模块，它可以让您检查例如调用堆栈。一个例子：

#!/usr/bin/env python

import inspect

def whocalled():
    return inspect.stack()[2][3]

def fib(n):
    print n, whocalled()
    if n < 2:
        return n
    return fib(n - 1) + fib(n - 2)

if __name__ == '__main__':
    fib(4)

会打印：

4 <module>
3 fib
2 fib
1 fib
0 fib
1 fib
2 fib
1 fib
0 fib

Answer 4

您可以通过inspect模块访问内存堆栈。

import inspect

def whoami():
    '''Returns the function name of the caller.'''

    return inspect.stack()[1][3] #Stack data for the name of the function

def caller():
    '''Returns the caller of a function.'''

    return inspect.stack()[2][3] #Stack data for the name of whatever calls this

caller = caller()

如果您尝试从 __main__ 调用它，您将收到索引超出范围错误。

when the caller of the function != whoami() -> no longer recursing.

Answer 5

我真的很奇怪你为什么把事情搞得这么复杂。您可以简单地将函数分为递归部分和非递归部分！这是一个简单的示例：

def countdown(n):
    " original function "
    print "Ok, start the countdown (not recursive)"

    msg = "We are at"

    def recursive( x ):
        " recursive part "
        if x >= 0:
            print msg, x, "(recursive)"
            return recursive(x-1)

    return recursive(n)

countdown(10)

实际上，您甚至不需要递归函数的许多参数，因为它是一个闭包，可以使用您在其名称空间中定义的任何内容。