利用python多线程实现拷贝文件夹的文件

import multiprocessing 

import threading

import os

import time

import random

import sys

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

    #print('正在复制：%s' % file_name)

    if os.path.isdir(source_folder_name + '/' + file_name):

        queue.put(file_name)

        return

    if not os.path.exists(dest_folder_name):

        os.mkdir(dest_folder_name)

    data_file = open(source_folder_name + '/' + file_name,'rb')

    cp_file = open(dest_folder_name + '/' + file_name,'wb')

    while True:

        content = data_file.read(4096)

        if len(content) == 0:

            break

        cp_file.write(content)

    data_file.close()

    cp_file.close()

    #print('%s文件复制完成！' % file_name)

    queue.put(file_name)

def main():

    source_folder_name = input("请输入文件夹:")

    dest_folder_name = source_folder_name +"附件"

    file_names = os.listdir(source_folder_name)

    #创建一个队里

    queue = multiprocessing.Manager().Queue(128)

    #创建进程池

    p = multiprocessing.Pool(1)

    for file_name in file_names:

        #print('put file_name %s' % file_name)

        p.apply_async(copy_file,args=(queue,file_name,source_folder_name,dest_folder_name))

    p.close()

    #p.join()

    all_file_nums = len(file_names)

    while True:

        file_name = queue.get()

        if file_name in file_names:

            file_names.remove(file_name)

        copy_rate = (all_file_nums - len(file_names)) *100 /all_file_nums

        print("\r%.2f=========>(%s) (%0.2f/%0.2f)" % (copy_rate, file_name,len(file_names),all_file_nums) + " " * 50,end="")

        #\r 默认表示将输出的内容返回到第一个指针，这样的话，后面的内容会覆盖前面的内容

        #sys.stdout.flush()

        #if len(file_names) < 5:

        #    print("remaining: %s" % file_names)

        if copy_rate >= 100:

            break

    print('拷贝文件完成!')

if __name__ == "__main__":

     main()