small implementation for time scaling context : 時間計測するcontext

timeit_context.py

import time
import contextlib

class TimeIt(object):
    def __init__(self):
        self._start = 0
        self._end = 0
    def begin(self):
        self._start = time.time()
        return self._start
    def finish(self):
        self._end = time.time()
        return self._end
    @property
    def time(self):
        if self._end:
            end = self._end
        else:
            end = time.time()
        return end - self._start
    def __enter__(self):
        self.begin()
        return self
    def __exit__(self, *tb):
        self.finish()
    def __str__(self):
        return '{:.2f} secs'.format(self.time)
        

if __name__ == '__main__':
    with TimeIt() as ti:
        time.sleep(1)
        print(ti.time)
        time.sleep(2)
        print(ti.time)
        time.sleep(3)
        print(ti.time)
    print(ti)