MeasureDuration.py

import time
import datetime


class MeasureDuration:
    def __init__(self, title: str = "Total time taken %s"):
        self.start = None
        self.end = None
        self.title = title

    def __enter__(self):
        self.start = time.time()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.end = time.time()
        print(self.title % self.duration())

    def duration(self):
        return str(datetime.timedelta(seconds=(self.end - self.start)))