python核心---多进程copy文件

最新推荐文章于 2024-11-28 09:12:09 发布

原创最新推荐文章于 2024-11-28 09:12:09 发布 · 202 阅读

0 ·

CC 4.0 BY-SA版权

python核心专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种利用Python多进程技术实现快速文件夹复制的方法。通过获取源文件夹名称、创建目标文件夹、列出所有文件并使用进程池进行并行复制，显著提高了复制效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# 1.获取要copy的文件夹的姓名
# 2.创建一个文件夹
# 3.获取old文件夹中的所有的文件名字
# 4.使用多进程的方式copy原文件夹中的所有文件到新文件夹中

# 导入需要的模块
from multiprocessing import Pool, Manager
import os


def copyFileTask(name, oldFolderName, NewFolderName, queue):
    # 完成copy一个文件的功能
    fr = open(oldFolderName + '/' + name)
    fw = open(NewFolderName + '/' + name, 'w')

    content = fr.read()
    fw.write(content)

    fr.close()
    fw.close()
    queue.put(name)


def main():
    # 获取要copy的文件夹名字1
    oldFolderName = input("请输入文件夹的名字:")
    # 创建一个文件夹
    newFolderName = oldFolderName + '-复件'
    os.mkdir(newFolderName)
    # 获取old文件夹中所有的文件名字
    fileNames = os.listdir(oldFolderName)
    # 使用多进程的方式copy原文件夹中的所有文件到文件夹中
    pool = Pool(5)
    queue = Manager().Queue()

    for name in fileNames:
        pool.apply_async(copyFileTask, args=(name, oldFolderName, newFolderName, queue))

    num = 0
    allNum = len(fileNames)
    while True:
        queue.get()
        num += 1
        copyRate = num/allNum
        print('copy的进度为:%.2f%%' %(copyRate*100), end='')
        if num == allNum:
            print('复制完成')
            break


if __name__=='__main__':
    main()