File size: 544 Bytes
113dbd0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from contextlib import contextmanager
from datetime import datetime
from loguru import logger
@contextmanager
def log_time(prefix=""):
"""log the time taken in a with block
prefix: the prefix text to show
"""
start_time = datetime.now()
logger.info("{}: start: {}".format(prefix, start_time))
try:
yield
finally:
end_time = datetime.now()
logger.info("{}: end: {}".format(prefix, end_time))
logger.info("{}: elapsed: {}".format(prefix, end_time - start_time))
|