- 使用进程的思想来实现多任务,从而来完成一个简单的文件下载器,可以实现将一个文件中的所有文件下载到指定文件夹下
- 代码实例
import os
import multiprocessing
def copy_file(q, file_name, old_folder_name, new_folder_name):
old_f = open(old_folder_name + "/" + file_name, "rb")
content = old_f.read()
old_f.close()
new_f = open(new_folder_name + "/" +file_name, "wb")
new_f.write(content)
new_f.close()
q.put(file_name)
def main():
old_folder_name = input("请输入要拷贝的文件夹名字:")
try:
new_folder_name = old_folder_name + "_other"
os.mkdir(new_folder_name)
except:
pass
file_names = os.listdir(old_folder_name)
po = multiprocessing.Pool(5)
q = multiprocessing.Manager().Queue()
for file_name in file_names:
po.apply_async(copy_file, (q, file_name, old_folder_name, new_folder_name))
po.close()
all_file_num = len(file_names)
cp_file_num = 0
temp = 1
while True:
q.get()
print("\r拷贝的进度为%.2f%%" % ((cp_file_num + 1) / all_file_num * 100),end="")
cp_file_num += 1
if cp_file_num == all_file_num:
print("")
break
if __name__ == '__main__':
main()