Таймаут и отмена

Решение

import asyncio
import logging

logger = logging.getLogger(__name__)


async def with_timeout(coro, timeout, default=None):
    try:
        return await asyncio.wait_for(coro, timeout=timeout)
    except asyncio.TimeoutError:
        logger.warning("операция не уложилась в %sс", timeout)
        return default


# Как правильно писать саму операцию, чтобы её можно было отменить:
async def slow_call(session, url):
    try:
        async with session.get(url) as response:
            return await response.json()
    except asyncio.CancelledError:
        logger.info("запрос к %s отменён", url)
        raise                     # обязательно пробрасываем дальше
    finally:
        # сюда попадаем и при отмене: закрываем то, что открыли
        pass


# Python 3.11+: таймаут на блок целиком
async def with_timeout_block(session, urls, timeout):
    async with asyncio.timeout(timeout):
        return [await slow_call(session, url) for url in urls]

Почему так

Как устроена отмена в asyncio

• wait_for по истечении времени вызывает cancel у задачи: внутрь корутины на ближайшем await бросается CancelledError
• Это обычное исключение, поэтому отработают все finally и async with — ресурсы закроются
• Прерывание возможно только на точках await: блокирующий вызов посреди корутины отменить нельзя, он доработает до конца

Почему CancelledError нужно пробрасывать

• Если перехватить и не пробросить, задача формально продолжит выполнение, а вызывающий уже считает её отменённой
• В 3.8+ CancelledError наследуется от BaseException именно для того, чтобы его не ловил случайный except Exception
• Ловить его допустимо только ради логирования и очистки — и сразу raise

Почему таймаут не означает «операция не выполнилась»

• Запрос мог дойти до сервера и быть обработан — вы просто не дождались ответа
• Для неидемпотентных операций это означает возможное двойное выполнение при повторе
• Отсюда ключ идемпотентности в платёжных API: это ровно та причина, по которой он существует