blob: 18b36f4f6c69c9486b872ecf6a8dd7ebd5a9f8ba (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
"""Utils related to logging."""
import logging
import time
from contextlib import contextmanager
from typing import Any, Iterator
_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))
|
