blob: cb3bb1954e872e1169f9f87c4415828c60dd7f56 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
"""Utils related to logging."""
import logging
import time
from collections.abc import Iterator
from contextlib import contextmanager
from typing import Any
_logger = logging.getLogger(__name__)
@contextmanager
def timed_info(msg: Any, *args: Any) -> Iterator[None]:
"""Context manager for logging slow operations, mentions duration."""
start = time.time()
try:
yield
finally:
elapsed = time.time() - start
_logger.info(msg + " (%.2fs)", *(*args, elapsed)) # noqa: G003
|