如何使用python在目录中使用glob模式执行递归删除?

Question

DirA
file1.txt
file2.conf
DirB
file3.txt
file4.sh
file5.sh
DirC
file6.bat
file7.txt

在上面的示例目录中，我需要使用 glob 模式执行递归删除。

pattern = ['*,txt','*.sh']

使用上述模式，我需要删除所有目录中所有具有 *.txt 和 *.sh 格式的文件

Answer 1

import os, glob
pattern = ['*.txt', '*.sh']
for p in pattern:
[os.remove(x) for x in glob.iglob('DirA/**/' + p, recursive=True)]

如果需要，可以使用列表推导来完成此任务。嘟嘟！

Answer 2

import os
import glob
for filename in glob.iglob('DirA/**/*.txt', recursive=True):
os.remove(filename)
for filename in glob.iglob('DirA/**/*.sh', recursive=True):
os.remove(filename)

这应该可以递归删除目录中的所有 txt 和 sh 文件。

或者，如果要指定带有模式的数组：

import os
import glob

def removeAll(pathToDir, patterns):
for pattern in patterns:
for filename in glob.iglob(pathToDir + '/**/' + pattern, recursive=True):
os.remove(filename)

patterns = ['*.txt', '*.sh']
removeAll('DirA', patterns)

Answer 3

您可以改用os.walk，这样您只需要遍历一次目录。使用glob，您必须遍历两次，因为您有两种模式要寻找。

import os
for root, _, files in os.walk('DirA'):
for file in files:
if any(file.endswith(ext) for ext in pattern):
os.remove(os.path.join(root, file))