贝利信息

如何让 rich.progress 在多线程中安全更新进度条

日期:2026-01-23 00:00 / 作者:冰川箭仙
多线程下需用 threading.Lock 串行化 rich.progress.update() 调用,主线程创建 Progress 和锁,子线程持 task_id 并在锁内更新;或改用 Live 配合线程安全状态管理;multiprocessing 不支持 Progress 共享。

rich.progress 多线程下直接更新会崩溃

直接在多个线程里调用 progress.update() 会导致 RuntimeError: Cannot update progress bar from multiple threads。这是因为 Progress 内部的渲染器(ConsoleRender)不是线程安全的,且进度条状态(如任务计数、已用时间)共享但未加锁。

用 threading.Lock 手动同步更新操作

最轻量、可控性最强的做法:所有对 progress.update() 的调用都包裹在一个全局 threading.Lock 里。这不是“绕过” rich 的限制,而是显式地串行化 UI 更新逻辑。

from rich.progress import Progress
import threading
import time

progress = Progress() lock = threading.Lock() task = progress.add_task("[cyan]Processing...", total=100)

def worker(i): time.sleep(0.1) # 模拟工作 with lock: progress.update(task, advance=1)

threads = [threading.Thread(target=worker, args=(i,)) for i in range(100)] for t in threads: t.start() for t in threads: t.join()

progress.stop()

改用 rich.live.Live 配合手动状态管理

当需要

更灵活的多线程 UI(比如显示多个实时指标、非进度类状态),LiveProgress 更适合——它只负责刷新整个渲染结果,不管理任务生命周期,因此线程安全责任完全落在你身上。

别碰 multiprocessing —— rich.progress 本身不支持进程间共享

Progress 对象无法被序列化(pickle),也不能跨进程传递。即使你用 ManagerQueue 试图同步进度,在子进程中调用 progress.update() 会直接报 AttributeError 或静默失败。

真正麻烦的不是“怎么更新”,而是“谁来决定更新时机”。rich 的设计默认把进度控制权交给单一线程,强行多线程更新容易掩盖数据竞争问题——比如两个线程同时 advance=1,却只生效一次。锁只是表层解法,背后得理清你的工作流是否真的需要并发更新 UI。