python 多进程读取文件
Python实现的多进程拷贝文件并显示百分比功能示例本文实例讲述了python实现的多进程拷贝文件并显示百分比功能。分享给大家供大家参考，具体如下：
centos7下查看cup核数：
|
|
# 总核数 = 物理cpu个数 x 每颗物理cpu的核数# 总逻辑cpu数 = 物理cpu个数 x 每颗物理cpu的核数 x 超线程数# 查看物理cpu个数cat /proc/cpuinfo| grep "physical id"| sort| uniq| wc -l# 查看每个物理cpu中core的个数(即核数)cat /proc/cpuinfo| grep "cpu cores"| uniq# 查看逻辑cpu的个数cat /proc/cpuinfo| grep "processor"| wc -l |
本示例拷贝文件采用的是最简单的但是有最复杂的方式：
1，根据要拷贝的文件夹名称，创建一个新的文件夹用来存储文件
2，读取源文件夹下的文件内容，将其写入目的文件夹下同名的文件中
直接上代码：
|
|
# coding=utf-8from multiprocessing import pool, managerimport osimport tracebackimport timedef copyfiletask(name, oldfoldername, newfoldername, queue): """copy文件""" fr = open(oldfoldername+"/"+name) fw = open(newfoldername+"/"+name, "w") content = fr.read() fw.write(content) fr.close() fw.close() # 将copy完成的文件名称put进队列 queue.put(name)def main(): try: # 0. 获取要copy的文件夹 # oldfoldername = input("请输入文件夹的名字：") # python3 oldfoldername = raw_input("请输入文件夹的名字：") # python2 # 1. 创建一个新的文件夹 print 'mkdir new file ------' newfoldername = str(oldfoldername) + "-copy" # print(newfoldername) os.mkdir(newfoldername) # 2. 获取old文件夹中的所有的文件名字 filenames = os.listdir(oldfoldername) # print(filenames) # 3. 使用多进程的方式copy 原文件夹中的所有文件到新的文件夹中 pool = pool(5) # 创建一个队列 queue = manager().queue() for name in filenames: pool.apply_async(copyfiletask, args=(name, oldfoldername, newfoldername, queue)) pool.close() # 关闭进程池，不再接受请求 pool.join() # 等待所有的子进程结束 num = 0 # 需要copy的文件总数 allnum = len(filenames) while num < allnum: # 收数据 queue.get() num += 1 copyrate = float(num) / allnum # 打印copy的进度 time.sleep(0.1) # 容易看出来打印百分比的变化，延长百分比更新的时间 print "\rcopy的进度是:%.2f%%" % (copyrate*100), print "\ncopy success ------" except: traceback.print_exc()if __name__ == "__main__": main() |
下面采用python模块shutil模块进行copy操作：
|
|
# coding=utf-8from multiprocessing import pool, managerimport osimport tracebackimport timeimport shutildef copyfiletask(name, old_path, new_path, queue): """copy文件""" src_path = os.path.join(old_path, name) dst_path = os.path.join(new_path, name) shutil.copy(src_path, dst_path) # 拷贝文件 # shitil.move(src_path, dst_path) # 移动文件 # 将copy完成的文件名称put进队列 queue.put(name)def main(): try: # 0. 获取要copy的文件夹 oldfoldername = raw_input("请输入文件夹的名字：") print '要拷贝的源文件夹： ', oldfoldername # 1. 创建一个新的文件夹 newfoldername = raw_input("请输入文件夹的名字：") print '目标文件夹：', newfoldername # 获取当前所在的路径 now_path = os.getcwd() print "当前所在路径： ", now_path old_path = os.path.join(now_path, oldfoldername) print '源文件夹路径： ', old_path new_path = os.path.join(now_path, newfoldername) print '目标文件夹路径： ', new_path if not os.path.exists(new_path): os.mkdir(new_path) # 3. 使用多进程的方式copy 原文件夹中的所有文件到新的文件夹中 pool = pool(5) # 创建5条进程 queue = manager().queue() # 创建一个队列 # 遍历源文件夹 for root, dirs, files in os.walk(oldfoldername): """ root 所指的是当前正在遍历的这个文件夹的本身的地址 dirs 是一个 list ，内容是该文件夹中所有的目录的名字(不包括子目录) files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录) """ print '源文件夹下的文件数量:：', len(files) s_t = time.time() for file in files: pool.apply_async(copyfiletask, args=(file, old_path, new_path, queue)) # 向进程池中添加任务 pool.close() # 关闭进程池，不再接受请求 pool.join() # 等待所有的子进程结束 print '耗时：{} 秒'.format(time.time() - s_t) num = 0 # 需要copy的文件总数 allnum = len(files) print 'len(filenames) = ', allnum while true: # 收数据 # print '收数据 ------' queue.get() # print 'queue.get() ------', queue.get() num += 1 copyrate = float(num) / allnum # 打印copy的进度 print "\rcopy的进度是:%.2f%%" % (copyrate*100), if num == allnum: break print "\ncopy success ------" except: traceback.print_exc()if __name__ == "__main__": main() |
输出：（由于我的cup核数为1核，在使用多进程的过程中，操作系统在轮询的时候，频繁的切换任务，浪费时间导致多进程比单进程耗时更长。）
[root@centos7 mnt]# python copy_file_test.py
请输入文件夹的名字：hdfs
要拷贝的源文件夹： hdfs
请输入文件夹的名字：test01
目标文件夹： test01
当前所在路径： /mnt
源文件夹路径： /mnt/hdfs
目标文件夹路径： /mnt/test01
源文件夹下的文件数量:： 8165
耗时：253.655323982 秒
len(filenames) = 8165
copy的进度是:100.00%
copy success ------
单进程拷贝文件：
|
|
# coding=utf-8from queue import queueimport osimport tracebackimport timeimport shutildef copyfiletask(name, old_path, new_path, queue): """copy文件""" src_path = os.path.join(old_path, name) dst_path = os.path.join(new_path, name) shutil.copy(src_path, dst_path) # 拷贝文件 # shitil.move(src_path, dst_path) # 移动文件 # 将copy完成的文件名称put进队列 queue.put(name)def main(): try: # 0. 获取要copy的文件夹 oldfoldername = raw_input("请输入文件夹的名字：") print '要拷贝的源文件夹： ', oldfoldername # 1. 创建一个新的文件夹 newfoldername = raw_input("请输入文件夹的名字：") print '目标文件夹：', newfoldername # 获取当前所在的路径 now_path = os.getcwd() print "当前所在路径： ", now_path old_path = os.path.join(now_path, oldfoldername) print '源文件夹路径： ', old_path new_path = os.path.join(now_path, newfoldername) print '目标文件夹路径： ', new_path if not os.path.exists(new_path): os.mkdir(new_path) queue = queue() # 创建一个队列 # 遍历源文件夹 for root, dirs, files in os.walk(oldfoldername): """ root 所指的是当前正在遍历的这个文件夹的本身的地址 dirs 是一个 list ，内容是该文件夹中所有的目录的名字(不包括子目录) files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录) """ print '源文件夹下的文件数量:：', len(files) s_t = time.time() # 调用函数拷贝文件 for file in files: copyfiletask(file, old_path, new_path, queue) print '耗时：{} 秒'.format(time.time() - s_t) num = 0 # 需要copy的文件总数 allnum = len(files) print 'len(filenames) = ', allnum while true: # 收数据 # print '收数据 ------' queue.get() # print 'queue.get() ------', queue.get() num += 1 copyrate = float(num) / allnum # 打印copy的进度 print "\rcopy的进度是:%.2f%%" % (copyrate*100), if num == allnum: break print "\ncopy success ------" except: traceback.print_exc()if __name__ == "__main__": main() |
输出：
[root@centos7 mnt]# python copy_file_test_dan.py
请输入文件夹的名字：hdfs
要拷贝的源文件夹： hdfs
请输入文件夹的名字：test02
目标文件夹： test02
当前所在路径： /mnt
源文件夹路径： /mnt/hdfs
目标文件夹路径： /mnt/test02
源文件夹下的文件数量:： 8165
耗时：122.284090996 秒
len(filenames) = 8165
copy的进度是:100.00%
copy success ------
我的虚拟机cpu核数：1
如下：
|
|
[root@centos7 mnt]# cat /proc/cpuinfo| grep "physical id"| sort| uniq| wc -l1[root@centos7 mnt]# cat /proc/cpuinfo| grep "cpu cores"| uniqcpu cores : 1[root@centos7 mnt]# cat /proc/cpuinfo| grep "processor"| wc -l1 |
希望本文所述对大家python程序设计有所帮助。
原文链接：https://blog.csdn.net/xuezhangjun0121/article/details/88777303