# 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")
python 递归复制源文件夹到目标文件夹,过滤文件大小和文件格式
最新推荐文章于 2023-06-25 11:20:23 发布