在递归中使用迭代器

今天下午想要复现一下学长的recursion file，想模仿源码里的精髓：迭代器

遇到了bug，花了一两个小时才解决。现总结如下：
首先，想要实现的功能是递归遍历文件夹，遇到满足条件的文件时，用yield返回该文件的位置。

如果不用递归器，可以这样实现：

path_list = []
def get_one_cage(root: str, cook_folder_name: str):
    for item in os.listdir(root).copy():
        item_path = os.path.join(root, item)
        if item == cook_folder_name:
            path_list.append(item_path)
            return
        elif os.path.isdir(item_path):
            get_one_cage(item_path, cook_folder_name)

即，深度优先遍历，满足要求时，将item_path补充到列表里，之后返回上一层。
这里有一个问题，需要有一个列表，把所有满足条件的地址全存起来，占内存。

使用迭代器可以用一个，遍历出来一个，省内存

替换为迭代器，最先想到的是，把 return 换成 yield，使用for循环调用迭代器函数

def get_one_cage(root: str, cook_folder_name: str):
    for item in os.listdir(root).copy():
        item_path = os.path.join(root, item)
        if item == cook_folder_name:
            yield item_path
        elif os.path.isdir(item_path):
            get_one_cage(item_path, cook_folder_name)

但是这样的程序跑到内嵌函数时，进不去，我百思不得其解

现在看，应该是因为迭代器函数不是一个函数，不是一个命令语句，它只是一个对象。

简单说就是，python程序一般遵循：动词+名词的结构，或者动词，比如：

a = 1

这句话实际上是把1赋值给了a，是有动词的。
迭代器只是一个名词，必须用for语句调用或者next()方法调用才会执行，或者是print，yield，return等等，反正得加个动词，不能孤零零一个名词。
而且上述代码还有一个漏洞。在第一段代码中，我们用一个全局变量存放遍历结果。在第二段代码里，我们本意是把结果yield到for循环调用的地方，但事实是，程序已经套了好几层了，每次yiled只能返回一层。如下图所示：

综上两点作出如下修改：

def get_one_cage(root: str, cook_folder_name: str):
    for item in os.listdir(root).copy():
        item_path = os.path.join(root, item)
        if item == cook_folder_name:
            yield item_path
        elif os.path.isdir(item_path):
            yield get_one_cage(item_path, cook_folder_name)

程序执行结果如下：

显然是返回了一个迭代器，不是一个str，其逻辑如下图所示：

就好比，本意是：
小明把沙袋传给小红，小红传给小兰
但现在是：
小明把沙袋传给了小红，小红被传了出去
修改如下：

def get_one_cage(root: str, cook_folder_name: str):
    for item in os.listdir(root).copy():
        item_path = os.path.join(root, item)
        if item == cook_folder_name:
            yield item_path
        elif os.path.isdir(item_path):
            yield next(get_one_cage(item_path, cook_folder_name))

逻辑如下：

标签：迭代,递归,item,使用,cook,path,folder,os,name
来源： https://blog.csdn.net/frank_haha/article/details/120383940