如何用GitPython实现Git仓库克隆的全局进度条？

这个问题我太熟了！Git克隆的进度条之所以会反复重置到0%，核心原因是克隆过程本身会拆分成多个独立的阶段——比如远程对象下载、本地文件检出、子模块处理（如果有的话）——每个阶段都会触发RemoteProgress.update方法，而你之前的代码每次都直接覆盖进度条的total和n，相当于每个阶段都重新初始化了进度条，自然会出现反复重置的情况。

下面给你两种解决方案，再推荐几个更顺手的进度条库：

改进tqdm的实现方案 #

我们可以通过判断操作阶段（GitPython的op_code参数会标识当前是哪个阶段），来分别处理不同阶段的进度更新，避免全局重置：

# clone_progress.py
from git import RemoteProgress
from tqdm import tqdm

class CloneProgress(RemoteProgress):
    def __init__(self):
        super().__init__()
        # 初始化进度条，默认显示克隆状态
        self.pbar = tqdm(desc="Cloning repo", unit="B", unit_scale=True)
        self.current_stage = None

    def update(self, op_code, cur_count, max_count=None, message=''):
        # 区分不同的操作阶段
        if op_code & RemoteProgress.REMOTE_COMPRESS:
            # 远程对象下载阶段：按字节数统计进度
            if self.current_stage != "download":
                self.current_stage = "download"
                self.pbar.set_description("Downloading objects")
                self.pbar.unit = "B"
                self.pbar.unit_scale = True
            if max_count:
                self.pbar.total = max_count
            self.pbar.n = cur_count
            self.pbar.refresh()
        elif op_code & RemoteProgress.CHECKOUT:
            # 文件检出阶段：按文件数量统计进度
            if self.current_stage != "checkout":
                self.current_stage = "checkout"
                self.pbar.set_description("Checking out files")
                self.pbar.unit = "file"
                self.pbar.unit_scale = False
            if max_count:
                self.pbar.total = max_count
            self.pbar.n = cur_count
            self.pbar.refresh()

    def __del__(self):
        # 确保进度条正常关闭
        self.pbar.close()

这个改进后的类会自动识别克隆的不同阶段，更新进度条的描述和统计单位，而且每个阶段只会初始化一次进度条的总长度，不会反复重置。

推荐更优的进度条库 #

如果想让进度条更美观、更灵活，推荐使用rich库——它支持彩色输出、多任务进度跟踪，而且和GitPython的集成非常顺畅：

首先安装rich：

pip install rich

然后实现对应的进度类：

# clone_progress.py
from git import RemoteProgress
from rich.progress import Progress, TaskID

class RichCloneProgress(RemoteProgress):
    def __init__(self, progress: Progress):
        super().__init__()
        self.progress = progress
        self.current_task: TaskID = None

    def update(self, op_code, cur_count, max_count=None, message=''):
        if op_code & RemoteProgress.REMOTE_COMPRESS:
            # 下载阶段：如果没有任务则创建新任务
            if not self.current_task:
                self.current_task = self.progress.add_task(
                    "[blue]Downloading objects...", total=max_count
                )
            # 更新当前任务进度
            self.progress.update(self.current_task, completed=cur_count, total=max_count)
        elif op_code & RemoteProgress.CHECKOUT:
            # 检出阶段：先关闭上一个任务，创建新任务
            if self.current_task:
                self.progress.remove_task(self.current_task)
                self.current_task = None
            self.current_task = self.progress.add_task(
                "[green]Checking out files...", total=max_count
            )
            self.progress.update(self.current_task, completed=cur_count)

使用方式也很简单：

# main.py
import git
from rich.progress import Progress

with Progress() as progress:
    git.Repo.clone_from(
        url=repo_url, 
        to_path=dir_cloned_repo, 
        progress=RichCloneProgress(progress)
    )

Rich的进度条会自动处理动画和更新，不需要手动调用refresh()，而且彩色的阶段提示会让整个克隆过程更清晰。

内容的提问来源于stack exchange，提问作者Admin