Python实现控制台中的进度条

相信很多人在写一些简单的python脚本的时候都希望能够在程序运行的过程中实现进度条的功能以便查看程序运行的速度或者进度

我之前一直想实现这样一个东西,也查看了许多博客但是都找不到一个完美的解决方案(当然,使用progressBar这个库是个选择,但很多时候我们需要一些定制功能的时候就需要考虑自己实现,其实也挺简单的,不想看废话的可以直接跳到最后

进度条最主要的问题就是所有字符全部在同一行,而且可以修改

然而当执行print语句的时候,python会在打印完这个语句的同时在结尾加上’\n’,也就是换行,这就导致在控制台下一旦被print之后就无法再修改了。所以我们现在的输出就不能再使用print来完成了。

我们要使用的是来自sys库的sys.stdout.write()函数,这个函数会在控制台输出这个字符串的同时不加上任何结尾,这就意味着这个输出还没有完全结束。通过sys.stdout.flush()函数可以把输出暂时打印在控制台中(造成print的假象,我们姑且先叫这个假输出)。那么如果我们使用’r’这个转义字符(回到行首),一切看起来是不是就合理很多了呢?

也就是说:打印字符串的时候,没有加上’\n’,同时让光标回到行首,再把当前缓冲区显示出来,也就好象是print了一样,但是这时候光标还在原来的位置。

举个例子:

在终端下执行这段代码就会得到简单的进度条效果。


 

接下来还需要解决两个问题:

一:清空缓冲区

有些聪明的读者可能发现,当新的字符串比之前短的时候会出现问题,比如下面这段代码:

运行后发现结果跟我们希望的不太一样。

其实是因为已经被flush出去的字符并不会主动清空,所以只有新写入的被修改了。针对这点我目前的解决方案是先输出一波空格把之前的字符串冲掉然后重新写:

二:固定底边输出

有时候我们希望在进度条加载的同时还有一些其他的输出。

我们不妨在刷新掉上一次输出之后输出所需输出的字符串,然后在假输出进度条。

采用如下代码:

就可以完成所需任务了。


怎么样,其实原理还是挺简单的吧?

这里给出一个自己实现的类用来打印进度条:

效果如下:


这样就可以方便的在一些任务中查看程序运行的进度了,比如爬虫、机器学习等并不知道要花多少时间等工作也都可以有形象的时间把握了。

如果有什么其他好的建议欢迎共同讨论~

3 8 收藏 12 评论

相关文章

可能感兴趣的话题



直接登录
最新评论
  • G.K. IT互联网 2015/12/18

    为什么我在python2.7 Windows环境下运行没有得到预期效果?

  • G.K. IT互联网 2015/12/18

    代码和原博客有出入 请改正

  • 幻影   2015/12/18

    第17行应为:sys.stdout.write(' ' * (self.width + 9) + '\r')
    第22行应为:sys.stdout.write('#' * progress + '-' * (self.width - progress) + '\r')
    第24行应为:sys.stdout.write('\n')
    都缺少了转义符号

  • zhgxun PHP开发 2015/12/19

    正在自己学习python,用的是3.5.1版本,亲测第19很明显,20行需要转化一下类型progress = self.width * self.count / self.total,默认是float,需要强制转换为int()后才有效果。效果棒棒的。

  • #!/bin/env python
    #-*- coding:utf-8 -*-
    import sys
    class Progressbar():
    def __init__(self,length,max=100):
    self.length=length
    self.max=max
    @property
    def max(self):
    return self.max
    @max.setter
    def max(self,max):
    self.max=max

    def update(self,num,**kw):
    self._len=num*self.length/self.max
    m=self.length-self._len
    percent=num*100/self.max
    bar='\r\33[K[%s|%s]:%s/%s|%s%%'%('#'*self._len,' '*m,num,self.max,percent)
    msg=kw.get('msg',None)
    line=kw.get('line',1)
    if msg:
    bar=('\r\33[K\33[1A'*line)+bar
    bar+=('\n'+msg)
    if num==self.max:
    bar+='\n'
    sys.stdout.write(bar)
    sys.stdout.flush()

    if __name__=='__main__':
    import time
    pb=Progressbar(length=80)
    for i in range(11):
    time.sleep(1)
    pb.update(i,msg='正在处理第%s条任务!'%i)

    这个是我写的progressbar,使用linux的shell的控制码~可以改变光标上移,清屏,并可以附带一则消息。
    只实用于Xinux

  • gnlm_admin 程序员 2015/12/27

    不错 不错!

  • 紫依   01/10

    不错不错,运行成功了

跳到底部
返回顶部