From eb449f0f48aa5221ff345f8c67290d7fa4f84a47 Mon Sep 17 00:00:00 2001
From: Abigail Gold <5366828+classabbyamp@users.noreply.github.com>
Date: Wed, 11 Dec 2019 02:33:11 -0500
Subject: [PATCH] remote lookup functions (#10)

* add dxcc lookup
* use new module `ctyparser` to simplify dxcc data management
* fix logic in dxcc command
* Fixes #21
---
 exts/lookup.py   | 90 ++++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |  1 +
 2 files changed, 91 insertions(+)
 create mode 100644 exts/lookup.py

diff --git a/exts/lookup.py b/exts/lookup.py
new file mode 100644
index 0000000..50d1c1d
--- /dev/null
+++ b/exts/lookup.py
@@ -0,0 +1,90 @@
+"""
+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()
diff --git a/requirements.txt b/requirements.txt
index 65d4abd..a31e785 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
 discord.py
+ctyparser
 beautifulsoup4
 lxml