递归目录树python

时间:2016-05-08 22:30:30

标签: python recursion os.walk

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 = next(os.walk(os.getcwd()+'/'+ list_of_folders [0]))[1] 终止代码并且不会抛出任何错误。我不知所措。我使用sudo运行代码。

1 个答案:

答案 0 :(得分:2)

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

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

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

此外,请记住,迭代器的结尾会引发StopIteration异常,因此这很可能发生了什么。你需要处理那个例外。