Python使用Rich實(shí)現(xiàn)美化終端顯示效果

更新時(shí)間：2024年02月13日 09:09:34 作者：wang_yb

Rich庫(kù)的功能就像它的名字一樣,使Python編程更加豐富（rich）,用來(lái)幫助開(kāi)發(fā)者在控制臺(tái)（命令行）輸出中創(chuàng)建豐富、多彩和具有格式化的文本,下面我們就來(lái)了解下它的具體使用吧

1. 安裝

通過(guò)pip安裝：

pip install rich

使用下面的命令驗(yàn)證是否安裝成功。

python -m rich

2. 應(yīng)用示例

Rich的功能很多，下面通過(guò)代碼示例來(lái)演示其中主要的功能。

2.1. 美化 REPL 輸出

安裝python之后，在命令行輸入python，就可以進(jìn)入python的交互式命令行環(huán)境（REPL）。

因?yàn)?code>python是解釋性語(yǔ)言，所以可以在REPL環(huán)境中交互式的運(yùn)行代碼：

注：REPL全稱: Read-Eval-Print-Loop (交互式解釋器)

默認(rèn)的REPL是沒(méi)有顏色的，使用Rich可以美化REPL的顯示，獲得更好的交互效果。

只需要導(dǎo)入Rich庫(kù)的pretty即可。

>>> from rich import pretty
>>> pretty.install()

再次運(yùn)行上面的代碼：

不同的數(shù)據(jù)類型會(huì)用不同的顏色來(lái)表示。

2.2. 查看對(duì)象信息

Rich庫(kù)中還提供了一個(gè)還有用的功能，用來(lái)查看python中各個(gè)變量或?qū)ο蟮脑敿?xì)信息。

使用前導(dǎo)入 inspect 函數(shù)。

>>> from rich import inspect

對(duì)于變量，查看其概括信息的方式如下：

>>> inspect(lst_var)

查看其包含的方法：

>>> inspect(lst_var, methods=True)

對(duì)于對(duì)象，也是一樣：

# -*- coding: utf-8 -*-
from rich import inspect

class Sample:
    def __init__(self, name, age):
        self._name = name
        self._age = age

    def info(self):
        print("姓名: {}， 年齡: {}".format(self.name, self.age))

    def set(self, name, age):
        self.name = name
        self.age = age

    def get(self):
        return {"name": self.name, "age": self.age}

    # 私有函數(shù)
    def _private(self):
        print("這是私有函數(shù)")

if __name__ == "__main__":
    sa = Sample("harry", 33)

    # 顯示對(duì)象概要信息
    inspect(sa)

    # 顯示對(duì)象方法信息
    inspect(sa, methods=True)

    # 顯示對(duì)象方法和私有變量，私有函數(shù)
    inspect(sa, methods=True, private=True)

Rich庫(kù)的inspect函數(shù)讓我們?cè)诿钚兄幸部梢垣@得非常好的閱讀體驗(yàn)。

2.3. 動(dòng)態(tài)顯示內(nèi)容

動(dòng)態(tài)顯示在命令行中一直是個(gè)難點(diǎn)，而Rich庫(kù)能幫助我們很容易的實(shí)現(xiàn)狀態(tài)和進(jìn)度的動(dòng)態(tài)顯示。

比如，如果有多個(gè)任務(wù)在執(zhí)行，可以用Rich來(lái)動(dòng)態(tài)顯示執(zhí)行的情況。

# -*- coding: utf-8 -*-

from time import sleep
from rich.console import Console

console = Console()
count = 5
tasks = [f"task {n}" for n in range(1, count + 1)]


with console.status("") as status:
    num = 1
    while tasks:
        status.update("[{}/{}] 已經(jīng)完成".format(num, count))
        task = tasks.pop(0)
        sleep(1)
        num += 1

print("所有任務(wù) 全部完成！")

這樣就不用打印出很多行的日志信息，而是動(dòng)態(tài)的顯示task完成的情況。

還有一個(gè)動(dòng)態(tài)的應(yīng)用是進(jìn)度條，進(jìn)度條雖然不能提高程序的性能，但是它讓我們了解到程序大概運(yùn)行到哪了，能夠減少等待的焦慮。

比如，傳輸大量文件或者下載大文件的過(guò)程中，沒(méi)有進(jìn)度條的話，常常會(huì)擔(dān)心程序是不是卡住了。

下面是一個(gè)模擬文件傳輸中使用進(jìn)度條的示例：

# -*- coding: utf-8 -*-

import time

from rich.progress import Progress

# 模擬2個(gè)文件，一個(gè)是文件名，一個(gè)是文件大小
files = [("windows.iso", 24000), ("debian.iso", 17000)]

with Progress() as progress:
    tasks = []
    for f in files:
        task_id = progress.add_task("copy {}".format(f[0]), filename=f[0], start=False)
        progress.update(task_id, total=f[1])

        progress.start_task(task_id)
        # 模擬讀取文件，每次讀取1024字節(jié)
        total = f[1]
        buffer = 1024
        while total > 0:
            read_bytes = 0
            if total > buffer:
                read_bytes = buffer
                total -= buffer
            else:
                read_bytes = total
                total = 0

            progress.update(task_id, advance=read_bytes)
            time.sleep(0.2)

        progress.console.log("{} 傳輸完成!".format(f[0]))
        progress.stop_task(task_id)

2.4. 復(fù)雜結(jié)構(gòu)顯示

Rich庫(kù)還可以幫助我們?cè)诳刂婆_(tái)顯示一些結(jié)構(gòu)化的內(nèi)容。

2.4.1. 表格

表格是最常用的一種表現(xiàn)形式，也是最常用的一種展示數(shù)據(jù)的方式。

# -*- coding: utf-8 -*-

from rich.console import Console
from rich.table import Table

table = Table(title="國(guó)內(nèi)生產(chǎn)總值指數(shù)")

table.add_column("年份", justify="left", style="cyan", no_wrap=True)
table.add_column("指標(biāo)", style="magenta")
table.add_column("數(shù)值", justify="right", style="green")

table.add_row("2022年", "國(guó)民總收入指數(shù)", "4432.1")
table.add_row("2021年", "國(guó)民總收入指數(shù)", "4319.7")
table.add_row("2020年", "國(guó)民總收入指數(shù)", "3979.1")
table.add_row("2019年", "國(guó)民總收入指數(shù)", "3912.1")

console = Console()
console.print(table)

2.4.2. 樹(shù)形

樹(shù)形結(jié)構(gòu)也是常用的結(jié)構(gòu)，比如展示文件夾結(jié)構(gòu)：

# -*- coding: utf-8 -*-

from rich.tree import Tree
from rich import print

# 根節(jié)點(diǎn)
tree = Tree("/")

# usr 分支
usr = tree.add("usr")
fonts = usr.add("fonts")
bin = usr.add("bin")
lib = usr.add("lib")

# home 分支
home = tree.add("home")
git = home.add("git")
python = home.add("python")
golang = home.add("golang")

# projects 分支
projects = home.add("projects")
samples = projects.add("samples")
work = projects.add("work")

print(tree)

2.5. 文檔和代碼顯示

在IT行業(yè)內(nèi)，markdown文檔和代碼幾乎是繞不開(kāi)的2個(gè)東西。

然而，直接在命令行中原樣顯示的markdown和代碼的話，是很難閱讀的。

利用Rich庫(kù)，可以幫助我們解析markdown的標(biāo)記，高亮不同編程語(yǔ)言的代碼，從而在命令行中獲得良好的閱讀體驗(yàn)。

# -*- coding: utf-8 -*-

from rich.console import Console
from rich.markdown import Markdown


md_sample = """這是一份[Markdown][1]的語(yǔ)法介紹。

# 段落

一個(gè)段落是由一個(gè)或多個(gè)連續(xù)的行構(gòu)成，段落間靠一個(gè)或以上視覺(jué)上的空行劃分。

    這是一個(gè)段落。它有兩個(gè)句子。

    這是另一個(gè)段落。它也有
    兩個(gè)句子。


## 代碼

`print("hello")`

### 列表

* 無(wú)序(沒(méi)有編號(hào)的)列表中的一項(xiàng)
    * 一個(gè)子項(xiàng)，要以一個(gè)制表符或者4個(gè)空格縮進(jìn)
* 無(wú)序列表中的另一個(gè)項(xiàng)
1. 有序(排好序，有編號(hào)的)列表中的一項(xiàng)
1. 有序列表中的另一個(gè)項(xiàng)


[1]: http://daringfireball.net/projects/markdown/
"""

console = Console()
markdown = Markdown(md_sample)
console.print(markdown)

可以看出，不同的markdown標(biāo)記會(huì)解析成不同的顯示效果。

同樣，對(duì)于不同的代碼，也可以高亮顯示：

# -*- coding: utf-8 -*-

from rich.console import Console
from rich.syntax import Syntax

py_code = """
def hello(name):
    print("hello ", name)

hello("world")
"""

java_code = """
public class HelloWorld
{
    public static void main(String[] args)
    {
        System.out.println("Hello,World!");
    }
}
"""

c_code = """
#include <stdio.h>
int main()
{
    printf("Hello,World!");
    return 1;
}
"""

console = Console()

console.print("========================")
console.print("[green]python 代碼")
console.print("========================")
syntax = Syntax(py_code, "python", theme="monokai", line_numbers=True)
console.print(syntax)

console.print("========================")
console.print("[blue]java 代碼")
console.print("========================")
syntax = Syntax(java_code, "java", theme="monokai", line_numbers=True)
console.print(syntax)

console.print("========================")
console.print("[red]c 代碼")
console.print("========================")
syntax = Syntax(c_code, "c", theme="monokai", line_numbers=True)
console.print(syntax)

這里只演示了3種代碼，但是Rich可以支持幾乎所有主流的代碼。

3. 總結(jié)

總的來(lái)說(shuō)，Python Rich庫(kù)是一個(gè)強(qiáng)大的工具，可以讓Python開(kāi)發(fā)者在命令行環(huán)境中擁有更加豐富和強(qiáng)大的輸出能力，使得數(shù)據(jù)呈現(xiàn)更加直觀，增強(qiáng)了代碼的可讀性和調(diào)試效率。

本篇演示了一些常用的功能，關(guān)于Rich庫(kù)的其他功能和每個(gè)功能的細(xì)節(jié)可以參考其官方文檔：https://rich.readthedocs.io/en/latest/

以上就是Python使用Rich實(shí)現(xiàn)美化終端顯示效果的詳細(xì)內(nèi)容，更多關(guān)于Python Rich美化終端顯示的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: