33 lines
922 B
Python
33 lines
922 B
Python
"""Contains the LookupCog, which deals with the lookup commands."""
|
|
from discord.ext import commands # Has the cog class
|
|
from discord_slash import cog_ext # Used for slash commands
|
|
|
|
from gwendolyn.utils import get_params # pylint: disable=import-error
|
|
|
|
params = get_params()
|
|
|
|
|
|
class LookupCog(commands.Cog):
|
|
"""Contains the lookup commands."""
|
|
|
|
def __init__(self, bot):
|
|
"""Initialize the cog."""
|
|
self.bot = bot
|
|
|
|
# Looks up a spell
|
|
@cog_ext.cog_slash(**params["spell"])
|
|
async def spell(self, ctx, query):
|
|
"""Look up a spell."""
|
|
await self.bot.lookup_funcs.spell_func(ctx, query)
|
|
|
|
# Looks up a monster
|
|
@cog_ext.cog_slash(**params["monster"])
|
|
async def monster(self, ctx, query):
|
|
"""Look up a monster."""
|
|
await self.bot.lookup_funcs.monster_func(ctx, query)
|
|
|
|
|
|
def setup(bot):
|
|
"""Add the cog to the bot."""
|
|
bot.add_cog(LookupCog(bot))
|