递归删除大型文件夹结构中第三级以下的文件夹

Question

我有一个只有数千个文件夹的大型结构，但我只对保持前三级的文件夹感兴趣，并删除其余的文件夹。我正在寻找一个递归python脚本来实现这一点。非常感谢您的帮助。

Answer 1

未测试，但使用os.walk():时可能会出现类似情况

import os
import shutil
BASE = '.'
for root, dirs, files in os.walk(BASE):
    n = 0
    head = root
    while head and head != BASE:
        head, _ = os.path.split(head)
        n += 1
    if n == 3:
        for dir in dirs:
            shutil.rmtree(os.path.join(root, dir))
        del dirs[:]      # clear dirs so os.walk() doesn't look for subdirectories

Answer 2

正确的方法是使用os.walk，但这里有一个便宜的答案：

>>> import os
>>> os.system('rm -rf */*/*/*/*')
>>> os.system('rmdir */*/*/*')

这将删除中至少四个级别的所有文件，然后尝试删除中至少三个级别的根目录。由于上一个命令将删除其内容，rmdir将成功（并抱怨它找到的所有非目录叶）。