Geoffrey-Django/cogs/Add_Commands.py

from discord.ext import commands
from BotErrors import *
from DiscordHelperFunctions import *
from bot import bot_commands


@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)
            bot_commands.register(player_name, ctx.message.author.id)
            await self.bot.say('{}, you have been added to the database.'.format(ctx.message.author.mention))
        except AttributeError:
            raise NotOnServerError
        except PlayerInDBError:
            await self.bot.say('{}, 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 name is optional.
            ?add_base [X Coordinate] [Z Coordinate] [Base Name]
        '''

        name = get_name(args)

        try:
            base = bot_commands.add_base(x_pos, z_pos, base_name=name, discord_uuid=ctx.message.author.id)
            await self.bot.say(
                '{}, 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 self.bot.say('{}, you already have one base in the database, you need to specify a base'
                              ' name'.format(ctx.message.author.mention))
            else:
                await self.bot.say('{}, 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 optional.
            ?add_shop [X Coordinate] [Z Coordinate] [Shop Name]
        '''

        name = get_name(args)

        try:
            shop = bot_commands.add_shop(x_pos, z_pos, shop_name=name, discord_uuid=ctx.message.author.id)
            await self.bot.say(
                '{}, 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 self.bot.say('{}, you already have one shop in the database, you need to specify a shop name'.format(
                    ctx.message.author.mention))
            else:
                await self.bot.say('{}, 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:
            bot_commands.add_tunnel(tunnel_color, tunnel_number, discord_uuid=ctx.message.author.id,
                                    location_name=loc_name)
            await self.bot.say('{}, your tunnel has been added to the database'.format(ctx.message.author.mention))
        except LocationLookUpError:
            await self.bot.say('{}, you do not have a location called **{}**.'.format(
                ctx.message.author.mention, loc_name))
        except LocationHasTunnelError:
            await self.bot.say('{}, **{}** already has a tunnel.'.format(ctx.message.author.mention, loc_name))
        except TunnelInitError:
            await self.bot.say('{}, invalid tunnel color.'.format(ctx.message.author.mention))
        except EntryNameNotUniqueError:
            await self.bot.say('{}, you have more than one location, you need to specify a location.'
                          .format(ctx.message.author.mention))
        except InvalidTunnelError:
            await self.bot.say('{}, **{}** 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.
        Quantity for Diamond Price.
        ?additem [Item Name] [Quantity] [Price] [Shop name]
        '''
        shop_name = get_name(args)
        try:
            bot_commands.add_item(item_name, quantity, diamond_price, shop_name=shop_name,
                                  discord_uuid=ctx.message.author.id)
            await self.bot.say('{}, **{}** has been added to the inventory of your shop.'.format(ctx.message.author.mention,
                                                                                            item_name))
        except PlayerNotFound:
            await self.bot.say('{}, you don\'t have any shops in the database.'.format(ctx.message.author.mention))
        except LocationInitError:
            await self.bot.say('{}, you have more than one shop in the database, please specify a shop name.'
                          .format(ctx.message.author.mention))
        except LocationLookUpError:
            await self.bot.say(
                '{}, 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))