testpython.git

@@ -5,11 +5,22 @@

 import os

 import sys

 import time

+import functools

 from argparse import ArgumentParser

 import counter.counter

 import sitesize.sitesize

+def timeit(f):

+    @functools.wraps(f)

+    def wrapper(*args, **kwargs):

+        s = time.time()

+        r = f(*args, **kwargs)

+        print(f'{f.__name__} took {time.time() - s:1.3f}s.')

+        return r

+    return wrapper

+

+

 def set_v_print(verbose: bool):

     """

     Defines the function v_print.

@@ -21,6 +32,7 @@ def set_v_print(verbose: bool):

     v_print = print if verbose else lambda *a, **k: None

+@timeit

 def main(debug: bool):

     start_time = time.time()

     localdir = os.path.abspath(os.path.dirname(sys.argv[0]))

@@ -28,7 +40,7 @@ def main(debug: bool):

     counter.counter.run()

     v_print("Running sitesize...")

     sitesize.sitesize.run()

-    print(f'Done in {localdir}.  That took {time.time() - start_time:1.2f}s.')

+    print(f'Done in {localdir}.')

 if __name__ == '__main__':