Getting our hands on decorators we enriched our Python toolkit! This one was also a lot of fun. See our solution here. We did a simple timeit one and a more complex mute_exception one. For the latter we needed this week's article: How to Write a Decorator with an Optional Argument?.
Some other things we learned:
Know the stdlib. Part of what makes these decorators useful is knowing about time, random, the @wraps decorator, partial, the awesome logging module, etc.
As we sensed using optional arguments made the mute_exception more versatile. You can run it in various ways:
@mute_exception # works: no args provided = takes defaults (no reraise, returns None) @mute_exception(reraise=True) # works: raises the ZeroDivisionError = crash @mute_exception(reraise=False, default_return=0) # works: does not reraise ZeroDivisionError and returns 0 in that case
We used f-strings! We are on Python 3.6 now so we just could no longer resist the temptation :)
Our Flake 8 Check Vim shortcut pays off: the code is more readable.
When you run our solution it will print (fake) timings and mute/log the ZeroDivisionError exception:
$ python decorator-pb.py div of args: (1, 4) took 0.755037784576416 div 1/4 = 0.25 div of args: (2, 5) took 0.763498067855835 div 2/5 = 0.4 div of args: (3, 0) took 0.09057903289794922 div 3/0 = 0
The program did not crash by the divide by 0, logging the exception:
$ tail decorators.log ... ... 00:58:36 root DEBUG div called for args (1, 4) 00:58:36 root DEBUG div called for args (2, 5) 00:58:37 root DEBUG div called for args (3, 0) 00:58:37 root ERROR div raised exception ZeroDivisionError for args: (3, 0)
But there is more ... we got a nice PR with two other cool decorators: