From 2b5e99b403b532e9b3a61797725329579c82557b Mon Sep 17 00:00:00 2001 From: Yukirin Date: Thu, 1 Feb 2018 06:12:59 +0800 Subject: [PATCH] Let's not be deprecated again Changed `aiohttp.get` to use `aiohttp.ClientSession()` --- imgwelcome/imgwelcome.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/imgwelcome/imgwelcome.py b/imgwelcome/imgwelcome.py index 52751cf..13f345e 100644 --- a/imgwelcome/imgwelcome.py +++ b/imgwelcome/imgwelcome.py @@ -56,6 +56,10 @@ class ImgWelcome: self.bot = bot self.settings = dataIO.load_json('data/imgwelcome/settings.json') self.version = "0.1.6" + self.session = aiohttp.ClientSession() + + def __unload(self): + self.session.close() async def save_settings(self): dataIO.save_json('data/imgwelcome/settings.json', self.settings) @@ -220,7 +224,7 @@ class ImgWelcome: await self.save_settings() async def _get_profile(self, url): - async with aiohttp.get(url) as r: + async with self.session.get(url) as r: image = await r.content.read() with open('data/imgwelcome/profilepic.png', 'wb') as f: f.write(image) @@ -407,7 +411,7 @@ class ImgWelcome: if success: try: - async with aiohttp.get(bg_url) as r: + async with self.session.get(bg_url) as r: image = await r.content.read() if not os.path.exists('data/imgwelcome/{}'.format(server.id)): os.makedirs('data/imgwelcome/{}'.format(server.id))