From c790e856d93b38d4fcf6031fca615b480ed73e24 Mon Sep 17 00:00:00 2001 From: 0x5c <0x5c.dev@gmail.com> Date: Fri, 18 Oct 2019 11:26:06 -0400 Subject: [PATCH] should be better --- cogs/qrzcog.py | 24 ++++++++++-------------- 1 file changed, 10 insertions(+), 14 deletions(-) diff --git a/cogs/qrzcog.py b/cogs/qrzcog.py index 8e06723..d6500f7 100644 --- a/cogs/qrzcog.py +++ b/cogs/qrzcog.py @@ -66,24 +66,20 @@ class QRZCog(commands.Cog): await ctx.send(embed=embed) async def get_session(self): - """QRZ API Session handling.""" + """Session creation and caching.""" + self.key = await qrz_login(self.gs.keys.qrz_user, self.gs.keys.qrz_pass) + with open('data/qrz_session', 'w') as qrz_file: + qrz_file.write(self.key) + + @tasks.loop(count=) + async def _qrz_session_init(self): + """Helper task to allow obtaining a session at cog instantiation.""" try: with open('data/qrz_session') as qrz_file: self.key = qrz_file.readline().strip() await qrz_test_session(self.key) - except FileNotFoundError: - self.key = await qrz_login(self.gs.keys.qrz_user, self.gs.keys.qrz_pass) - with open('data/qrz_session', 'w') as qrz_file: - qrz_file.write(self.key) - except ConnectionError: - self.key = await qrz_login(self.gs.keys.qrz_user, self.gs.keys.qrz_pass) - with open('data/qrz_session', 'w') as qrz_file: - qrz_file.write(self.key) - - @tasks.loop(count=1) - async def _qrz_session_init(self): - """Helper task to allow initialisation of the session at cog instantiation.""" - await self.qrz_get_session() + except (FileNotFoundError, ConnectionError): + await self.qrz_get_session() async def qrz_login(user: str, passwd: str):