124-带进度条的文件拷贝

【123-进度条】中介绍了tqdm进度条,那么怎么把它与其他代码结合起来使用呢?
下面举一个在拷贝过程中增加进度条的示例。
首先,确定拷贝文件时每次从源文件读取数据的长度length,如4096字节。
然后,确定需要从源文件读取多少次。读取次数用源文件的总大小除以length即可得到。源文件大小可以这样获得:

[root@room8pc16 ~]# python3
>>> import os
>>> os.stat('/bin/ls')
os.stat_result(st_mode=33261, st_ino=134897105, st_dev=64768, st_nlink=1, st_uid=0, st_gid=0, st_size=117616, st_atime=1545969810, st_mtime=1447997566, st_ctime=1494055224)
# 上面输出的st_size就是文件大小
>>> size = os.stat('/bin/ls').st_size
>>> print(size)
117616

获取了文件的大小,接下来求出读取次数times:

>>> length = 4096
>>> times, extra = divmod(size, length)   # 同时获取商和余数
>>> if extra:
...     times += 1   #  如果余数不为0,次数加1

有了上面的思路,下面看看完整的代码:

[root@room8pc16 ~]# vim /tmp/cp.py
import os
import sys
from tqdm import tqdm

def copy(src_fname, dst_fname, length=4096):
    size = os.stat(src_fname).st_size
    times, extra = divmod(size, length)
    if extra:
        times += 1

    with open(src_fname, 'rb') as src_fobj:
        with open(dst_fname, 'wb') as dst_fobj:
            for i in tqdm(range(times)):
                data = src_fobj.read(length)
                dst_fobj.write(data)

if __name__ == '__main__':
    copy(sys.argv[1], sys.argv[2])
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 一、Python简介和环境搭建以及pip的安装 4课时实验课主要内容 【Python简介】: Python 是一个...
    _小老虎_阅读 5,863评论 0 10
  • Swift1> Swift和OC的区别1.1> Swift没有地址/指针的概念1.2> 泛型1.3> 类型严谨 对...
    cosWriter阅读 11,160评论 1 32
  • 一、目录结构 1.Windows文件系统 看到的是一个个驱动器盘符,例如:C盘,D盘等每个驱动器都有自己的根目录结...
    郑元吉阅读 684评论 0 0
  • 今年九月我准备放下所有,决心笃定去做我这一生都想完成的事情,我喜欢那个地方,喜欢那里的生活,喜欢那里的方式,我愿...
    未央_0db9阅读 376评论 0 0
  • 0. Vector是什么 动态数组 实现List接口 内容改变相关的方法均为synchronized 1. 实现的...
    烟小花飞花阅读 216评论 0 0