util.py

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file util.py

import os
import contextlib
import sys

try:
    from halo import Halo
except:
    Halo = None


@contextlib.contextmanager
def Spinner(text, persist=True, *args, **kwargs):
    stream = kwargs.get("stream", sys.stdout)
    if stream.isatty() and Halo is not None:
        spinner = Halo(text, *args, **kwargs)
        spinner.start()
        try:
            yield
            if persist:
                spinner.succeed()
        except:
            if persist:
                spinner.fail()
            raise
        finally:
            if not persist:
                spinner.stop()
    else:
        stream.write(text + "\n")
        yield