前往小程序,Get更优阅读体验!
立即前往
发布
社区首页 >专栏 >124-带进度条的文件拷贝

124-带进度条的文件拷贝

作者头像
凯茜的老爸
发布2019-01-03 15:15:37
发布2019-01-03 15:15:37
1.3K00
代码可运行
举报
文章被收录于专栏:python百例python百例
运行总次数:0
代码可运行

【123-进度条】中介绍了tqdm进度条,那么怎么把它与其他代码结合起来使用呢?

下面举一个在拷贝过程中增加进度条的示例。

首先,确定拷贝文件时每次从源文件读取数据的长度length,如4096字节。

然后,确定需要从源文件读取多少次。读取次数用源文件的总大小除以length即可得到。源文件大小可以这样获得:

代码语言:javascript
代码运行次数:0
复制
[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:

代码语言:javascript
代码运行次数:0
复制
>>> length = 4096
>>> times, extra = divmod(size, length)   # 同时获取商和余数
>>> if extra:
...     times += 1   #  如果余数不为0,次数加1

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

代码语言:javascript
代码运行次数:0
复制
[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])
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2018.12.29 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档