Geoffrey-Django/geoffrey/cogs/Add_Commands.py

from discord.ext import commands

from geoffrey.BotErrors import *
from geoffrey.DiscordHelperFunctions import *


@commands.cooldown(5, 60, commands.BucketType.user)
class Add_Commands:
    """
    Commands for adding things to Geoffrey.
    *You must use ?register before using any of these commands!*
    """

    def __init__(self, bot):
        self.bot = bot

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def register(self, ctx):
        """
        Registers your Discord and Minecraft account with the the database.
        You must do this before adding entries to the database.
        """

        try:
            player_name = get_nickname(ctx.message.author, self.bot.special_users)
            self.bot.bot_commands.register(player_name, ctx.message.author.id)
            await ctx.send('{}, you have been added to the database.'.format(ctx.message.author.mention))
        except AttributeError:
            raise NotOnServerError
        except PlayerInDBError:
            await ctx.send('{}, you are already in the database. Ding dong.'.format(ctx.message.author.mention))

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def add_base(self, ctx, x_pos: int, z_pos: int, *args):
        """
        Adds your base to the database. The base name is optional if this your first base.
            ?add_base [X Coordinate] [Z Coordinate] [Base Name]
        """

        name = get_name(args)

        try:
            base = self.bot.bot_commands.add_base(x_pos, z_pos, base_name=name, discord_uuid=ctx.message.author.id)
            await ctx.send(
                '{}, your base has been added to the database: \n\n{}'.format(ctx.message.author.mention, base))
        except LocationInitError:
            raise commands.UserInputError
        except EntryNameNotUniqueError:
            if name is None:
                await ctx.send('{}, you already have one base in the database, you need to specify a base'
                                   ' name'.format(ctx.message.author.mention))
            else:
                await ctx.send(
                    '{}, a base called **{}** already exists. You need to specify a different name.'.format(
                        ctx.message.author.mention, name))

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def add_shop(self, ctx, x_pos: int, z_pos: int, *args):
        """
        Adds your shop to the database. The name is shop optional if this your first shop.
            ?add_shop [X Coordinate] [Z Coordinate] [Shop Name]
        """

        name = get_name(args)

        try:
            shop = self.bot.bot_commands.add_shop(x_pos, z_pos, shop_name=name, discord_uuid=ctx.message.author.id)
            await ctx.send(
                '{}, your shop has been added to the database: \n\n{}'.format(ctx.message.author.mention, shop))
        except LocationInitError:
            raise commands.UserInputError
        except EntryNameNotUniqueError:
            if name is None:
                await ctx.send(
                    '{}, you already have one shop in the database, you need to specify a shop name'.format(
                        ctx.message.author.mention))
            else:
                await ctx.send(
                    '{}, a shop called **{}** already exists. You need to specify a different name.'.format(
                        ctx.message.author.mention, name))

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def add_tunnel(self, ctx, tunnel_color: str, tunnel_number: int, *args):
        """
        Adds your tunnel to the database. If you only have one location, you do not need to specify a location name.
            ?tunnel [Tunnel Color] [Tunnel Number] [Location Name]
        """

        loc_name = get_name(args)
        try:
            self.bot.bot_commands.add_tunnel(tunnel_color, tunnel_number, discord_uuid=ctx.message.author.id,
                                    location_name=loc_name)
            await ctx.send('{}, your tunnel has been added to the database'.format(ctx.message.author.mention))
        except LocationLookUpError:
            await ctx.send('{}, you do not have a location called **{}**.'.format(
                ctx.message.author.mention, loc_name))
        except LocationHasTunnelError:
            await ctx.send('{}, **{}** already has a tunnel.'.format(ctx.message.author.mention, loc_name))
        except TunnelInitError:
            await ctx.send('{}, invalid tunnel color.'.format(ctx.message.author.mention))
        except EntryNameNotUniqueError:
            await ctx.send('{}, you have more than one location, you need to specify a location.'
                               .format(ctx.message.author.mention))
        except InvalidTunnelError:
            await ctx.send(
                '{}, **{}** is an invalid tunnel color.'.format(ctx.message.author.mention, tunnel_color))

    @commands.command(pass_context=True)
    @commands.cooldown(5, 60, commands.BucketType.user)
    async def add_item(self, ctx, item_name: str, quantity: int, diamond_price: int, *args):
        """
        Adds an item to a shop's inventory. If you have one shop, the shop name is not required.
        Quantity for Diamond Price. eg. 32 Dirt for 1D
            ?add_item [Item Name] [Quantity] [Price] [Shop name]
        """
        shop_name = get_name(args)
        try:
            self.bot.bot_commands.add_item(item_name, quantity, diamond_price, shop_name=shop_name,
                                  discord_uuid=ctx.message.author.id)
            await ctx.send(
                '{}, **{}** has been added to the inventory of your shop.'.format(ctx.message.author.mention,
                                                                                  item_name))
        except PlayerNotFound:
            await ctx.send('{}, you don\'t have any shops in the database.'.format(ctx.message.author.mention))
        except LocationInitError:
            await ctx.send('{}, you have more than one shop in the database, please specify a shop name.'
                               .format(ctx.message.author.mention))
        except LocationLookUpError:
            await ctx.send(
                '{}, you don\'t have any shops named **{}** in the database.'.format(ctx.message.author.mention,
                                                                                     shop_name))


def setup(bot):
    bot.add_cog(Add_Commands(bot))