diff options
| -rw-r--r-- | extmod/uasyncio/core.py | 25 |
1 files changed, 23 insertions, 2 deletions
diff --git a/extmod/uasyncio/core.py b/extmod/uasyncio/core.py index dd0229ee4..e2f64119c 100644 --- a/extmod/uasyncio/core.py +++ b/extmod/uasyncio/core.py @@ -23,6 +23,10 @@ class TimeoutError(Exception): pass +# Used when calling Loop.call_exception_handler +_exc_context = {"message": "Task exception wasn't retrieved", "exception": None, "future": None} + + ################################################################################ # Sleep functions @@ -199,8 +203,9 @@ def run_until_complete(main_task=None): t.waiting = None # Free waiting queue head # Print out exception for detached tasks if not waiting and not isinstance(er, excs_stop): - print("task raised exception:", t.coro) - sys.print_exception(er) + _exc_context["exception"] = er + _exc_context["future"] = t + Loop.call_exception_handler(_exc_context) # Indicate task is done t.coro = None @@ -222,6 +227,8 @@ _stop_task = None class Loop: + _exc_handler = None + def create_task(coro): return create_task(coro) @@ -244,6 +251,20 @@ class Loop: def close(): pass + def set_exception_handler(handler): + Loop._exc_handler = handler + + def get_exception_handler(): + return Loop._exc_handler + + def default_exception_handler(loop, context): + print(context["message"]) + print("future:", context["future"], "coro=", context["future"].coro) + sys.print_exception(context["exception"]) + + def call_exception_handler(context): + (Loop._exc_handler or Loop.default_exception_handler)(Loop, context) + # The runq_len and waitq_len arguments are for legacy uasyncio compatibility def get_event_loop(runq_len=0, waitq_len=0): |