Python获取指定目录下的文件

转载自:https://www.cnblogs.com/testlearn/p/14426580.html

import os


def find_file(search_path, include_str=None, filter_strs=None):
    """
    查找指定目录下所有的文件(不包含以__开头和结尾的文件)或指定格式的文件,若不同目录存在相同文件名,只返回第1个文件的路径
    :param search_path: 查找的目录路径
    :param include_str: 获取包含字符串的名称
    :param filter_strs: 过滤包含字符串的名称
    """
    if filter_strs is None:
        filter_strs = []

    files = []
    # 获取路径下所有文件
    names = os.listdir(search_path)
    for name in names:
        path = os.path.abspath(os.path.join(search_path, name))
        if os.path.isfile(path):
            # 如果不包含指定字符串则
            if include_str is not None and include_str not in name:
                continue

            # 如果未break,说明不包含filter_strs中的字符
            for filter_str in filter_strs:
                if filter_str in name:
                    break
            else:
                files.append(path)
        else:
            files += find_file(path, include_str=include_str, filter_strs=filter_strs)
    return files


if __name__ == '__main__':
    # 获取全部文件
    f = find_file("./test")
    print(f)

    # 获取包含指定字符的文件
    f = find_file("./test", include_str=".py")
    print(f)

    # 获取不包含指定字符的文件
    f = find_file("./test", filter_strs=[".pyc", "__init__"])
    print(f)

    # 获取包含指定字符且不包含某些指定字符的文件
    f = find_file("./test", include_str=".py", filter_strs=[".pyc", "__init__"])
    print(f)

 

posted @ 2022-03-11 14:22  Jaetyn  阅读(3797)  评论(0编辑  收藏  举报