# @Time: 2020/1/17
# @Author: Ellen
# @File: 多任务文件夹复制.py
# @Software: PyCharm
'''
多任务文件夹复制
1.获取用户要复制的文件夹名字
2.创建一个新的文件夹
3.获取文件夹待拷贝的文件名字 读文件再写
4.创建进程池
5.添加拷贝任务
'''
import multiprocessing
import os
def copy_file(file_name, new_folder_name, old_folder_name):
'''完成文件拷贝'''
print('拷贝的文件名称为:%s' % file_name)
with open(old_folder_name + '/' + file_name, 'rb') as f:
content = f.read()
# old_file = open(old_folder_name + '/' + file_name,'rb')
# content = old_file.read()
# old_file.close()
# 保存到新的文件夹中
new_file = open(new_folder_name + '/' + file_name, 'wb')
new_file.write(content)
new_file.close()
def main():
# 获取用户要复制的文件夹名字 test
old_folder_name = input('请输入要复制的文件夹名字:')
# 创建一个新的文件夹 test[复件]
new_folder_name = old_folder_name + '复件'
if not os.path.exists(new_folder_name):
os.mkdir(new_folder_name)
# 获取文件夹的所有待拷贝的文件名字
file_names = os.listdir(old_folder_name)
print(file_names)
# 创建进程池
po = multiprocessing.Pool(3)
# 添加拷贝任务
for file_name in file_names:
po.apply_async(copy_file, args=(file_name, new_folder_name, old_folder_name))
po.close()
po.join()
if __name__ == '__main__':
main()
执行结果:
请输入要复制的文件夹名字:demo
[‘ast.py’, ‘_bootlocale.py’, ‘_collections_abc.py’, ‘_compat_pickle.py’, ‘_compression.py’, ‘_dummy_thread.py’, ‘_markupbase.py’, ‘_osx_support.py’, ‘future.py’, ‘phello.foo.py’]
拷贝的文件名称为:ast.py
拷贝的文件名称为:_bootlocale.py
拷贝的文件名称为:_collections_abc.py
拷贝的文件名称为:_compat_pickle.py
拷贝的文件名称为:_compression.py
拷贝的文件名称为:_dummy_thread.py
拷贝的文件名称为:_markupbase.py
拷贝的文件名称为:_osx_support.py
拷贝的文件名称为:future.py
拷贝的文件名称为:phello.foo.py
Process finished with exit code 0