Source code for trinity.utils.timer

"""Timer context manager"""

import time




class Timer:


    def __init__(self, metrics_dict, key_name):
        self.metrics = metrics_dict
        self.key = key_name


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

    def __exit__(self, exc_type, exc_val, exc_tb):
        end_time = time.time()
        elapsed_time = end_time - self.start_time
        self.metrics[self.key] = elapsed_time