mirror of
https://github.com/miaowware/qrm2.git
synced 2024-11-29 19:28:42 -05:00
eb449f0f48
* add dxcc lookup * use new module `ctyparser` to simplify dxcc data management * fix logic in dxcc command * Fixes #21
91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
"""
|
|
Lookup extension for qrm
|
|
---
|
|
Copyright (C) 2019 Abigail Gold, 0x5c
|
|
|
|
This file is part of discord-qrm2 and is released under the terms of the GNU
|
|
General Public License, version 2.
|
|
"""
|
|
|
|
from datetime import datetime
|
|
import threading
|
|
|
|
import discord
|
|
from discord.ext import commands, tasks
|
|
from ctyparser import BigCty
|
|
|
|
import common as cmn
|
|
|
|
|
|
class LookupCog(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
try:
|
|
self.cty = BigCty('./data/cty.json')
|
|
except OSError:
|
|
self.cty = BigCty()
|
|
|
|
# TODO: See #107
|
|
# @commands.command(name="sat", category=cmn.cat.lookup)
|
|
# async def _sat_lookup(self, ctx: commands.Context, sat_name: str, grid1: str, grid2: str = None):
|
|
# '''Links to info about satellite passes on satmatch.com.'''
|
|
# now = datetime.utcnow().strftime('%Y-%m-%d%%20%H:%M')
|
|
# if grid2 is None or grid2 == '':
|
|
# await ctx.send(f'http://www.satmatch.com/satellite/{sat_name}/obs1/{grid1}'
|
|
# f'?search_start_time={now}&duration_hrs=24')
|
|
# else:
|
|
# await ctx.send(f'http://www.satmatch.com/satellite/{sat_name}/obs1/{grid1}'
|
|
# f'/obs2/{grid2}?search_start_time={now}&duration_hrs=24')
|
|
|
|
@commands.command(name="dxcc", aliases=['dx'], category=cmn.cat.lookup)
|
|
async def _dxcc_lookup(self, ctx: commands.Context, query: str):
|
|
'''Gets info about a DXCC prefix.'''
|
|
with ctx.typing():
|
|
query = query.upper()
|
|
full_query = query
|
|
embed = discord.Embed(title=f'DXCC Info for ',
|
|
timestamp=datetime.utcnow())
|
|
embed.set_footer(text=f'{ctx.author.name}',
|
|
icon_url=str(ctx.author.avatar_url))
|
|
embed.description = f'*Last Updated: {self.cty.formatted_version}*'
|
|
embed.colour = cmn.colours.bad
|
|
while query:
|
|
if query in self.cty.keys():
|
|
data = self.cty[query]
|
|
embed.add_field(name="Entity",
|
|
value=data['entity'])\
|
|
.add_field(name="CQ Zone",
|
|
value=data['cq'])\
|
|
.add_field(name="ITU Zone",
|
|
value=data['itu'])\
|
|
.add_field(name="Continent",
|
|
value=data['continent'])\
|
|
.add_field(name="Time Zone",
|
|
value=f'+{data["tz"]}' if data['tz'] > 0 else str(data['tz']))
|
|
embed.title += query
|
|
embed.colour = cmn.colours.good
|
|
break
|
|
else:
|
|
query = query[:-1]
|
|
else:
|
|
embed.title += full_query + ' not found'
|
|
embed.colour = cmn.colours.bad
|
|
await ctx.send(embed=embed)
|
|
|
|
@tasks.loop(hours=24)
|
|
async def _update_cty(self):
|
|
update = threading.Thread(target=run_update, args=(self.cty, "./data/cty.json"))
|
|
update.start()
|
|
|
|
|
|
def run_update(cty_obj, dump_loc):
|
|
update = cty_obj.update()
|
|
if update:
|
|
cty_obj.dump(dump_loc)
|
|
|
|
|
|
def setup(bot: commands.Bot):
|
|
lookupcog = LookupCog(bot)
|
|
bot.add_cog(lookupcog)
|
|
lookupcog._update_cty.start()
|