如何阻止os.walk走下所有子目录?

时间:2014-06-04 04:19:23

标签: python

使用:

def simpleFunc(dirName):
    import os
    fileList=[]
    for dir, dirs, filenames in os.walk(dirName):
        for filename in filenames:
            filepath=dir+'/'+filename
            fileList.append(filepath)
    print fileList


simpleFunc(os.path.dirname('/entire/path/to/file.ext'))

问题是os.walk不会停留在/entire/path/to/目录级别,而是一直向下到它可以找到的最低子目录。因此,如果有/entire/path/to/subdir1/subdir2/subdir3,那么除/entire/path/to/之外,将搜索所有三个子目录:./subdir1/././subdir2/./././subdir3/。 问题:如何确保函数在指定的目录级别停止:/entire/path/to/并且不会完全停止?

4 个答案:

答案 0 :(得分:2)

from os import listdir
from os.path import isfile, join
path='/entire/path/to/'
files = [ join(path,f) for f in listdir(path) if isfile(join(path,f)) ]

文件返回您的文件。无需声明新文件列表

答案 1 :(得分:2)

根据您编写的内容,如果您只想搜索指定的目录。最好使用os.listdir,然后在os.path.isfile上过滤,例如,像这样:

def simple_func(dirpath):
    paths = os.listdir(dirpath)
    # create absolute paths
    abs_paths = [os.path.join(dirpath, p) for p in paths]
    # filter for paths that are files
    file_list = [p for p in paths if os.path.isfile(p)]
    return file_list

这样你就不必处理停止递归的问题,而且很清楚你想要哪些文件。您可能想要查看以查看多个isfile调用是否会损害任何内容。

答案 2 :(得分:1)

听起来你应该使用os.listdir

import os

my_path = '/entire/path/to/files/'
file_list = []
for filename in os.listdir(my_path):
  filepath = os.path.join(my_path, filename)
  if os.path.isfile(filepath):
    fileList.append(filepath)

答案 3 :(得分:0)

您需要从cPtr生成器中选择唯一的第一项:

bPtr

然后将root目录添加到每个文件名:

os.walk

或只是:

import os

root, _, files = next(os.walk('/entire/path/to'))