进程池完成文件夹拷贝器
import os
import shutil # 文件操作的高级模块
import multiprocessing
import time
# 拷贝文件的任务
def copy_file_work(src_dir, dst_dir, file_name):
# 查看当前进程号
print(multiprocessing.current_process().pid)
# 源文件的路径
src_file_path = src_dir + "/" + file_name
# 目标文件的路径
dst_file_path = dst_dir + "/" + file_name
with open(dst_file_path, "wb") as dst_file:
with open(src_file_path, "rb") as src_file:
while True:
src_file_data = src_file.read(1024)
if src_file_data:
# 读取到原文件的数据写入到目标文件里面
dst_file.write(src_file_data)
else:
break
time.sleep(0.1)
if __name__ == '__main__':
# 源目录
src_dir = "test"
# 目标目录
dst_dir = "/home/python/Desktop/test"
if os.path.exists(dst_dir):
# 删除指定目录的文件夹，把文件夹里面的所有内容全部删除
shutil.rmtree(dst_dir)
# 创建指定目录的文件夹
os.mkdir(dst_dir)
# 获取原目录的文件名列表
file_name_list = os.listdir(src_dir)
# 创建进程池
pool = multiprocessing.Pool(3)
# 遍历文件名列表获取对应的文件名
for file_name in file_name_list:
# 使用进程池完成文件的异步拷贝
pool.apply_async(copy_file_work, (src_dir, dst_dir, file_name))
# 关闭进程池
pool.close()
# 主进程等待进程池执行完成以后程序再退出
pool.join()
