Medios-Macina/SYS/progress.py

"""Rich-only progress helpers.

These functions preserve the legacy call signatures used around the codebase,
but all rendering is performed via Rich (no ASCII progress bars).
"""

from __future__ import annotations

import sys

from models import ProgressBar


_BAR = ProgressBar()


def print_progress(
    filename: str, current: int, total: int, speed: float = 0, end: str = "\r"
) -> None:
    _BAR.update(
        downloaded=int(current),
        total=int(total) if total else None,
        label=str(filename or "progress"),
        file=sys.stderr,
    )


def print_final_progress(filename: str, total: int, elapsed: float) -> None:
    _BAR.finish()