python 递归复制源文件夹到目标文件夹，过滤文件大小和文件格式

# coding=utf-8
import os
import sys
import shutil
import tqdm

img_formats = ['.png', '.jpg', '.jpeg', '.webp', '.gif', '.bmp', '.mpg']


def copy_dir(src_root, target_root, max_file_size=1024, exclude_file_format=img_formats):
    """
    递归复制源文件夹到目标文件夹，过滤文件大小和文件格式
    """
    src_root = os.path.abspath(src_root)
    target_root = os.path.abspath(target_root)
    if os.path.exists(target_root):
        raise ValueError("{} already exists.".format(target_root))
    for root, dirs, files in os.walk(src_root):
        print(root)
        _target_root = root.replace(src_root, target_root, 1)
        os.mkdir(_target_root)
        for f in tqdm.tqdm(files):
            p = os.path.join(root, f)
            p_size = os.path.getsize(p)/float(1024)
            p_format = os.path.splitext(p)[-1]
            if p_size < max_file_size and p_format not in exclude_file_format:
                target_p = os.path.join(_target_root, f)
                shutil.copy(p, target_p)


if __name__ == '__main__':
    # if len(sys.argv) < 3:
    #     print('Usage: ./copy_dir src_root target_root')
    #     exit(1)
    # copy_dir(sys.argv[1], sys.argv[2])
    copy_dir("./", "./1")