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

master
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

View File

@ -165,13 +165,15 @@ class LogHandlerContext:
def main_decorator(subject, *args, **kwargs):
'''
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 wrapped(argv):
(context, argv) = main_log_context(argv, subject, *args, **kwargs)
with context:
return main(argv)
status = main(argv)
log.info('Program finished, returned %s.', status)
return status
return wrapped
return wrapper