miaowware/qrm2
1
0
Fork 0
mirror of https://github.com/miaowware/qrm2.git synced 2025-07-30 11:52:26 -04:00
Code Issues Actions Projects Releases Wiki Activity

should be better

Browse Source
This commit is contained in:
0x5c 2019-10-18 11:26:06 -04:00
parent daf461b6ac
commit c790e856d9
No known key found for this signature in database
GPG Key ID: CCE14303E194D25E
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
cogs/qrzcog.py
View File

@ -66,24 +66,20 @@ class QRZCog(commands.Cog):
await ctx.send(embed=embed) await ctx.send(embed=embed)
async def get_session(self): 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: try:
with open('data/qrz_session') as qrz_file: with open('data/qrz_session') as qrz_file:
self.key = qrz_file.readline().strip() self.key = qrz_file.readline().strip()
await qrz_test_session(self.key) await qrz_test_session(self.key)
except FileNotFoundError: except (FileNotFoundError, ConnectionError):
self.key = await qrz_login(self.gs.keys.qrz_user, self.gs.keys.qrz_pass) await self.qrz_get_session()
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()
async def qrz_login(user: str, passwd: str): async def qrz_login(user: str, passwd: str):