Wow, what a cool original idea

timer.py

import time

class timer:
    def __init__(self, name="timer", format="{:0.1f}"):
        self.name = name
        self.format = format
 
    def __enter__(self):
        self.start = time.perf_counter()
        return self
 
    def __exit__(self, *args):
        self.t = time.perf_counter() - self.start
        t = self.format.format(self.t)
        print(self.name, "took", t, "s")