BioArchLinux/nvchecker
2
0
Fork 0
mirror of https://github.com/lilydjwg/nvchecker.git synced 2025-03-10 06:14:02 +00:00
Code Issues Projects Releases Packages Wiki Activity

fix aiohttp_httpclient

Browse source
This commit is contained in:
lilydjwg 2020-08-20 19:29:00 +08:00
parent 58a7ae05ca
commit 57626faecf
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
nvchecker/httpclient/aiohttp_httpclient.py
View file

@ -5,12 +5,14 @@ import atexit
import asyncio
from typing import Optional, Dict
import structlog
import aiohttp
from .base import BaseSession, TemporaryError, Response
__all__ = ['session']
logger = structlog.get_logger(logger_name=__name__)
connector = aiohttp.TCPConnector(limit=20)
class AiohttpSession(BaseSession):
@ -31,12 +33,14 @@ class AiohttpSession(BaseSession):
) -> Response:
kwargs = {
'headers': headers,
'params': params,
}
if proxy is not None:
kwargs['proxy'] = proxy
try:
logger.debug('send request', method=method, url=url, kwargs=kwargs)
res = await self.session.request(
method, url, **kwargs)
except (
@ -55,6 +59,6 @@ class AiohttpSession(BaseSession):
@atexit.register
def cleanup():
loop = asyncio.get_event_loop()
loop.run_until_complete(session.close())
loop.run_until_complete(session.session.close())
session = AiohttpSession()