Geoffrey-Django/geoffrey/cogs/Search_Commands.py

148 lines
5.5 KiB
Python
Raw Normal View History

from discord.ext import commands
2018-08-12 19:00:04 +00:00
2018-08-21 20:36:51 +00:00
from geoffrey.BotErrors import *
from geoffrey.DiscordHelperFunctions import *
class Search_Commands:
"""
Commands to find stuff.
"""
def __init__(self, bot):
self.bot = bot
@commands.command(pass_context=True)
@commands.cooldown(5, 60, commands.BucketType.user)
async def find(self, ctx, *args):
"""
Finds all the locations matching the search term
?find [Search]
"""
search = get_name(args)
try:
if search is None:
raise commands.UserInputError
result = self.bot.bot_commands.find(search)
await ctx.send(
'{}, The following entries match **{}**:\n{}'.format(ctx.message.author.mention, search, result))
except LocationLookUpError:
await ctx.send(
'{}, no matches to **{}** were found in the database.'.format(ctx.message.author.mention, search))
@commands.command(pass_context=True)
@commands.cooldown(5, 60, commands.BucketType.user)
async def tunnel(self, ctx, player: str):
"""
Finds all the tunnels a player owns
?tunnel [Player]
"""
try:
result = self.bot.bot_commands.tunnel(player)
await ctx.send(
2018-09-15 15:24:39 +00:00
'{}, **{}** owns the following tunnel(s): \n{}'.format(ctx.message.author.mention, player, result))
except LocationLookUpError:
await ctx.send('{}, no tunnels for **{}** were found in the database.'
.format(ctx.message.author.mention, player))
@commands.command(pass_context=True)
@commands.cooldown(5, 60, commands.BucketType.user)
async def find_around(self, ctx, x_pos: int, z_pos: int, *args):
"""
Finds all the locations around a certain point.
The radius defaults to 200 blocks if no value is given
Default dimension is the overworld
2018-08-18 17:01:59 +00:00
?find_around [X Coordinate] [Z Coordinate] [Radius] [Optional Flags]
2018-08-18 17:01:59 +00:00
Optional Flags:
-d [dimension]
"""
radius = 200
dimension = 'Overworld'
try:
if len(args) > 0:
if args[0] == '-d':
dimension = args[1]
if len(args) > 1:
radius = int(args[2])
else:
radius = int(args[0])
if len(args) > 1:
if args[1] == '-d':
dimension = args[2]
base_string = self.bot.bot_commands.find_around(x_pos, z_pos, radius, dimension)
if len(base_string) != 0:
2018-09-16 00:54:03 +00:00
await ctx.send('{}, the following locations(s) are within **{}** blocks of that point: \n {}'.format(
ctx.message.author.mention, radius, base_string))
else:
await ctx.send('{}, there are no locations within {} blocks of that point'
.format(ctx.message.author.mention, radius))
except ValueError:
await ctx.send(
'{}, invalid radius, the radius must be a whole number.'.format(ctx.message.author.mention,
radius))
except InvalidDimError:
await ctx.send('{}, {} is an invalid dimension.'.format(ctx.message.author.mention, dimension))
@commands.command(pass_context=True)
@commands.cooldown(5, 60, commands.BucketType.user)
async def selling(self, ctx, *args):
"""
Lists all the shops selling an item
2018-08-18 17:01:59 +00:00
?selling [item]
"""
item_name = get_name(args)
try:
result = self.bot.bot_commands.selling(item_name)
await ctx.send(
2018-09-15 15:24:39 +00:00
'{}, the following shop(s) sell **{}**: \n{}'.format(ctx.message.author.mention, item_name, result))
except ItemNotFound:
await ctx.send('{}, no shop sells **{}**.'.format(ctx.message.author.mention, item_name))
@commands.command(pass_context=True)
@commands.cooldown(5, 60, commands.BucketType.user)
async def info(self, ctx, *args):
"""
Displays info about a location.
If the location is a shop, it displays the shop's inventory
2018-08-18 17:01:59 +00:00
?info [Location Name]
"""
loc = get_name(args)
try:
if loc is None:
raise commands.UserInputError
info_str = self.bot.bot_commands.info(loc)
await ctx.send(info_str)
except IndexError:
await ctx.send('{}, no locations in the database match **{}**.'.format(ctx.message.author.mention, loc))
2018-08-10 23:03:30 +00:00
@commands.command(pass_context=True)
@commands.cooldown(5, 60, commands.BucketType.user)
2018-08-10 23:03:30 +00:00
async def me(self, ctx):
"""
2018-08-10 23:03:30 +00:00
Displays all your locations in the database
"""
2018-08-10 23:03:30 +00:00
try:
loc_str = self.bot.bot_commands.me(discord_uuid=ctx.message.author.id)
2018-09-15 15:24:39 +00:00
await ctx.send('{}, here are your location(s) in the database: \n {}'.format(ctx.message.author.mention,
loc_str))
2018-08-10 23:03:30 +00:00
except PlayerNotFound:
await ctx.send('{}, you don\'t have any locations in the database.'.format(ctx.message.author.mention))
2018-08-10 23:03:30 +00:00
def setup(bot):
bot.add_cog(Search_Commands(bot))