qrm2/exts/ham.py

118 lines
4.5 KiB
Python
Raw Normal View History

2019-10-05 19:36:45 -04:00
"""
Ham extension for qrm
2019-10-05 19:36:45 -04:00
---
Copyright (C) 2019-2020 Abigail Gold, 0x5c
2019-10-05 19:36:45 -04:00
This file is part of qrm2 and is released under the terms of
the GNU General Public License, version 2.
2019-10-05 19:36:45 -04:00
"""
from datetime import datetime
2019-10-05 19:36:45 -04:00
import discord.ext.commands as commands
import common as cmn
from resources import callsign_info
from resources import phonetics
from resources import qcodes
2019-10-05 19:36:45 -04:00
class HamCog(commands.Cog):
def __init__(self, bot: commands.Bot):
2019-10-05 19:36:45 -04:00
self.bot = bot
@commands.command(name="qcode", aliases=["q"], category=cmn.cat.ref)
async def _qcode_lookup(self, ctx: commands.Context, qcode: str):
"""Looks up the meaning of a Q Code."""
qcode = qcode.upper()
embed = cmn.embed_factory(ctx)
if qcode in qcodes.qcodes:
embed.title = qcode
embed.description = qcodes.qcodes[qcode]
embed.colour = cmn.colours.good
else:
embed.title = f"Q Code {qcode} not found"
embed.colour = cmn.colours.bad
2019-10-05 19:36:45 -04:00
await ctx.send(embed=embed)
@commands.command(name="phonetics", aliases=["ph", "phoneticize", "phoneticise", "phone"], category=cmn.cat.ref)
async def _phonetics_lookup(self, ctx: commands.Context, *, msg: str):
"""Returns NATO phonetics for a word or phrase."""
result = ""
for char in msg.lower():
if char.isalpha():
result += phonetics.phonetics[char]
else:
result += char
result += " "
embed = cmn.embed_factory(ctx)
embed.title = f"Phonetics for {msg}"
embed.description = result.title()
embed.colour = cmn.colours.good
2019-10-05 19:36:45 -04:00
await ctx.send(embed=embed)
@commands.command(name="utc", aliases=["z"], category=cmn.cat.ref)
async def _utc_lookup(self, ctx: commands.Context):
"""Returns the current time in UTC."""
now = datetime.utcnow()
result = "**" + now.strftime("%Y-%m-%d %H:%M") + "Z**"
embed = cmn.embed_factory(ctx)
embed.title = "The current time is:"
embed.description = result
embed.colour = cmn.colours.good
2019-10-05 19:36:45 -04:00
await ctx.send(embed=embed)
@commands.command(name="prefixes", aliases=["vanity", "pfx", "vanities", "prefix"], category=cmn.cat.ref)
async def _vanity_prefixes(self, ctx: commands.Context, country: str = None):
"""Lists valid callsign prefixes for different countries."""
if country is None:
await ctx.send_help(ctx.command)
return
embed = cmn.embed_factory(ctx)
if country.lower() not in callsign_info.options:
embed.title = f"{country} not found!"
embed.description = f"Valid countries: {', '.join(callsign_info.options.keys())}"
embed.colour = cmn.colours.bad
else:
embed.title = callsign_info.options[country.lower()][0]
embed.description = callsign_info.options[country.lower()][1]
embed.colour = cmn.colours.good
for name, val in callsign_info.options[country.lower()][2].items():
embed.add_field(name=name, value=val, inline=False)
await ctx.send(embed=embed)
2019-12-15 12:51:43 -05:00
@commands.command(name="contests", aliases=["cc", "tests"], category=cmn.cat.ref)
async def _contests(self, ctx: commands.Context):
embed = cmn.embed_factory(ctx)
embed.title = "Contest Calendar"
2019-12-15 12:51:43 -05:00
embed.description = ("*We are currently rewriting the old, Chrome-based `contests` command. In the meantime, "
"use [the website](https://www.contestcalendar.com/weeklycont.php).*")
embed.colour = cmn.colours.good
2019-12-15 12:51:43 -05:00
await ctx.send(embed=embed)
@commands.command(name="phoneticweight", aliases=["pw"], category=cmn.cat.ref)
async def _weight(self, ctx: commands.Context, *, msg: str):
"""Calculates the phonetic weight of a callsign or message."""
embed = cmn.embed_factory(ctx)
msg = msg.upper()
weight = 0
for char in msg:
try:
weight += phonetics.pweights[char]
except KeyError:
embed.title = "Error in calculation of phonetic weight"
embed.description = f"Unknown character `{char}` in message"
embed.colour = cmn.colours.bad
await ctx.send(embed=embed)
return
embed.title = f"Phonetic Weight of {msg}"
embed.description = f"The phonetic weight is **{weight}**"
embed.colour = cmn.colours.good
await ctx.send(embed=embed)
2019-10-05 19:36:45 -04:00
def setup(bot: commands.Bot):
2019-10-05 19:36:45 -04:00
bot.add_cog(HamCog(bot))