Python腳本使用進度條

在下載下傳某些檔案的時候你一定會不時盯着進度條，在寫代碼的時候使用進度條可以便捷的觀察任務處理情況，除了使用print來列印之外，今天本文就介紹幾種給你的Python代碼加上酷炫的進度條的方式。

自定義ProgressBar

最原始的辦法就是不借助任何第三方工具，自己寫一個進度條函數，使用time子產品配合sys子產品即可

import sys
import time
def progressbar(it, prefix="", size=60, file=sys.stdout):
    count = len(it)
    def show(j):
        x = int(size*j/count)
        file.write("%s[%s%s] %i/%i\r" % (prefix, "#"*x, "."*(size-x), j, count))
        file.flush()        
    show(0)
    for i, item in enumerate(it):
        yield item
        show(i+1)
    file.write("\n")
    file.flush()
    
for i in progressbar(range(15), "Computing: ", 40):
    do_something()
    time.sleep(0.1)      

自己定義的好處就是可以将進度條定義成我們想要的形式比如上面就是使用#與·來輸出，為什麼不用print？因為sys.stdout就是print的一種預設輸出格式，而sys.stdout.write()可以不換行列印，sys.stdout.flush()可以立即重新整理輸出的内容。當然也可以封裝成類來更好的使用[1] ，但效果是類似的。

from __future__ import print_function
import sys
import re
class ProgressBar(object):
    DEFAULT = 'Progress: %(bar)s %(percent)3d%%'
    FULL = '%(bar)s %(current)d/%(total)d (%(percent)3d%%) %(remaining)d to go'
    def __init__(self, total, width=40, fmt=DEFAULT, symbol='=',
                 output=sys.stderr):
        assert len(symbol) == 1
        self.total = total
        self.width = width
        self.symbol = symbol
        self.output = output
        self.fmt = re.sub(r'(?P<name>%\(.+?\))d',
            r'\g<name>%dd' % len(str(total)), fmt)
        self.current = 0
    def __call__(self):
        percent = self.current / float(self.total)
        size = int(self.width * percent)
        remaining = self.total - self.current
        bar = '[' + self.symbol * size + ' ' * (self.width - size) + ']'
        args = {
            'total': self.total,
            'bar': bar,
            'current': self.current,
            'percent': percent * 100,
            'remaining': remaining
        }
        print('\r' + self.fmt % args, file=self.output, end='')
    def done(self):
        self.current = self.total
        self()
        print('', file=self.output)
        
from time import sleep
progress = ProgressBar(80, fmt=ProgressBar.FULL)
for x in range(progress.total):
    progress.current += 1
    progress()
    sleep(0.1)
progress.done()      

tqdm

之前我們說了，自定義的好處就是可以自己修改，那麼使用第三方庫的好處就是可以偷懶，不用自己寫，拿來就能用。比如提到Python進度條那肯定會想到常用的tqdm，安裝很簡單pip install tqdm即可，使用也很簡單，幾行代碼即可實作上面的進度條

from tqdm import trange
import time
for i in trange(10): 
    time.sleep(1)      

當然tqdm作為老牌的Python進度條工具，循環處理、多程序、多線程、遞歸處理等都是支援的，你可以在官方GitHub上學習[2] 、解鎖更多的玩法。

Rich

上面兩種實作Python進度條的方法都學會了嗎，雖然簡單但是看上去并不漂亮，顔色也比較單調。是以最後壓軸出場的就是一款比較小衆的第三方庫Rich[3] 。Rich主要是用于在終端中列印豐富多彩的文本(最高支援1670萬色)

是以當然可以使用Rich列印進度條，顯示完成百分比，剩餘時間，資料傳輸速度等都可以。并且樣式更加酷炫，并且它是高度可配置的，是以我們可以對其進行自定義以顯示所需的任何資訊。使用也很簡單，比如我們使用Rich來實作一個最簡單的進度條

from rich.progress import track
import  time
for step in track(range(30)):
    print('Python')
    time.sleep(0.5)      

同時Rich支援多個進度條，這在多任務情況下監控的進度很有用(使用方法見官方文檔)