python 使用 walk 搜索文件

作为程序员，我们经常会搜索特定的文件，并做特殊的处理。
python 搜索文件特别方便，可以直接使用 walk API. 代码如下

• walk 通过在目录树中游走输出在目录中的文件名，向上或者向下

import os
import time


def find_files_with_suffix(target_dir, target_suffix="txt"):
    """ 查找以 target_suffix 为后缀的文件，并返加 """
    find_res = []
    target_suffix_dot = "." + target_suffix
    walk_generator = os.walk(target_dir)
    for root_path, dirs, files in walk_generator:
        if len(files) < 1:
            continue
        for file in files:
            file_name, suffix_name = os.path.splitext(file)
            if suffix_name == target_suffix_dot:
                file_path = os.path.join(root_path, file)
                find_res.append(file_path)
    return find_res


def get_file_size(file):
    """ 得到文件的大小，单位 M """
    assert os.path.exist(file), r'输入的文件不存在 {}'.format(file)
    return os.stat(file).st_size / (1021 * 1024)


def get_file_name_and_suffix(file_name):
    """ 输入的文件名，输出文件名以及后缀 """
    split_list = os.path.splitext(file_name)
    file_name_pre = split_list[-2]
    file_suffix = split_list[-1]
    return file_name_pre, file_suffix


if __name__ == '__main__':

    begin_time = time.time()

    dir = r'F:/'
    files = find_files_with_suffix(dir, target_suffix='mp4')
    for item in files:
        print(item)
    print('cost time: {} s'.format(time.time() - begin_time))

写完这篇Blog, 是今年的第 60 篇博客，可以收尾跨年到 2022 年了。来年再见！