一個多進程拷貝某個目錄下的文件的程序代碼(有進度顯示)
import os
import multiprocessing
def copy_file(q, file_name, old_folder_name, new_folder_name):
# 完成文件的複製的任務
# print("=========>>>>>模擬拷貝文件:%s------->%s 文件名:%s" % (old_folder_name, new_folder_name, file_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 + "副本"
os.mkdir(new_folder_name)
except:
pass
# 獲取文件夾中要拷貝的文件的名字
file_names = os.listdir(old_folder_name)
#print(file_names)
# 創建進程池
po = multiprocessing.Pool(5)
# 創建隊列
q = multiprocessing.Manager().Queue()
# 向進程池中添加要拷貝文件的任務
for file_name in file_names:
po.apply_async(copy_file, args=(q, file_name, old_folder_name, new_folder_name))
po.close()
# po.join()
all_file_num = len(file_names)
copy_ok_num = 0
while True:
file_name = q.get()
# print("已經完成複製:%s" % file_name)
copy_ok_num += 1
print("\r拷貝的進度:%.2f%%" % (copy_ok_num*100 / all_file_num),end="")
if copy_ok_num >= all_file_num:
break
print()
if __name__ == '__main__':
main()
大家在寫的時候注意下文件的路徑問題