Почему возникает ошибка «'NoneType' object has no attribute 'cancelled'» в asyncpg?

Question

[Zo0meeer]

У меня есть метод класса работы с БД:

async def foo(self) -> None:
    """Копирует таблицу в файл."""
    await self.pool.copy_from_table(table_name='table', output=Path('files', 'file.csv'),
                                    schema_name='public', format='csv', delimiter=';',
                                    header=True, encoding='utf-8')

При тестировании на локальной машине файл стабильно создавался со всеми записями. В продакшене создаётся файл лишь с 50 записями, а в логах вижу ошибку:

File "/home/bot/venv/lib/python3.10/site-packages/asyncpg/pool.py", line 661, in copy_from_table
    return await con.copy_from_table(
  File "/home/bot/venv/lib/python3.10/site-packages/asyncpg/connection.py", line 760, in copy_from_table
    return await self._copy_out(copy_stmt, output, timeout)
  File "/home/bot/venv/lib/python3.10/site-packages/asyncpg/connection.py", line 1050, in _copy_out
    return await self._protocol.copy_out(copy_stmt, writer, timeout)
  File "asyncpg/protocol/protocol.pyx", line 363, in copy_out
  File "asyncpg/protocol/protocol.pyx", line 364, in asyncpg.protocol.protocol.BaseProtocol.copy_out
  File "asyncpg/protocol/coreproto.pyx", line 92, in asyncpg.protocol.protocol.CoreProtocol._read_server_messages
  File "asyncpg/protocol/coreproto.pyx", line 361, in asyncpg.protocol.protocol.CoreProtocol._process__copy_out_data
  File "asyncpg/protocol/coreproto.pyx", line 430, in asyncpg.protocol.protocol.CoreProtocol._parse_copy_data_msgs
  File "asyncpg/protocol/protocol.pyx", line 890, in asyncpg.protocol.protocol.BaseProtocol._on_result
  File "asyncpg/protocol/protocol.pyx", line 812, in asyncpg.protocol.protocol.BaseProtocol._dispatch_result
AttributeError: 'NoneType' object has no attribute 'cancelled'

Чувствую, что баг. С моей стороны есть способ исправить?

Comments

[Алан Гибизов]

А что за "продакшен"? Возможно, там есть ряд ограничений, не связанных напрямую с программой. Просто прога натыкается на это ограничение и говорит "ой, всё".

[Zo0meeer]

Алан Гибизов, обычный vds сервер. Да и по Traceback видно, что в самой библиотеке собака зарыта.
UPD: Вот ссылка на эту проблему на git.

[mkone112]

Zo0meeer, и там вроде есть решение

[Zo0meeer]

mkone112, какое? Изменение файла протокола на 700+ строчке? Пробовал, не помогло :(

Answer 1

[Сергей Горностаев]

Вероятно, это баг в asyncpg. Попробуйте использовать uvloop.

Comments

[Zo0meeer]

asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())

Поставил uvloop, проблема решилась, спасибо! Решение было на git, но когда читал, не мог поверить, что проблема описана ещё в 2018, а решается установкой сторонней библиотеки. Хотя, безусловно, использование uvloop лишь улучшает и ускоряет проект.
Извиняюсь за вопрос вне темы, но не можете подсказать как скормить uvloop для aiogram?

[Сергей Горностаев]

Zo0meeer, aiogram автоматически подключает uvloop, если он установлен.

[Zo0meeer]

Сергей Горностаев, благодарю!