aiohttp pass proxy to ClientSession

aiohttp_client_proxy.py

from aiohttp.client import ClientSession


class CustomClientSession(ClientSession):
    """
        Use proxy on every request when passed to Client

        Usage
        >> async with CustomClientSession(proxy="http://1.1.1.1:3128") as session:
    """
    def __init__(self, *args, **kwargs):
        if 'proxy' in kwargs:
            self.proxy = kwargs.pop('proxy')
        super().__init__(
            *args, **kwargs
        )

    async def _request(self, *args, **kwargs):
        if hasattr(self, 'proxy'):
            kwargs.update({
                "proxy": self.proxy
            })
        return await super()._request(*args, **kwargs)