Source code for progressbar.shortcuts

from __future__ import annotations

import os
import typing

from . import (
    bar,
    fast as fast_module,
)

if typing.TYPE_CHECKING:
    from . import widgets as widgets_module

T = typing.TypeVar('T')




def progressbar(
    iterator: typing.Iterator[T],
    min_value: bar.NumberT = 0,
    max_value: bar.ValueT = None,
    widgets: typing.Sequence[widgets_module.WidgetBase | str] | None = None,
    prefix: str | None = None,
    suffix: str | None = None,
    fast: bool | None = None,
    **kwargs: typing.Any,
) -> typing.Iterator[T]:
    # Auto-dispatch to the lean FastProgressBar for the simple, common case;
    # anything that needs the full widget machinery uses ProgressBar.
    use_fast = (
        widgets is None
        and fast is not False
        and not kwargs.get('variables')
        and not os.environ.get('PROGRESSBAR_DISABLE_FASTPATH')
    )
    cls = fast_module.FastProgressBar if use_fast else bar.ProgressBar
    progressbar_ = cls(
        min_value=min_value,
        max_value=max_value,
        widgets=widgets,
        prefix=prefix,
        suffix=suffix,
        **kwargs,
    )
    return iter(progressbar_(iterator))