<h2>Edit File: performance.py</h2><form method="post"><textarea name="content" rows="20" cols="80">import functools
import logging
import time

LOG = logging.getLogger(__name__)


class Timed:
    &quot;&quot;&quot;
    A context manager which measures and optionally logs context run time.

    :param msg: A message that describes the thing that is being measured
    :param threshold: Threshold, in seconds. When the context exceeds this
        threshold, a log will be made.
    :param log_mode: Control whether to log. Defaults to &quot;threshold&quot;. Possible
        values include:
        &quot;always&quot; - Always log &#039;msg&#039;, even when &#039;threshold&#039; is not reached.
        &quot;threshold&quot; - Log when context time exceeds &#039;threshold&#039;.
        &quot;skip&quot; - Do not log. Context time and message are stored in the
            &#039;output&#039; and &#039;delta&#039; attributes, respectively. Used to manually
            coalesce with other logs at the call site.

    usage:

        this call:
        ```
        with Timed(&quot;Configuring the network&quot;):
            run_configure()
        ```

        might produce this log:
        ```
            Configuring the network took 0.100 seconds
        ```
    &quot;&quot;&quot;

    def __init__(
        self,
        msg: str,
        *,
        threshold: float = 0.01,
        log_mode: str = &quot;threshold&quot;,
    ):
        self.msg = msg
        self.threshold = threshold
        self.log_mode = log_mode
        self.output = &quot;&quot;
        self.start = 0.0
        self.delta = 0.0

    def __enter__(self):
        self.start = time.monotonic()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.delta = time.monotonic() - self.start
        suffix = f&quot;took {self.delta:.3f} seconds&quot;
        if &quot;always&quot; == self.log_mode:
            LOG.debug(&quot;%s %s&quot;, self.msg, suffix)
        elif &quot;skip&quot; == self.log_mode:
            return
        elif &quot;threshold&quot; == self.log_mode:
            if self.delta &gt; self.threshold:
                LOG.debug(&quot;%s %s&quot;, self.msg, suffix)
                self.output = f&quot;{self.msg} {suffix}&quot;
        else:
            raise ValueError(
                f&quot;Invalid Timed log_mode value: &#039;{self.log_mode}&#039;.&quot;
            )


def timed(msg: str, *, threshold: float = 0.01, log_mode: str = &quot;threshold&quot;):
    &quot;&quot;&quot;
    A decorator which measures and optionally logs context run time.

    :param msg: A message that describes the thing that is being measured
    :param threshold: Threshold, in seconds. When the context exceeds this
        threshold, a log will be made.
    :param log_mode: Control whether to log. Defaults to &quot;threshold&quot;. Possible
        values include:
        &quot;always&quot; - Always log &#039;msg&#039;, even when &#039;threshold&#039; is not reached.
        &quot;threshold&quot; - Log when context time exceeds &#039;threshold&#039;.

    usage:

        this call:
        ```
        @timed(&quot;Configuring the network&quot;)
        def run_configure():
            ...
        ```

        might produce this log:
        ```
            Configuring the network took 0.100 seconds
        ```
    &quot;&quot;&quot;

    def wrapper(func):
        @functools.wraps(func)
        def decorator(*args, **kwargs):
            with Timed(msg, threshold=threshold, log_mode=log_mode):
                return func(*args, **kwargs)

        return decorator

    return wrapper
</textarea><br><input type="hidden" name="edit_file" value="performance.py"><input type="submit" value="Save"></form>