A simple progress bar that just shows the total number of "things" processed and the time elapsed

progress_bar.py

"""
A simple progress bar that just shows the total number of "things" processed
and the time elapsed
"""
import time
from datetime import timedelta

from rich.progress import Progress, ProgressColumn, Task, TaskID, Text, TextColumn


class TimeElapsedColumn(ProgressColumn):
    """Renders time elapsed."""

    # Only refresh twice a second to prevent jitter
    max_refresh = 0.5

    def render(self, task: "Task") -> Text:
        """Show time elapsed."""
        elapsed = task.elapsed
        if elapsed is None:
            return Text("-:--:--", style="progress.remaining")
        elapsed_delta = timedelta(seconds=int(elapsed))
        return Text(str(elapsed_delta), style="progress.remaining")


progress = Progress(
    TextColumn("[bold black]{task.description}", justify="right",),
    "•",
    TextColumn(
        "[bold red]{task.fields[completed_description]} {task.completed}",
        justify="right",
    ),
    "•",
    TimeElapsedColumn(),
)


def run(task_id: TaskID) -> None:
    progress.update(task_id, total=100)
    progress.start_task(task_id)
    for x in range(0, 50):
        time.sleep(0.1)
        progress.update(task_id, advance=1)


def download():
    with progress:
        task_id = progress.add_task(
            "Downloading rows", completed_description="Records processed", start=False
        )
        run(task_id)
    print("now doing other stuff")
    time.sleep(3)
    print("Done")


if __name__ == "__main__":
    download()