第六节 多进程拷贝文件

import os
import multiprocessing
import os
import time

def copy_file(old_file, new_file, q):
        """拷贝数据"""
        old_f = open(old_file, 'rb')
        old_content = old_f.read()
        old_f.close()
        # 打开新文件开始拷贝
        # print(os.listdir(new_file))
        new_filex = open(new_file, 'wb')
        new_filex.write(old_content)
        new_filex.close()
        q.put(1)
        time.sleep(1)

def main():
    # 1、获取文件夹内每个文件名
    old_path = r'G:	est'
    old_filename = os.listdir(old_path)
    # print(old_filename)
    # 2、创建一个新的文件夹用于存储拷贝的文件
    new_path = r'G:\'
    new_filename = 'testcopy'
    new_filename_path = os.path.join(new_path, new_filename)
    os.mkdir(new_filename_path)
    # 3、拷贝函数
    # copy_file(old_filename, old_path, new_filename)
    # 4、创建进程池
    po = multiprocessing.Pool(3)
    q = multiprocessing.Manager().Queue()
    for x in old_filename:
        old_file = os.path.join(old_path, x)
        new_file = os.path.join(new_filename_path, x)
        po.apply_async(copy_file, (old_file, new_file, q))
    po.close()
    # po.join()
    # 6、显示拷贝进度,取消po.join()，通过while循环来阻塞主进程结束
    while True:
        num = q.qsize()
        num_all = len(old_filename)
        print('
拷贝进度为：%.02f %%' % (num*100/num_all))
        if num ==num_all:
            break

if __name__ == '__main__':
    main()