voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Log the program's final return when using @main_decorator.

Browse source
This commit is contained in:
voussoir 2021-07-04 15:44:50 -07:00
parent f18f0fd2df
commit e9d450029d
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
voussoirkit/operatornotify.py
View file

@ -165,13 +165,15 @@ class LogHandlerContext:
def main_decorator(subject, *args, **kwargs): def main_decorator(subject, *args, **kwargs):
''' '''
Add this decorator to your application's main function to automatically Add this decorator to your application's main function to automatically
wrap it in a main_log_context. wrap it in a main_log_context and log the final return value.
''' '''
def wrapper(main): def wrapper(main):
def wrapped(argv): def wrapped(argv):
(context, argv) = main_log_context(argv, subject, *args, **kwargs) (context, argv) = main_log_context(argv, subject, *args, **kwargs)
with context: with context:
return main(argv) status = main(argv)
log.info('Program finished, returned %s.', status)
return status
return wrapped return wrapped
return wrapper return wrapper