diff --git a/voussoirkit/operatornotify.py b/voussoirkit/operatornotify.py index 07efb70..194d428 100644 --- a/voussoirkit/operatornotify.py +++ b/voussoirkit/operatornotify.py @@ -178,6 +178,10 @@ def main_decorator(subject, *args, **kwargs): def wrapper(main): def wrapped(argv): (context, argv) = main_log_context(argv, subject, *args, **kwargs) + + if isinstance(context, contextlib.nullcontext): + return main(argv) + with context: status = main(argv) log.info('Program finished, returned %s.', status)