Source code for catatom2osm.progressbar

import logging

from tqdm import tqdm

from catatom2osm import config

log = logging.getLogger(config.app_name)


[docs]class FakeTqdm:
    """Fake class to disable progress bars."""

    def __init__(self, *args, **kwargs):
        pass

[docs]    def set_description(self, *args, **kwargs):
        pass

[docs]    def set_postfix(self, *args, **kwargs):
        pass

[docs]    def update(self, *args, **kwargs):
        pass

[docs]    def close(self):
        pass


[docs]def get(*args, **kwargs):
    if config.show_progress_bars:
        leave = getattr(log, "app_level", logging.DEBUG) <= logging.DEBUG
        kwargs.update({"leave": leave})
        return tqdm(*args, **kwargs)
    return FakeTqdm()