You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.0 KiB
36 lines
1.0 KiB
2 years ago
|
import functools
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
class Clock(object):
|
||
|
|
"""
|
||
|
|
This is a decorator for the purpose of timing
|
||
|
|
functions and classes, it can be added to any function
|
||
|
|
and during runtime will spit out a formatted str
|
||
|
|
that displays function(arguments) and results with
|
||
|
|
a time delta. Uses pref_counter() not time()
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, func):
|
||
|
|
self.func = func
|
||
|
|
functools.update_wrapper(self, func)
|
||
|
|
|
||
|
|
def __call__(self, *args, **kwargs):
|
||
|
|
t0 = time.perf_counter()
|
||
|
|
result = self.func(*args, **kwargs)
|
||
|
|
elapsed = time.perf_counter() - t0
|
||
|
|
name = self.func.__name__
|
||
|
|
arg_lst = []
|
||
|
|
if args:
|
||
|
|
arg_lst.extend(repr(arg) for arg in args)
|
||
|
|
if kwargs:
|
||
|
|
arg_lst.extend("{}={}".format(k, w) for k, w in kwargs.items())
|
||
|
|
arg_str = ", ".join(arg_lst)
|
||
|
|
print(
|
||
|
|
"TIME TRIAL: {:s}({:.30s}~) -> {!r:.30}~ dt=[{:.8}]".format(
|
||
|
|
name, arg_str, result, elapsed
|
||
|
|
)
|
||
|
|
)
|
||
|
|
print()
|
||
|
|
return result
|