aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/prompt-toolkit/py2/prompt_toolkit/eventloop/utils.py
blob: 56b4646d1cb842a347f2ec2dc857d58a2a12a6f6 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from __future__ import unicode_literals 
import time 
 
__all__ = ( 
    'TimeIt', 
) 
 
 
class TimeIt(object): 
    """ 
    Context manager that times the duration of the code body. 
    The `duration` attribute will contain the execution time in seconds. 
    """ 
    def __init__(self): 
        self.duration = None 
 
    def __enter__(self): 
        self.start = time.time() 
        return self 
 
    def __exit__(self, *args): 
        self.end = time.time() 
        self.duration = self.end - self.start