Get execution time of any part of your code with this context manager for python

execution_timer_with_context_manager.py

import logging
from contextlib import contextmanager

log = logging.getLogger(__name__)

@contextmanager
def timed_log(name, time_chunk='seconds'):
    """Context manager to get execution time of parts of codes.
    To use, simply declares the context manager:
        ```with timed_log(name='useful', time_chunck='minutes'):
            -- your code to get execution time here
        ```
    
    Do not forget to set your logging lib to print INFO logs.
    Parameters
    ----------
    name : str
        a useful name for you to know which part of the code you are measuring
    time_chunck : str, optional
        can be `minutes` or `seconds`, by default 'seconds'
    """
    
    printlog = lambda name, t, time_chunk: log.info(' '.join(map(str, [name, 'took', t, time_chunk])))
    
    start_time = time.time()
    try:
        yield start_time
    finally:
        
        total_time = time.time() - start_time
        
        if time_chunk == 'seconds':
            printlog(name, round(total_time, 1), time_chunk)
        elif time_chunk == 'minutes':
            printlog(name, round(total_time / 60, 2), time_chunk)