如何在Python中找到递归的空目录?

时间:2014-11-06 08:32:01

标签: python directory-structure subdirectory

与GNU find' s find . -type d -empty -delete类似,我想找到空目录,包括那些带有空子目录的目录(以及包含emtpy子目录的子目录等),但不删除它们。是否有任何现有的解决方案,或者我是否必须手动使用os.walk(可能使用topdown=False并跟踪到目前为止找到的空子目录)?

2 个答案:

答案 0 :(得分:7)

以下是使用生成器os.walk

的简单解决方案
import os

def find_empty_dirs(root_dir='.'):
    for dirpath, dirs, files in os.walk(root_dir):
        if not dirs and not files:
            yield dirpath

print list(find_empty_dirs())

我不明白为什么topdown=False是必要的,我不认为它会改变任何事情。

这确实认为只包含空目录的目录本身不是空的,但find . -type d -empty也是如此。

虽然进行了一些测试,但我看到find . -type d -empty -delete 确实首先删除空子目录,然后删除较高目录,如果它们为空。但是使用os.walk不会起作用,因为它在下降之前读取子目​​录列表,即使使用topdown=False也是如此。

删除空子目录树的递归解决方案可以是:

import os

def recursive_delete_if_empty(path):
    """Recursively delete empty directories; return True
    if everything was deleted."""

    if not os.path.isdir(path):
        # If you also want to delete some files like desktop.ini, check
        # for that here, and return True if you delete them.
        return False

    # Note that the list comprehension here is necessary, a
    # generator expression would shortcut and we don't want that!
    if all([recursive_delete_if_empty(os.path.join(path, filename))
            for filename in os.listdir(path)]):
        # Either there was nothing here or it was all deleted
        os.rmdir(path)
        return True
    else:
        return False

答案 1 :(得分:2)

好的,这是我使用os.walk的手动解决方案。当然可以修改函数is_empty,例如排除隐藏文件,或在我的示例中desktop.ini

import os


def empty_dirs(root_dir='.', recursive=True):
    empty_dirs = []
    for root, dirs, files in os.walk(root_dir, topdown=False):
        #print root, dirs, files
        if recursive:
            all_subs_empty = True  # until proven otherwise
            for sub in dirs:
                full_sub = os.path.join(root, sub)
                if full_sub not in empty_dirs:
                    #print full_sub, "not empty"
                    all_subs_empty = False
                    break
        else:
            all_subs_empty = (len(dirs) == 0)
        if all_subs_empty and is_empty(files):
            empty_dirs.append(root)
            yield root


def is_empty(files):
    return (len(files) == 0 or files == ['desktop.ini'])


def find_empty_dirs(root_dir='.', recursive=True):
    return list(empty_dirs(root_dir, recursive))


print find_empty_dirs(recursive=False)