updated from Atlas

09-closure-deco/clock/clockdeco.py

@@ -0,0 +1,17 @@
+import time
+import functools
+
+
+def clock(func):
+    @functools.wraps(func)
+    def clocked(*args, **kwargs):
+        t0 = time.perf_counter()
+        result = func(*args, **kwargs)
+        elapsed = time.perf_counter() - t0
+        name = func.__name__
+        arg_lst = [repr(arg) for arg in args]
+        arg_lst.extend(f'{k}={v!r}' for k, v in kwargs.items())
+        arg_str = ', '.join(arg_lst)
+        print(f'[{elapsed:0.8f}s] {name}({arg_str}) -> {result!r}')
+        return result
+    return clocked

09-closure-deco/clock/clockdeco0.py

@@ -0,0 +1,13 @@
+import time
+
+
+def clock(func):
+    def clocked(*args):  # <1>
+        t0 = time.perf_counter()
+        result = func(*args)  # <2>
+        elapsed = time.perf_counter() - t0
+        name = func.__name__
+        arg_str = ', '.join(repr(arg) for arg in args)
+        print(f'[{elapsed:0.8f}s] {name}({arg_str}) -> {result!r}')
+        return result
+    return clocked  # <3>

09-closure-deco/clock/clockdeco_cls.py

@@ -0,0 +1,43 @@
+# clockdeco_class.py
+
+"""
+>>> snooze(.1)  # doctest: +ELLIPSIS
+[0.101...s] snooze(0.1) -> None
+>>> clock('{name}: {elapsed}')(time.sleep)(.2)  # doctest: +ELLIPSIS
+sleep: 0.20...
+>>> clock('{name}({args}) dt={elapsed:0.3f}s')(time.sleep)(.2)
+sleep(0.2) dt=0.201s
+"""
+
+# tag::CLOCKDECO_CLS[]
+import time
+
+DEFAULT_FMT = '[{elapsed:0.8f}s] {name}({args}) -> {result}'
+
+class clock:  # <1>
+
+    def __init__(self, fmt=DEFAULT_FMT):  # <2>
+        self.fmt = fmt
+
+    def __call__(self, func):  # <3>
+        def clocked(*_args):
+            t0 = time.perf_counter()
+            _result = func(*_args)  # <4>
+            elapsed = time.perf_counter() - t0
+            name = func.__name__
+            args = ', '.join(repr(arg) for arg in _args)
+            result = repr(_result)
+            print(self.fmt.format(**locals()))
+            return _result
+        return clocked
+# end::CLOCKDECO_CLS[]
+
+if __name__ == '__main__':
+
+    @clock()
+    def snooze(seconds):
+        time.sleep(seconds)
+
+    for i in range(3):
+        snooze(.123)
+

09-closure-deco/clock/clockdeco_demo.py

@@ -0,0 +1,16 @@
+import time
+from clockdeco0 import clock
+
+@clock
+def snooze(seconds):
+    time.sleep(seconds)
+
+@clock
+def factorial(n):
+    return 1 if n < 2 else n*factorial(n-1)
+
+if __name__ == '__main__':
+    print('*' * 40, 'Calling snooze(.123)')
+    snooze(.123)
+    print('*' * 40, 'Calling factorial(6)')
+    print('6! =', factorial(6))

09-closure-deco/clock/clockdeco_param.py

@@ -0,0 +1,40 @@
+# clockdeco_param.py
+
+"""
+>>> snooze(.1)  # doctest: +ELLIPSIS
+[0.101...s] snooze(0.1) -> None
+>>> clock('{name}: {elapsed}')(time.sleep)(.2)  # doctest: +ELLIPSIS
+sleep: 0.20...
+>>> clock('{name}({args}) dt={elapsed:0.3f}s')(time.sleep)(.2)
+sleep(0.2) dt=0.201s
+"""
+
+# tag::CLOCKDECO_PARAM[]
+import time
+
+DEFAULT_FMT = '[{elapsed:0.8f}s] {name}({args}) -> {result}'
+
+def clock(fmt=DEFAULT_FMT):  # <1>
+    def decorate(func):      # <2>
+        def clocked(*_args): # <3>
+            t0 = time.perf_counter()
+            _result = func(*_args)  # <4>
+            elapsed = time.perf_counter() - t0
+            name = func.__name__
+            args = ', '.join(repr(arg) for arg in _args)  # <5>
+            result = repr(_result)  # <6>
+            print(fmt.format(**locals()))  # <7>
+            return _result  # <8>
+        return clocked  # <9>
+    return decorate  # <10>
+
+if __name__ == '__main__':
+
+    @clock()  # <11>
+    def snooze(seconds):
+        time.sleep(seconds)
+
+    for i in range(3):
+        snooze(.123)
+
+# end::CLOCKDECO_PARAM[]

09-closure-deco/clock/clockdeco_param_demo1.py

@@ -0,0 +1,9 @@
+import time
+from clockdeco_param import clock
+
+@clock('{name}: {elapsed}s')
+def snooze(seconds):
+    time.sleep(seconds)
+
+for i in range(3):
+    snooze(.123)

09-closure-deco/clock/clockdeco_param_demo2.py

@@ -0,0 +1,9 @@
+import time
+from clockdeco_param import clock
+
+@clock('{name}({args}) dt={elapsed:0.3f}s')
+def snooze(seconds):
+    time.sleep(seconds)
+
+for i in range(3):
+    snooze(.123)