Python是否允許遞歸的__iter__函數？

Question

我正在嘗試編寫一個__iter__函數，該函數應該遞歸遍歷一個目錄（包括子目錄），並且由於它的結構是任意的，我認為遞歸函數將是最佳選擇。 但它沒有用。

這就是我所擁有的：

class Dummy(object):

    def __init__(self, directory):
        self.directory = directory

    def _iterate_on_dir(self, path):
        '''
        Internal helper recursive function.
        '''
        for filename in os.listdir(path):
            full_path = os.path.join(path, filename)
            if os.path.isdir(full_path):
                self._iterate_on_dir(full_path)
            else:
                yield full_path

    def __iter__(self):
        '''
        Yield filenames
        '''
        return self._iterate_on_dir(self.directory)

一些print語句告訴我，遞歸調用被簡單地忽略了。

我怎么能做到這一點？

Answer 1

現在當你遞歸調用_iterate_on_dir你只是創建一個生成器對象，而不是實際迭代它。

修復： self._iterate_on_dir(full_path)應該變為：

for thing in self._iterate_on_dir(full_path):
    yield thing

如果您使用的是Python 3，則可以將其替換為：

yield from self._iterate_on_dir(full_path)