comparison mercurial/profiling.py @ 32786:0ead06d54ffe

profile: make the contextmanager object available to the callers

This will allow calling methods on the object in the code using the context
manager.

32785:37ec8f24d912

    def __enter__(self):
        self._entered = True
        if self._enabled:
            self.start()

    def start(self):
        """Start profiling.

        The profiling will stop at the context exit.

    The purpose of this context manager is to make calling code simpler:
    just use a single code path for calling into code you may want to profile
    and this function determines whether to start profiling.
    """
    with profile(ui, enabled=ui.configbool('profiling', 'enabled')):
        yield

32786:0ead06d54ffe

    def __enter__(self):
        self._entered = True
        if self._enabled:
            self.start()
        return self

    def start(self):
        """Start profiling.

        The profiling will stop at the context exit.

    The purpose of this context manager is to make calling code simpler:
    just use a single code path for calling into code you may want to profile
    and this function determines whether to start profiling.
    """
    with profile(ui, enabled=ui.configbool('profiling', 'enabled')) as p:
        yield p