mirror of https://github.com/miaowware/qrm2.git
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
"""
|
|
Prefixes Lookup extension for qrm
|
|
---
|
|
Copyright (C) 2021 Abigail Gold, 0x5c
|
|
|
|
This file is part of qrm2 and is released under the terms of
|
|
the GNU General Public License, version 2.
|
|
"""
|
|
|
|
|
|
import discord.ext.commands as commands
|
|
|
|
import common as cmn
|
|
from resources import callsign_info
|
|
|
|
|
|
class PrefixesCog(commands.Cog):
|
|
def __init__(self, bot: commands.Bot):
|
|
self.bot = bot
|
|
self.pfxs = callsign_info.options
|
|
|
|
@commands.command(name="prefixes", aliases=["vanity", "pfx", "vanities", "prefix"], category=cmn.Cats.REF)
|
|
async def _vanity_prefixes(self, ctx: commands.Context, country: str = ""):
|
|
"""Lists valid callsign prefixes for different countries."""
|
|
country = country.lower()
|
|
embed = cmn.embed_factory(ctx)
|
|
if country not in self.pfxs:
|
|
desc = "Possible arguments are:\n"
|
|
for key, val in self.pfxs.items():
|
|
desc += f"`{key}`: {val.title}{(' ' + val.emoji if val.emoji else '')}\n"
|
|
embed.title = f"{country} Not Found!"
|
|
embed.description = desc
|
|
embed.colour = cmn.colours.bad
|
|
await ctx.send(embed=embed)
|
|
return
|
|
else:
|
|
data = self.pfxs[country]
|
|
embed.title = data.title + (" " + data.emoji if data.emoji else "")
|
|
embed.description = data.desc
|
|
embed.colour = cmn.colours.good
|
|
|
|
for name, val in data.calls.items():
|
|
embed.add_field(name=name, value=val, inline=False)
|
|
await ctx.send(embed=embed)
|
|
|
|
|
|
def setup(bot: commands.Bot):
|
|
bot.add_cog(PrefixesCog(bot))
|