python文件夹复制_python文件夹copy器（多进程版）

本节的练习的要求如下：

输入要拷贝文件夹的文件名称

读取该文件夹下的所有文件

启动5个进程来拷贝文件夹，将拷贝成功的文件名称放入队列中

主进程中显示文件拷贝的进度

代码如下：

import multiprocessing

import os

import time

import random

def copy_file(q, file_name, source_folder_name, dest_folder_name):

"""拷贝文件"""

# print("正在拷贝文件:%s，%s ==> %s" % (file_name, source_folder_name, dest_folder_name))

f_read = open(source_folder_name + "/" + file_name, "rb")

f_write = open(dest_folder_name + "/" + file_name, "wb")

while True:

content = f_read.read(1024)

if content:

f_write.write(content)

else:

break

f_write.close()

f_read.close()

time.sleep(random.random())

# 放入已经拷贝完成的文件

q.put(file_name)

def main():

# 获取要复制的文件夹名称

source_folder_name = input("请输入要复制的文件夹名称：")

# 目标文件夹

dest_folder_name = source_folder_name + "[副本]"

if not os.path.exists(dest_folder_name):

os.mkdir(dest_folder_name)

# 获取文件夹中的所有普通文件名

file_names = os.listdir(source_folder_name)

# 创建Queue

q = multiprocessing.Manager().Queue()

# 创建pool

po = multiprocessing.Pool(5)

for file_name in file_names:

po.apply_async(copy_file, args=(q, file_name, source_folder_name, dest_folder_name))

po.close()

all_file_num = len(file_names)

copy_ok_num = 0

while True:

file_name = q.get()

copy_ok_num += 1

print("\r拷贝的进度为：%.2f%%" % (copy_ok_num * 100 / all_file_num), end="")

if copy_ok_num >= all_file_num:

break

print("")

if __name__ == "__main__":

main()