Measure a function execution time

measure_time.py

import logging


logging.basicConfig(level=logging.INFO)


def measure_time(func):
    def wrapper(*args, **kwargs):
        f = datetime.datetime.fromtimestamp

        begin = time.time()
        logging.info(f'-- {func} started at {f(begin)} {"-" * 10}')

        result = func(*args, **kwargs)

        end = time.time()
        elapsed = end - begin
        logging.info(f'{func} done at {f(end)}. Elapsed: {elapsed}')

        return result
    return wrapper


if __name__ == '__main_':
    import os
    @measure_time
    print(list(os.listdir()))