一个简单的小案例,主要复习一下基础命令。
import multiprocessing
import os,time
'''
获取旧文件夹里的文件名(列表格式),
新建一个文件夹保存复制过来的文件
创建进程池执行读取保存文件方法,记得关闭和等待
创建一个进度条,感觉拉低了运行速度。
'''
def copy_file(q,file_name,old_folder_name,new_folder_name):
'''教程用的是open'''
with open(old_folder_name+ "/" + file_name, "rb") as f:
content = f.read()
with open(new_folder_name + "/" + file_name,"wb") as f:
f.write(content)
#如果拷贝完就像队列中写入一个消息表示完成
q.put(file_name)
def main():
# 获取需要copy的文件夹名
old_folder_name = input("请输入要copy的文件夹名字:")
# 创建一个新的文件夹
try:
new_folder_name = old_folder_name+"复刻版"
os.mkdir(new_folder_name)
except:
pass
# 获取旧文件夹的文件
file_names = os.listdir(old_folder_name)
# 创建进程池
po = multiprocessing.Pool(2)
# 创建一个队列
q = multiprocessing.Manager().Queue()
# 向进程池中添加copy文件的任务
for file_name in file_names:
# print(file_name)
# time.sleep(0.2) #放慢速度看效果
po.apply_async(copy_file,args=(q,file_name,old_folder_name,new_folder_name))
po.close()
po.join()
copy_ok_num=0
while True:
file_name = q.get()
# print("已完成",file_name)
copy_ok_num += 1
print("\r拷贝进度:%.2f %% " % (copy_ok_num*100/len(file_names)), end='')
if copy_ok_num >= len(file_names):
break
print()
if __name__ == '__main__':
main()