Progress bar for Python Trio tasks using tqdm

trio_progress_bar.py

import trio
import tqdm

class TrioProgress(trio.abc.Instrument):

    def __init__(self, total, notebook_mode=False, **kwargs):
        if notebook_mode:
            from tqdm import tqdm_notebook as tqdm
        else:
            from tqdm import tqdm

        self.tqdm = tqdm(total=total, **kwargs)

    def task_exited(self, task):
        self.tqdm.update(1)