import logging from collections.abc import AsyncIterator from contextlib import asynccontextmanager from typing import Optional, List, Tuple, Any import aiohttp from pyhon import const from pyhon.connection.handler.base import ConnectionHandler from pyhon.typedefs import Callback _LOGGER = logging.getLogger(__name__) class HonAuthConnectionHandler(ConnectionHandler): _HEADERS = {"user-agent": const.USER_AGENT} def __init__(self, session: Optional[aiohttp.ClientSession] = None) -> None: super().__init__(session) self._called_urls: List[Tuple[int, str]] = [] @property def called_urls(self) -> List[Tuple[int, str]]: return self._called_urls @called_urls.setter def called_urls(self, called_urls: List[Tuple[int, str]]) -> None: self._called_urls = called_urls @asynccontextmanager async def _intercept( self, method: Callback, *args: Any, **kwargs: Any ) -> AsyncIterator[aiohttp.ClientResponse]: kwargs["headers"] = kwargs.pop("headers", {}) | self._HEADERS async with method(*args, **kwargs) as response: self._called_urls.append((response.status, str(response.request_info.url))) yield response