一日一技：命令行进度条是什么原理？

青南

发布于 2021-11-19 11:38:22

9170

发布于 2021-11-19 11:38:22

文章被收录于专栏：未闻Code

摄影：产品经理

这个饭吃起来有点油

今年4月份，我写了一篇文章：字符串离奇消失之谜，有同学在评论里面问我：

\r有什么用？我们今天就来看看，如何使用\r来实现一个简易的命令行进度条。运行效果如下面这个动图所示：

这个SimpleProcessBar是怎么实现的呢？其实它的代码非常简单：

import time


class SimpleProcessBar:
    def __init__(self, total, finished='■', unfinished='□'):
        self.total = total
        self.finished = 0
        self.unfinished_char = unfinished
        self.finished_char = finished
        self.bar = [self.unfinished_char] * 100

    def incr(self, value=1):
        self.finished += value
        if self.finished > 100:
            self.finished = 100
        bar = self.generate_bar()
        print(bar, end='\r')

    def generate_bar(self):
        finished = int(self.finished / self.total * 100)
        self.bar[:finished] = self.finished_char * finished
        return ''.join(self.bar) + f' {finished} %'

x = SimpleProcessBar(30)
for i in range(30):
    x.incr()
    time.sleep(1)

真正的核心代码就只有一行：print(bar, end='\r')。其它代码都是辅助这一行的。我们试一试直接手动执行一下：

import time

print("□" * 10, end='\r')
time.sleep(1)
print(f'{"■" * 3}{"□" * 7}', end='\r')
time.sleep(1)
print(f'{"■" * 6}{"□" * 4}', end='\r')
time.sleep(1)
print(f'{"■" * 9}{"□" * 1}', end='\r')
time.sleep(1)
print("■" * 10, end='\r')

运行效果如下所示：