From c1e5822f07e2fe4ca47633ed3dfda7d7bee64b6c Mon Sep 17 00:00:00 2001 From: Felix Yan Date: Thu, 29 Nov 2018 06:39:14 +0800 Subject: [PATCH] Handle graceful exit and timeout Timeout was refactored and the defaults work correctly here. --- nvchecker/source/aiohttp_httpclient.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/nvchecker/source/aiohttp_httpclient.py b/nvchecker/source/aiohttp_httpclient.py index a950e17..8b34271 100644 --- a/nvchecker/source/aiohttp_httpclient.py +++ b/nvchecker/source/aiohttp_httpclient.py @@ -2,6 +2,7 @@ # Copyright (c) 2013-2017 lilydjwg , et al. import atexit +import asyncio import aiohttp connector = aiohttp.TCPConnector(limit=20) @@ -24,5 +25,9 @@ class BetterClientSession(aiohttp.ClientSession): raise HTTPError(res.status, res.reason, res) return res -session = BetterClientSession(connector=connector, read_timeout=10, conn_timeout=5) -atexit.register(session.close) +session = BetterClientSession(connector=connector) + +@atexit.register +def cleanup(): + loop = asyncio.get_event_loop() + loop.run_until_complete(session.close())