使用多进程(multiprocess)进行文件夹下文件的拷贝

    技术2022-07-10  149

    使用多进程(multiprocess)进行文件夹下文件的拷贝

    import os import multiprocessing # 多多进程的python 包 def copy_file(q, source_file, old_fold_name, des_folder): # 完成文件复制 # print('---------' + source_file) old_file = open(old_fold_name + '/' + source_file, "rb") content = old_file.read() old_file.close() new_file = open(des_folder + '/' + source_file, 'wb') new_file.write(content) new_file.close() q.put(source_file) pass def main(): # 1.获取用户想要复制的文件夹的名字 old_folder_name = input("请输入要复制的文件夹的名字:") # 2. 创建一个新的文件夹 try: os.mkdir(old_folder_name + "[copy]") except: pass # 3. 获取文件夹的所有的代copy的文件夹的名字 listdir() file_names = os.listdir(old_folder_name) # print(file_names) # 4. 创建进程池,最多同时执行5个进程 po = multiprocessing.Pool(5) # 5. 创建一个队列 q = multiprocessing.Manager().Queue() # 5. 向进程池中添加copy文件的任务 for file_name in file_names: po.apply_async(copy_file, args=(q, file_name, old_folder_name, old_folder_name + "[copy]")) po.close() # po.join() # 让主线程等待,防止主进程结束导致子进程结束 file_num = len(file_names) copy_ok_num = 0 # 下面的while用来显示进度,属于附加功能 while True: file_name = q.get() copy_ok_num += 1 print("\r完成进度:%.2f %%" % (copy_ok_num * 100 / file_num),end='') if copy_ok_num >= file_num: break if __name__ == "__main__": main()
    Processed: 0.013, SQL: 9