Nestable Halo spinners

demo.py

from halo_nestable import HaloNestable
import time

with HaloNestable("Top level") as spinner:
    time.sleep(2)
    with HaloNestable("Nested"):
        time.sleep(2)
    time.sleep(2)

spinner.succeed("Great success!")

halo_nestable.py

from halo import Halo

_stack = []

class HaloNestable:
    def __init__(self, *args, **kwargs):
        self.halo = Halo(*args, **kwargs)

    def __getattr__(self, attr):
        return getattr(self.halo, attr)

    def __enter__(self):
        if _stack:
            _stack[-1].stop()
        _stack.append(self.halo)
        return self.halo.start()

    def __exit__(self, type, value, traceback):
        self.halo.stop()
        _stack.pop()
        if _stack:
            _stack[-1].start()