基于 .txt 文件的 os.walk 文件夹排除

Question

我想要一个 Folders_To_Skip.txt 文件，其中包含由新行分隔的目录列表

例如：

A:\\stuff\a\b\
A:\\junk\a\b\

我有一些文件正在破坏我的 .csv 记录编译，我想排除那些我无论如何都不需要阅读的目录。

在locate 函数中，我有我试图从Excluding directories in os.walk 实现的功能，但我似乎无法让它与list 中的目录一起使用，更不用说从文本文件列表中读取的时候了print 访问的文件仍然包含我试图排除的目录中的文件。

您能否解释一下解决方案是特定的排除目录（不是世界末日），还是可以操作排除子目录（会更方便）。

现在locate 之前的代码允许轻松查找控制文本文件，然后将这些项目作为列表加载到脚本的其余部分以运行，假设所有控制文件都在同一位置，但位置可以根据运行脚本的人员和位置而改变。

也出于测试目的，Drive_Locations.txt 设置为：

A
B

这是当前脚本：

import os
from tkinter import filedialog
import fnmatch

input('Press Enter to select any file in writing directory or associated control files...')
fname = filedialog.askopenfilename()
fpath = os.path.split(fname)

# Set location for Drive Locations to scan
Disk_Locations = os.path.join(fpath[0], r'Drive_Locations.txt')
# Set location for Folders to ignore such as program files
Ignore = os.path.join(fpath[0], r'Folders_To_Skip.txt')

# Opens list of Drive Locations to be sampled
with open(Disk_Locations, 'r') as Drives:
    Drive = Drives.readlines()
    Drive = [x.replace('\n', '') for x in Drive]
# Iterable list for directories to be excluded
with open(Ignore, 'r') as SkipF1:
    Skip_Fld = SkipF1.readlines()
    Skip_Fld = [x.replace('\n', '') for x in Skip_Fld]

# Locates file in entire file tree from previously established parent directory.
def locate(pattern, root=os.curdir):
    for path, dirs, files in os.walk(os.path.abspath(root), topdown=True):
        dirs[:] = [d for d in dirs if d not in Skip_Fld]
        for filename in fnmatch.filter(files, pattern):
            yield os.path.join(path, filename)

for disk in Drive:
    # Formats Drive Location for acceptance
    disk = str.upper(disk)
    if str.find(disk, ':') < 0:
        disk = disk + ':'
    # Changes the current disk drive
    if os.path.exists(disk):
        os.chdir(disk)
    # If disk incorrect skip to next disk
    else:
        continue
    for exist_csv in locate('*.csv'):
        # Skip compiled record output files in search
            print(exist_csv)

Answer 1

这里的主要错误是os.walk() 返回相对目录名称的列表。因此，例如当您在目录A:\stuff\a 中时，您要跳过的目录只是列为b，而不是A:\stuff\a\b；因此，您的跳过逻辑当然不会从当前目录的子目录列表中删除任何内容。

这是一个直接检查当前目录的重构。

for path, dirs, files in os.walk(os.path.abspath(root), topdown=True):
    if path not in Skip_Fld:
        for filename in fnmatch.filter(files, pattern):
            yield os.path.join(path, filename)

abspath 电话很重要；很高兴您将其包含在您的尝试中。

您要跳过的目录列表应该有单个反斜杠，或者可能是正斜杠，并且可能没有最终目录分隔符（幸运的是，我无法检查 Windows 上 os.walk() 是如何报告这些的）。