专栏首页python3多任务拷贝小案例

多任务拷贝小案例

使用到:

进程池(重复利用进程),队列(用于进程间的通信,显示进度)

功能:

拷贝文件夹下的所有文件,显示进度

注意:

如涉及到进程池同时需要使用到队列,使用multiprocessing.Manager().Queue(),而非multiprocessing.Queue()

import os
import multiprocessing


def copy_file(q, file_name, old_folder_name, new_folder_name):
    """完成文件的copy"""
    # print("=====>模拟copy文件 %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():
    # 1.获取用户要copy的文件夹的名字
    old_folder_name = input("请输入要copy的文件夹的名字:")

    # 2.创建一个新的文件夹
    new_folder_name = old_folder_name + "[复件]"
    try:
        os.mkdir(new_folder_name)
    except:
        pass

    # 3.获取文件夹的所有的待copy的文件夹名字 listdir()
    file_names = os.listdir(old_folder_name)
    # print(file_names)

    # 4.创建进程池
    po = multiprocessing.Pool(2)

    # 5.创建一个队列
    q = multiprocessing.Manager().Queue()

    # 6.向进程池中添加cpoy文件的任务
    for file_name in file_names:
        po.apply_async(copy_file, args=(q, file_name, old_folder_name, new_folder_name))

    po.close()
    # 让主进程无线循环的去获取进度,故将join()注释
    # po.join()

    # 获取所有文件的个数
    all_file_num = len(file_names)
    copy_ok_num = 0
    while True:
        file_name = q.get()
        # print("已经完成copy:%s" % file_name)
        copy_ok_num += 1
        print("\r拷贝进度:%.2f%%" % (copy_ok_num/all_file_num*100), end="")
        if copy_ok_num >= all_file_num:
            break

    print()


if __name__ == '__main__':
    main()

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • python遍历文件

    py3study
  • python3第七天(输入和输出)

    输出值的方式:1,表达式。2,print()函数。3,文件对象的write()函数

    py3study
  • python实现磁盘日志清理

    py3study
  • R语言实现eQTL分析

    今天给大家介绍一个GWAS分析过程中的一个重要的环节eQTL(表达数量性状位点)分析。eQTL指的是染色体上一些能特定调控mRNA和蛋白质表达水平的区域,其mR...

    一粒沙
  • [附录代码] shell脚本一步完成多层ssh跳转时的文件传输

    PedroQin
  • 多图带你读懂 Transformers 的工作原理

    Transformer是一类神经网络架构,现在越来越受欢迎了。Transformer最近被OpenAI用于训练他们的语言模型,同时也被DeepMind的Alph...

    AI研习社
  • 使用Python构造hive insert语句说明

    mysql可以使用nevicat导出insert语句用于数据构造,但是hive无法直接导出insert语句。我们可以先打印在hive命令行,然后使用脚本拼装成i...

    砸漏
  • Android性能优化

    讲到Android开发,就不得不谈一下Android的优化,不管是平时开发中我们需要注意的一些Android对Java的一些类的优化,还是实际开发中对性能的优化...

    xiangzhihong
  • Angular2里获取(input file)上传文件的内容

    最近在用Angular2,需要有一个上传文件的功能,但是又不想用库,所以直接用原生的input file 标签。

    Tuzei
  • 万能的AI之根据语音识别人脸

    近期,麻省理工CSAIL(人工智能实验室),最近就发布了这样一个让人难以置信的研究。只需要听6秒的声音片段,AI就能推断出说话者的容貌。 详细解释之前,咱们一起...

    神无月

扫码关注云+社区

领取腾讯云代金券