python多进程文件备份

19881103liujia

1. from multiprocessing import  Pool,Manager
   
2. import  os
   
3. 
   
4. def copyFileTask(name,oldFileName,newFilename,queue):
   
5.     #print(name)
   
6.     fr=open(oldFileName+"\"+name,"r")
   
7.     fw =open(newFilename+"\"+name, "w")
   
8.     fileData=fr.read()
   
9.     fw.write(fileData)
   
10.     fr.close()
    
11.     fw.close()
    
12.     queue.put(name)
    
13. 
    
14. 
    
15. def main():
    
16.     #0获取文件夹名字
    
17.     oldName=input("请输入文件夹名字:")
    
18.     #1创建一个空文件夹
    
19.     newName=oldName+"--备份"
    
20.     os.mkdir(newName)
    
21. 
    
22.     #2获取old文件夹所有的文件名
    
23.     fileNames=os.listdir(oldName)
    
24. 
    
25.     #3多进程的方式copy文件
    
26.     pool=Pool(5)
    
27.     queue=Manager().Queue()  #进程池里用消息队列需要用Manager类
    
28.     i = 0
    
29.     for name in fileNames:
    
30.         pool.apply_async(copyFileTask,args=(name,oldName,newName,queue))
    
31.         i+=1
    
32. 
    
33.    #4获取备份情况
    
34.     while True:
    
35.         getName=queue.get()
    
36.         if getName !="":
    
37.             print("正在备份-->%s"%getName)
    
38.             i-=1
    
39.             if i==0:
    
40.                 break
    
41.     pool.close()
    
42.     pool.join()
    
43.     print("备份完毕")
    
44. 
    
45. 
    
46. if __name__=="__main__":
    
47.     main()

复制代码