继续浏览精彩内容
慕课网APP
程序员的梦工厂
打开
继续
感谢您的支持,我会继续努力的
赞赏金额会直接到老师账户
将二维码发送给自己后长按识别
微信支付
支付宝支付

124-带进度条的文件拷贝

慕姐8265434
关注TA
已关注
手记 1147
粉丝 221
获赞 1064

中介绍了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.pyimport osimport sysfrom tqdm import tqdmdef 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])



作者:凯茜的老爸
链接:https://www.jianshu.com/p/da813405f582


打开App,阅读手记
0人推荐
发表评论
随时随地看视频慕课网APP