Python：最大递归深度

Question

下面给出了一个用于返回其参数大小总和的函数，该参数可以是单个文件/目录或文件/目录列表。该代码给出了错误消息RuntimeError: maximum recursion depth exceeded while calling a Python object 但是我尝试对此进行测试。

如何解决这个问题？

谢谢

放心

#!/usr/bin/python3.1
import os

def fileSizes(f):
    if hasattr(f,'__iter__'):
        return sum(filter(fileSizes,f))
    if os.path.isfile(f):
        return os.path.getsize(f)
    elif os.path.isdir(f):
        total_size = os.path.getsize(f)
        for item in os.listdir(f):
            total_size += fileSizes(os.path.join(f, item))
        return total_size

Answer 1

使用内置的os.walk(Documentation) 方法，而不是编写自己的临时目录横向方法。

事实上，文档中的示例（上面的链接）计算了非目录文件的总大小。

Answer 2

上次我检查时，默认的最大递归深度是 1000。你有几个选择：

1. 使用Stackless Python，它不施加递归限制（允许无限递归）。
2. 以迭代方式而不是递归方式重写您的函数，这样一开始就不会溢出堆栈。
3. 使用sys.setrecursionlimit 设置最大递归限制。请注意，如果滥用，这可能会导致分段错误。

Answer 3

问题出在一行：

if hasattr(f,'__iter__'):
    return sum(filter(fileSizes,f))

由于f是一个路径，它是一个字符串，它有__iter__的属性，所以你无限循环到那里。

Answer 4

有很多技巧可以避免深度递归的问题，但我怀疑这不是你的核心问题。 （也许是如果你有非常深的目录结构..所以先在浅目录上测试）。

我猜您在目录树中发现了由链接引起的循环结构。 IE。指向父目录的符号链接目录。 （这种结构并不少见）。

您可以通过在关注目录之前检查目录是否为符号链接来避免无限循环。

另一种可能性是你得到“。”或“..”列表中的某处。这些也会产生无限循环。我认为这里不会发生这种情况，但值得考虑。 os.listdir 似乎没有返回它们。