从文件夹中复制部分文件
代码参数为原文件目录JSON_from_PATH
,复制到的文件夹目录JSON_to_PATH
和要复制的文件名列表json_name_list
import os
import random
import shutil
IMAGE_PATH = "./Images"
JSON_from_PATH = "./jsons_tmp"
JSON_to_PATH = "./jsons"
def copy_file(from_dir, to_dir, Name_list):
if not os.path.isdir(to_dir):
os.mkdir(to_dir)
# 1
# name_list = os.listdir(from_dir)
# # 2
# sample = random.sample(pathDir, 2)
# print(sample)
# 3
for name in Name_list:
try:
# print(name)
if not os.path.isfile(os.path.join(from_dir, name)):
print("{} is not existed".format(os.path.join(from_dir, name)))
shutil.copy(os.path.join(from_dir, name), os.path.join(to_dir, name))
# print("{} has copied to {}".format(os.path.join(from_dir, name), os.path.join(to_dir, name)))
except:
# print("failed to move {}".format(from_dir + name))
pass
# shutil.copyfile(fileDir+name, tarDir+name)
print("{} has copied to {}".format(from_dir, to_dir))
if __name__ == '__main__':
name_list = os.listdir(IMAGE_PATH)
# print(name_list)
# process json files
json_name_list = [name[:-4] + ".json" for name in name_list]
# print(json_name_list)
copy_file(JSON_from_PATH, JSON_to_PATH, json_name_list)