递归目录树python

Question

def mapping(list_of_files , list_of_folders, max1):
    print max1
    max1 += 1

    if len(list_of_folders) == 0:
        folder = os.walk(os.getcwd()).next()[1]
        files = os.listdir(os.getcwd())
    elif len(list_of_files) != 0:
        print list_of_folders[0]
        ## for some reason this of line of code stops working without error on 20150 iteration 
        folder = next(os.walk(os.getcwd() +'/' + list_of_folders[0]))[1]
        #folder = os.walk(os.getcwd() +'/' + list_of_folders[0] ).next()[1]
        #print os.listdir(os.getcwd() + '/' +list_of_folders[0])
        files = os.listdir(os.getcwd() + '/' +list_of_folders[0])


    length = len(folder)
    length1 = len (files)

    y = 0
    if folder == files:
        del files[:] 
    else :
        for x in range(length):
            while y != length1:
                if files[y] == folder[x]:
                    files.pop(y)
                    length1 = length1 - 1
                    y = 0
                y += 1
            y = 0

    #print folder
    #print files

    if len(list_of_folders) == 0:
        list_of_files = add_to_main_lists(list_of_files,'', files)
        #print list_of_files
        list_of_folders = add_to_main_lists(list_of_folders, '', folder)
        #print list_of_folders
    else:
        list_of_files = add_to_main_lists(list_of_files,list_of_folders[0], files)
        #print list_of_files
        list_of_folders = add_to_main_lists(list_of_folders, list_of_folders[0], folder)
        #print list_of_folders
        list_of_folders.pop(0)


    if len(list_of_folders) == 0:
        print "got to here"
        return  list_of_files 
    else:
        print "file length: " + str(len(list_of_files))
        print "folder length: " + str(len(list_of_folders))
        mapping(list_of_files , list_of_folders, max1)

    return list_of_files

list_of_files = []
list_of_folders = []
list_of_files = mapping(list_of_files, list_of_folders , max1)
print (list_of_files)

文件应该映射出文件所在的文件夹及其所有子目录。出于某种原因，当代码运行其20150迭代时 folder=下一个(os.步行(操作系统getcwd（）+'/'+文件夹列表[0]））[1] 终止代码并且不抛出任何错误。我不知所措。我用sudo运行代码。你知道吗

Answer 1

我可以在代码中看到两件事：

• 在同一次迭代中，您在生成器上调用了.next()两次。我不认为这是您想要做的，您的迭代将解析两个不同的os.walk输出。将os.walk(os.getcwd()).next()的输出保存在var中并使用它。

• 仅当当前级别中没有文件夹时才处理文件列表，因为您使用elif来执行list_of_files。您应该使用if，因为您无论如何都要处理它。

另外，请记住迭代器的结尾会抛出一个StopIteration异常，所以很可能就是这样。你需要处理这个异常。你知道吗