Python 彩色终端

简介

Python程序运行在终端上时,只有背景色和前景色2种颜色,通常是黑底白字

我们可以使用 rich 这个库,让 Python程序有更多的色彩和样式,视觉上更好看。

如果你是在Windows上运行程序,推荐使用 Windows terminal 终端软件, 而不是传统的 Console Host 。 因为前者对终端功能的支持更丰富。


安装 rich 库很简单, 执行 pip install rich 即可

进度条

rich支持彩色进度条。

如果,只有一个任务需要显示进度,可以使用快捷函数 track

比如,

import time
from rich.progress import track

for i in track(range(20), description="进度..."): 
    time.sleep(0.5) # 模拟执行任务耗费的时间

上面代码,for 循环每执行一次,进度条对应循环的完成次数

比如,上例中一共 循环 20次, 第1次循环完,进度条完成 1/20 第2次循环完,进度条完成 2/20 第3次循环完,进度条完成 3/20


sleep是模拟执行任务耗费的时间,真实有对应的任务,不需要sleep的。


下面是一个批量下载文件的代码示例

import urllib.request
from rich.progress import track

for i in track(range(5,16), description="进度..."):    
    url = f"https://mirrors.kernel.org/gnu/cpio/cpio-2.{i}.tar.gz"
    print(url)
    content = urllib.request.urlopen(url).read()
    with open(f'd:/tmp/3/cpio-2.{i}.tar.gz','wb') as f:
        f.write(content)

如果你不喜欢默认进度条色彩,可以给track函数 指定颜色参数,比如
track(range(5,16), description="进度...",
        complete_style="green", finished_style="green")

complete_style 和 finished_style 值分别指定了未完成和结束时的进度条前景色。可以指定一种颜色,支持256种单词表示颜色


如果你同时执行要显示多个任务的进度,需要多个进度条

import time
from rich.progress import Progress

with Progress() as progress:

    task1 = progress.add_task("[red]下载...", total=10)
    task2 = progress.add_task("[green]处理...", total=10)
    task3 = progress.add_task("[yellow]保存...", total=10)

    while not progress.finished:
        progress.update(task1, advance=0.5)
        progress.update(task2, advance=0.2)
        progress.update(task3, advance=1)
        time.sleep(1)

add_task 方法的 total指明了任务的总体进度值

update 方法 的 advance 参数 表示增加多少完成量,

也可以使用 completed 参数,表示目前该任务总共的完成量。这个值就是进度条已完成部分的长度。

比如

import time
from rich.progress import Progress

with Progress() as progress:

    task1 = progress.add_task("[red]下载...", total=10)
    task2 = progress.add_task("[green]处理...", total=10)
    task3 = progress.add_task("[yellow]保存...", total=10)

    t1 = 0
    while not progress.finished:
        t1 += 2
        progress.update(task1, completed=t1)
        progress.update(task2, advance=0.2)
        progress.update(task3, advance=1)
        time.sleep(1)
上一页