Python: Aioimaplib ловить исключения

Question

Я пытаюсь проверить несколько данных для входа в систему imap асинхронно с aioimaplib. Этот код работает до тех пор, пока серверы imap достижимы и / или клиенты не истекают.

Как правильно ловить исключения?

Пример исключения:

ERROR:asyncio:Task exception was never retrieved future: <Task finished coro=<BaseEventLoop.create_connection() done, defined at G:\WinPython-3.5.4\python-3.5.4.amd64\lib\asyncio\base_events.py:679> exception=TimeoutError(10060, "Connect call failed ('74.117.114.100', 993)")>

Код:

account_infos = [
    # User            Password     Server
    ('[e-mail]', 'password1', 'imap.google.com'),
    ('[e-mail]', 'password2', 'imap.yandex.com'),
    ('[e-mail]', 'password3', 'imap.server3.com'),
]


class MailLogin:
    def __init__(self):
        self.loop = asyncio.get_event_loop()
        self.queue = asyncio.Queue(loop=self.loop)
        self.max_workers = 2

    async def produce_work(self):
        for i in account_infos:
            await self.queue.put(i)
        for _ in range(max_workers):
            await self.queue.put((None, None, None))

    async def worker(self):
        while True:
            (username, password, server) = await self.queue.get()
            if username is None:
                break

            while True:
                try:
                    s = IMAP4_SSL(server)
                    await s.wait_hello_from_server()
                    r = await s.login(username, password)
                    await s.logout()
                    if r.result != 'NO':
                        print('Information works')
                except Exception as e:
                    # DOES NOT CATCH
                    print(str(e))
                else:
                    break

    def start(self):
        try:
            self.loop.run_until_complete(
                asyncio.gather(self.produce_work(), *[self.worker() for _ in range(self.max_workers)],
                               loop=self.loop, return_exceptions=True)
            )
        finally:
            print('Done')


if __name__ == '__main__':
    MailLogin().start()

Answer 1

Есть несколько способов сделать это, но TimeoutError вероятно пойман в вашем except, Вы не видите этого, потому что str (e) - пустая строка.

Вы можете видеть стеки, включающие режим отладки asyncio.

Во-первых, вы можете поймать исключение, как вы это сделали:

async def fail_fun():
    try:
        imap_client = aioimaplib.IMAP4_SSL(host='foo', timeout=1)
        await imap_client.wait_hello_from_server()
    except Exception as e:
        print('Exception : ' + str(e))

if __name__ == '__main__':
    get_event_loop().run_until_complete(fail_fun())

Во-вторых, вы можете поймать исключение на run_until_complete

async def fail_fun():
    imap_client = aioimaplib.IMAP4_SSL(host='foo', timeout=1)
    await imap_client.wait_hello_from_server()

if __name__ == '__main__':
    try:
        get_event_loop().run_until_complete(fail_fun())
    except Exception as e:
        print('Exception : ' + str(e))

Соединение установлено, оборачивая loop.create_connection сопрограмма с create_task: мы хотели установить соединение в конструкторе IMAP4 и __init__ должен вернуть None.

Поэтому, если у вашего хоста неверное значение, вы можете проверить его раньше или дождаться истечения времени ожидания:

socket.gaierror: [Errno -5] No address associated with hostname

если хост не отвечает до истечения времени ожидания, вы можете увеличить время ожидания. И если соединение потеряно во время соединения, вы можете добавить функцию обратного вызова потерянного соединения в конструкторе IMAP4.