Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Track:
- def print_result(self, name, start_time, end_time):
- time_delta = end_time - start_time
- result = '%r took: %2.4f sec' % (name, time_delta)
- print(result)
- def time(self, func):
- @wraps(func)
- def func_wrapper(*args, **kwargs):
- start_time = time.time()
- func_result = func(*args, **kwargs)
- self.print_result(func.__name__, start_time, time.time())
- return func_result
- return func_wrapper
- def partial(self, partial_name):
- _self = self
- class class_wrapper:
- def __enter__(self):
- self.start_time = time.time()
- return self
- def __exit__(self, type, value, traceback):
- _self.print_result(partial_name, self.start_time, time.time())
- return class_wrapper()
- track = Track()
Add Comment
Please, Sign In to add comment