stopwatch.py

import datetime
import time

class stopwatch(object):
  def __init__(self, jobname):
    self.jobname = jobname

  def __enter__(self):
    self.starttime = (datetime.datetime.utcnow() - datetime.datetime(1970, 1, 1)).total_seconds()

  def __exit__(self, type, value, traceback):
    self.endtime = (datetime.datetime.utcnow() - datetime.datetime(1970, 1, 1)).total_seconds()
    print(f'{self.jobname} took {round(self.endtime - self.starttime, 2)}s')

with stopwatch('wow intensive computations'):
  time.sleep(1)