code timer for python, base on timeit

code_timer.py

import time
import contextlib
import gc

@contextlib.contextmanager
def disable_gc():
    if gc.isenabled():
        gc.disable()
        yield
        gc.enable()
    else:
        yield

class CodeTimer:
    def __init__(self):
        self._total = None
        self._begin = None

    @contextlib.contextmanager
    def begin(self):
        if self._begin is not None:
            raise RuntimeError
        pfc = time.perf_counter
        with disable_gc():
            self._begin = pfc()
            yield self
            self._total = pfc() - self._begin

    @property
    def time(self):
        if self._total is None:
            raise RuntimeError
        return self._total
    
    def used(self):
        if self._begin is None:
            raise RuntimeError
        return time.perf_counter() - self._begin