mirror of https://github.com/miaowware/qrm2.git
88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
"""
|
|
Lookup extension for qrm
|
|
---
|
|
Copyright (C) 2019-2020 Abigail Gold, 0x5c
|
|
|
|
This file is part of qrm2 and is released under the terms of
|
|
the GNU General Public License, version 2.
|
|
"""
|
|
|
|
|
|
import threading
|
|
from pathlib import Path
|
|
|
|
from ctyparser import BigCty
|
|
|
|
from discord.ext import commands, tasks
|
|
|
|
import common as cmn
|
|
|
|
|
|
cty_path = Path("./data/cty.json")
|
|
|
|
|
|
class LookupCog(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
try:
|
|
self.cty = BigCty(cty_path)
|
|
except OSError:
|
|
self.cty = BigCty()
|
|
|
|
# TODO: See #107
|
|
# @commands.command(name="sat", category=cmn.Cats.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.Cats.LOOKUP)
|
|
async def _dxcc_lookup(self, ctx: commands.Context, query: str):
|
|
"""Gets DXCC info about a callsign prefix."""
|
|
query = query.upper()
|
|
full_query = query
|
|
embed = cmn.embed_factory(ctx)
|
|
embed.title = "DXCC Info for "
|
|
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"])
|
|
embed.add_field(name="CQ Zone", value=data["cq"])
|
|
embed.add_field(name="ITU Zone", value=data["itu"])
|
|
embed.add_field(name="Continent", value=data["continent"])
|
|
embed.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, cty_path))
|
|
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()
|