Внутреннее исключение не вызывается с помощью asyncio.gather - PullRequest
Новая
       64

Внутреннее исключение не вызывается с помощью asyncio.gather

0 голосов
/ 15 апреля 2019

Используя Python 3.7, я пытаюсь поймать исключение и повторно вызвать его на , следуя примеру, найденному в StackOverflow .Хотя пример работает, он не работает для всех ситуаций.Ниже у меня есть два асинхронных скрипта Python, которые пытаются повторно вызвать исключения.Первый пример работает, он напечатает как внутреннее, так и внешнее исключение. 

import asyncio

class Foo:
    async def throw_exception(self):
        raise Exception("This is the inner exception")

    async def do_the_thing(self):
        try:
            await self.throw_exception()
        except Exception as e:
            raise Exception("This is the outer exception") from e

async def run():
    await Foo().do_the_thing()

def main():
    loop = asyncio.get_event_loop()
    loop.run_until_complete(run())

if __name__ == "__main__":
    main()

При выполнении этого правильно будет выведена следующая трассировка стека исключений: 

$ py test.py
Traceback (most recent call last):
  File "test.py", line 9, in do_the_thing
    await self.throw_exception()
  File "test.py", line 5, in throw_exception
    raise Exception("This is the inner exception")
Exception: This is the inner exception

The above exception was the direct cause of the following exception:
Traceback (most recent call last):
  File "test.py", line 21, in <module>
    main()
  File "test.py", line 18, in main
    loop.run_until_complete(run())
  File "C:\Python37\lib\asyncio\base_events.py", line 584, in run_until_complete
    return future.result()
  File "test.py", line 14, in run
    await Foo().do_the_thing()
  File "test.py", line 11, in do_the_thing
    raise Exception("This is the outer exception") from e
Exception: This is the outer exception

Однако в моем следующем PythonСценарий, у меня есть несколько задач, которые я ставлю в очередь и хочу получить аналогичную трассировку стека исключений.По сути, я, кроме вышеприведенной трассировки стека, должен быть напечатан 3 раза (по одному разу для каждой задачи в следующем скрипте).Единственное различие между приведенными выше и ниже сценариями заключается в функции run(). 

import asyncio

class Foo:
    async def throw_exception(self):
        raise Exception("This is the inner exception")

    async def do_the_thing(self):
        try:
            await self.throw_exception()
        except Exception as e:
            raise Exception("This is the outer exception") from e

async def run():
    tasks = []

    foo = Foo()

    tasks.append(asyncio.create_task(foo.do_the_thing()))
    tasks.append(asyncio.create_task(foo.do_the_thing()))
    tasks.append(asyncio.create_task(foo.do_the_thing()))

    results = await asyncio.gather(*tasks, return_exceptions=True)

    for result in results:
        if isinstance(result, Exception):
            print(f"Unexpected exception: {result}")

def main():
    loop = asyncio.get_event_loop()
    loop.run_until_complete(run())

if __name__ == "__main__":
    main()

Приведенный выше фрагмент кода приводит к разочаровывающе коротким исключениям, в которых отсутствуют следы стека. 

$ py test.py
Unexpected exception: This is the outer exception
Unexpected exception: This is the outer exception
Unexpected exception: This is the outer exception

Если я изменяюreturn_exceptions будет False, я выведу исключения и трассировку стека один раз, затем выполнение остановится, а оставшиеся две задачи будут отменены.Вывод идентичен выводу из первого скрипта.Недостатком этого подхода является то, что я хочу продолжить обработку задач, даже если встречаются исключения, а затем отобразить все исключения в конце, когда все задачи завершены.

1 Ответ

1 голос
/ 16 апреля 2019

asyncio.gather остановится на первом исключении, если вы не предоставите аргумент return_exceptions=True, поэтому ваш подход правильный: сначала нужно собрать все результаты и исключения, а затем отобразить их.

Чтобы получить полную отсутствующую трассировку стека, вам нужно будет сделать больше, чем просто «напечатать» исключение.Взгляните на модуль traceback в stdlib, в котором есть все, что вам нужно для этого: https://docs.python.org/3/library/traceback.html

Вы также можете использовать logging.exception, который будет делать более или менее то же самое.

...