Geoffrey-Django/Geoffrey.py

456 lines
16 KiB
Python
Raw Normal View History

2018-05-21 15:59:35 +00:00
from discord.ext import commands
from Commands import *
from BotErrors import *
2018-06-03 02:42:31 +00:00
from MinecraftAccountInfoGrabber import *
from itertools import zip_longest
2018-07-21 18:28:31 +00:00
from BotConfig import *
import threading
2018-07-21 18:28:31 +00:00
2018-05-21 15:59:35 +00:00
command_prefix = '?'
2018-05-25 16:32:29 +00:00
description = '''
Geoffrey started his life as an inside joke none of you will understand.
At some point, she was to become an airhorn bot. Now, they know where your bases/shops are.
2018-05-22 02:41:15 +00:00
Please respect Geoffrey, the bot is very sensitive.
*You must use ?register before adding entries into Geoffrey*
2018-05-22 02:41:15 +00:00
'''
2018-05-25 16:32:29 +00:00
2018-05-25 22:00:43 +00:00
bad_error_message = 'OOPSIE WOOPSIE!! Uwu We made a fucky wucky!! A wittle fucko boingo! The admins at our ' \
2018-06-30 15:07:56 +00:00
'headquarters are working VEWY HAWD to fix this! (Error in command {}: {})'
2018-05-25 22:00:43 +00:00
2018-05-25 16:32:29 +00:00
bot = commands.Bot(command_prefix=command_prefix, description=description, case_insensitive=True)
# Bot Commands ******************************************************************'
2018-06-03 02:42:31 +00:00
2018-05-21 15:59:35 +00:00
@bot.event
async def on_ready():
2018-05-25 16:32:29 +00:00
print('GeoffreyBot')
print('Username: ' + bot.user.name)
print('ID: ' + bot.user.id)
@bot.event
async def on_command_error(error, ctx):
2018-05-25 20:30:47 +00:00
if isinstance(error, commands.CommandNotFound):
error_str = 'Command not found, ding dongs like you can use ?help to see all the commands this bot can do.'
elif isinstance(error, commands.CommandOnCooldown):
return
elif isinstance(error, commands.UserInputError):
error_str = 'Invalid syntax for **{}** you ding dong, please read ?help {}.'\
.format(ctx.invoked_with, ctx.invoked_with)
2018-06-03 02:42:31 +00:00
elif isinstance(error.original, UsernameLookupFailed):
error_str = error.original.__doc__
elif isinstance(error.original, PlayerNotFound):
error_str = 'Make sure to use ?register first you ding dong.'
elif isinstance(error.original, EntryNameNotUniqueError):
error_str = 'An entry in the database already has that name ding dong.'
elif isinstance(error.original, DatabaseValueError):
2018-07-23 00:01:49 +00:00
error_str = 'Use a shorter name or a smaller value, dong ding.'
2018-05-25 20:30:47 +00:00
else:
2018-06-30 15:07:56 +00:00
error_str = bad_error_message.format(ctx.invoked_with, error)
2018-05-25 20:30:47 +00:00
2018-07-23 00:01:49 +00:00
await bot.send_message(ctx.message.channel, '{} **Error Running Command:** {}'.format(ctx.message.author.mention,
error_str))
2018-05-25 16:32:29 +00:00
2018-05-21 15:59:35 +00:00
@commands.cooldown(5, 60, commands.BucketType.user)
2018-05-21 15:59:35 +00:00
@bot.command()
async def test():
2018-05-25 21:06:00 +00:00
'''
Checks if the bot is alive.
'''
2018-05-25 16:32:29 +00:00
await bot.say('I\'m here you ding dong')
@commands.cooldown(5, 60, commands.BucketType.user)
@bot.command(pass_context=True)
async def register(ctx):
'''
Registers your Discord and Minecraft account with the the database.
You must do this before adding entries to the database.
'''
try:
2018-07-21 03:10:13 +00:00
player_name = get_nickname(ctx.message.author)
bot_commands.register(player_name, ctx.message.author.id)
await bot.say('{}, you have been added to the database.'.format(ctx.message.author.mention))
2018-07-21 03:10:13 +00:00
except AttributeError:
await bot.say('{}, run this command on 24CC whoever you are'.format(ctx.message.author.mention))
except LocationInitError:
raise commands.UserInputError
except PlayerInDB:
await bot.say('{}, you are already in the database. Ding dong.'.format(ctx.message.author.mention))
@commands.cooldown(5, 60, commands.BucketType.user)
2018-05-21 15:59:35 +00:00
@bot.command(pass_context=True)
2018-08-01 01:16:14 +00:00
async def add_base(ctx, x_pos: int, z_pos: int, * args):
2018-05-25 16:32:29 +00:00
'''
Adds your base to the database.
The name is optional.
2018-08-01 01:16:14 +00:00
?add_base [X Coordinate] [Z Coordinate] [Base Name]
2018-05-25 16:32:29 +00:00
'''
name = get_name(args)
try:
2018-08-01 00:22:17 +00:00
base = bot_commands.add_base(x_pos, z_pos, base_name=name, discord_uuid=ctx.message.author.id)
await 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 bot.say('{}, you already have one base in the database, you need to specify a base'
' name'.format(ctx.message.author.mention))
else:
await bot.say('{}, a base called **{}** already exists. You need to specify a different name.'.format(
ctx.message.author.mention, name))
@commands.cooldown(5, 60, commands.BucketType.user)
2018-06-23 17:33:02 +00:00
@bot.command(pass_context=True)
2018-08-01 01:16:14 +00:00
async def add_shop(ctx, x_pos: int, z_pos: int, *args):
2018-06-23 17:33:02 +00:00
'''
Adds your shop to the database.
The name is optional.
2018-08-01 01:16:14 +00:00
?add_shop [X Coordinate] [Z Coordinate] [Shop Name]
2018-06-23 17:33:02 +00:00
'''
name = get_name(args)
2018-06-23 17:33:02 +00:00
try:
2018-08-01 01:16:14 +00:00
shop = bot_commands.add_shop(x_pos, z_pos, shop_name=name, discord_uuid=ctx.message.author.id)
await bot.say('{}, your shop has been added to the database: \n\n{}'.format(ctx.message.author.mention, shop))
2018-06-23 17:33:02 +00:00
except LocationInitError:
raise commands.UserInputError
except EntryNameNotUniqueError:
if name is None:
await bot.say('{}, you already have one shop in the database, you need to specify a shop name'.format(
ctx.message.author.mention))
else:
await bot.say('{}, a shop called **{}** already exists. You need to specify a different name.'.format(
ctx.message.author.mention, name))
2018-05-25 16:32:29 +00:00
@commands.cooldown(5, 60, commands.BucketType.user)
@bot.command(pass_context=True)
2018-08-01 01:16:14 +00:00
async def add_tunnel(ctx, tunnel_color: str, tunnel_number: int, *args):
'''
Adds your tunnel to the database.
The location name is optional.
?tunnel [Tunnel Color] [Tunnel Number] [Location Name]
'''
2018-08-01 01:16:14 +00:00
loc_name = get_name(args)
try:
2018-08-01 00:22:17 +00:00
bot_commands.add_tunnel(tunnel_color, tunnel_number, discord_uuid=ctx.message.author.id, location_name=loc_name)
await bot.say('{}, your tunnel has been added to the database'.format(ctx.message.author.mention))
except EntryNameNotUniqueError:
await bot.say('{}, you already have one tunnel in the database, please specify a location.'.format(
ctx.message.author.mention))
return
except LocationLookUpError:
await bot.say('{}, you do not have a location called **{}**.'.format(
2018-08-01 00:22:17 +00:00
ctx.message.author.mention, loc_name))
except LocationHasTunnelError:
2018-08-01 00:22:17 +00:00
await bot.say('{}, **{}** already has a tunnel.'.format(ctx.message.author.mention, loc_name))
except TunnelInitError:
await bot.say('{}, invalid tunnel color.'.format(ctx.message.author.mention))
except InvalidTunnelError:
2018-08-01 01:16:14 +00:00
await bot.say('{}, **{}** is an invalid tunnel color.'.format(ctx.message.author.mention, tunnel_color))
@commands.cooldown(5, 60, commands.BucketType.user)
2018-05-25 16:32:29 +00:00
@bot.command(pass_context=True)
async def find(ctx, * args):
2018-05-25 21:06:00 +00:00
'''
Finds all the locations and tunnels matching the search term
?find [Search]
2018-05-25 16:32:29 +00:00
'''
2018-08-01 01:16:14 +00:00
search = get_name(args)
2018-06-30 15:07:56 +00:00
try:
if search is None:
raise commands.UserInputError
result = bot_commands.find(search)
await bot.say('{}, The following entries match **{}**:\n{}'.format(ctx.message.author.mention, search, result))
except LocationLookUpError:
2018-07-29 15:50:24 +00:00
await bot.say('{}, no matches to **{}** were found in the database.'.format(ctx.message.author.mention, search))
2018-05-25 16:32:29 +00:00
@commands.cooldown(5, 60, commands.BucketType.user)
@bot.command(pass_context=True)
async def tunnel(ctx, player: str):
'''
Finds all the tunnels a player owns.
?tunnel [Player]
'''
try:
result = bot_commands.tunnel(player)
await bot.say('{}, **{}** owns the following tunnels: \n{}'.format(ctx.message.author.mention, player, result))
except LocationLookUpError:
2018-07-29 15:50:24 +00:00
await bot.say('{}, no tunnels for the player **{}** were found in the database/'
.format(ctx.message.author.mention, player))
@commands.cooldown(5, 60, commands.BucketType.user)
2018-05-25 21:33:18 +00:00
@bot.command(pass_context=True)
async def delete(ctx, * args):
2018-05-25 21:33:18 +00:00
'''
2018-06-23 21:51:00 +00:00
Deletes a location from the database.
?delete [Location name]
2018-05-25 21:33:18 +00:00
'''
2018-08-01 01:16:14 +00:00
loc = get_name(args)
try:
if loc is None:
raise commands.UserInputError
bot_commands.delete(loc, discord_uuid=ctx.message.author.id)
await bot.say('{}, your location named **{}** has been deleted.'.format(ctx.message.author.mention, loc))
2018-06-30 15:07:56 +00:00
except (DeleteEntryError, PlayerNotFound):
await bot.say('{}, you do not have a location named **{}**.'.format(ctx.message.author.mention, loc))
2018-05-25 21:33:18 +00:00
2018-06-02 17:40:55 +00:00
2018-05-26 14:26:24 +00:00
@bot.command(pass_context=True)
@commands.cooldown(5, 60, commands.BucketType.user)
2018-08-01 01:16:14 +00:00
async def find_around(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 overworld.
2018-08-01 01:16:14 +00:00
?find_around [X Coordinate] [Z Coordinate] [Radius] [Optional Flags]
Optional Flags:
-d [dimension]
'''
2018-07-21 18:49:10 +00:00
try:
radius = 200
dimension = 'Overworld'
2018-07-21 18:49:10 +00:00
if len(args) > 0:
if args[0] == '-d':
dimension = args[1]
else:
radius = int(args[0])
2018-07-21 18:49:10 +00:00
if len(args) > 1:
if args[1] == '-d':
dimension = args[2]
2018-08-01 00:22:17 +00:00
base_string = bot_commands.find_around(x_pos, z_pos, radius, dimension)
2018-05-26 14:26:24 +00:00
if len(base_string) != 0:
await bot.say('{}, the following locations(s) within **{}** blocks of that point: \n {}'.format(
ctx.message.author.mention, radius, base_string))
2018-07-21 18:49:10 +00:00
else:
await bot.say('{}, there are no locations within {} blocks of that point'
.format(ctx.message.author.mention, radius))
except InvalidDimError:
await bot.say('{}, {} is an invalid dimension.'.format(ctx.message.author.mention, dimension))
2018-05-26 14:26:24 +00:00
@bot.command(pass_context=True)
@commands.cooldown(5, 60, commands.BucketType.user)
2018-08-01 01:16:14 +00:00
async def add_item(ctx, item_name: str, quantity: int, diamond_price: int, * args):
2018-06-23 21:51:00 +00:00
'''
Adds an item to a shop's inventory.
Quantity for Diamond Price.
?additem [Item Name] [Quantity] [Price] [Shop name]
2018-06-23 21:51:00 +00:00
'''
2018-08-01 01:16:14 +00:00
shop_name = get_name(args)
2018-06-30 15:07:56 +00:00
try:
2018-08-01 00:22:17 +00:00
bot_commands.add_item(item_name, quantity, diamond_price, shop_name=shop_name,
discord_uuid=ctx.message.author.id)
await bot.say('{}, **{}** has been added to the inventory of your shop.'.format(ctx.message.author.mention,
item_name))
2018-06-30 15:07:56 +00:00
except PlayerNotFound:
await bot.say('{}, you don\'t have any shops in the database.'.format(ctx.message.author.mention))
except LocationInitError:
await bot.say('{}, you have more than one shop in the database, please specify a shop name.'
.format(ctx.message.author.mention))
2018-06-30 15:07:56 +00:00
except LocationLookUpError:
await bot.say('{}, you don\'t have any shops named **{}** in the database.'.format(ctx.message.author.mention,
shop_name))
@commands.cooldown(5, 60, commands.BucketType.user)
@bot.command(pass_context=True)
async def selling(ctx, item_name: str):
2018-06-23 21:51:00 +00:00
'''
Lists all the shops selling an item
?selling [item]
2018-06-23 21:51:00 +00:00
'''
try:
result = bot_commands.selling(item_name)
await bot.say('{}, the following shops sell **{}**: \n{}'.format(ctx.message.author.mention, item_name, result))
except ItemNotFound:
await bot.say('{}, no shop sells **{}**.'.format(ctx.message.author.mention, item_name))
@commands.cooldown(5, 60, commands.BucketType.user)
@bot.command(pass_context=True)
2018-07-21 22:25:18 +00:00
async def info(ctx, * args):
'''
Displays info about a location.
If the location is a shop, it displays the shop's inventory.
?info [Location Name]
'''
2018-08-01 01:16:14 +00:00
loc = get_name(args)
2018-07-21 03:10:13 +00:00
try:
if loc is None:
raise commands.UserInputError
info_str = bot_commands.info(loc)
await bot.say(info_str)
2018-07-21 03:10:13 +00:00
except IndexError:
2018-08-01 00:22:17 +00:00
await bot.say('{}, no locations in the database match {}.'.format(ctx.message.author.mention, loc))
2018-07-21 03:10:13 +00:00
return
2018-08-01 00:22:17 +00:00
@commands.cooldown(5, 60, commands.BucketType.user)
@bot.command(pass_context=True)
async def edit_pos(ctx, x_pos: int, y_pos: int, * args):
'''
Edits the position of a location
?edit_pos [X Coordinate] [Z Coordinate] [Location Name]
'''
2018-08-01 01:16:14 +00:00
loc = get_name(args)
2018-08-01 00:22:17 +00:00
try:
loc_str = bot_commands.edit_pos(x_pos, y_pos, loc, discord_uuid=ctx.message.author.id)
await bot.say('{}, the following location has been updated: \n\n{}'.format(ctx.message.author.mention, loc_str))
except LocationLookUpError:
await bot.say('{}, you do not have a location called **{}**.'.format(
ctx.message.author.mention, loc))
@commands.cooldown(5, 60, commands.BucketType.user)
@bot.command(pass_context=True)
async def edit_tunnel(ctx, tunnel_color: str, tunnel_number: int, * args):
'''
Edits the tunnel of a location
?edit_tunnel [Tunnel Color] [Tunnel Number] [Location Name]
'''
2018-08-01 01:16:14 +00:00
loc = get_name(args)
2018-08-01 00:22:17 +00:00
try:
loc_str = bot_commands.edit_tunnel(tunnel_color, tunnel_number, loc, discord_uuid=ctx.message.author.id)
await bot.say('{}, the following location has been updated: \n\n{}'.format(ctx.message.author.mention, loc_str))
except LocationLookUpError:
await bot.say('{}, you do not have a location called **{}**.'.format(
ctx.message.author.mention, loc))
2018-08-01 00:31:21 +00:00
except InvalidTunnelError:
2018-08-01 01:16:14 +00:00
await bot.say('{}, **{}** is an invalid tunnel color.'.format(ctx.message.author.mention, tunnel_color))
2018-08-01 00:22:17 +00:00
@commands.cooldown(5, 60, commands.BucketType.user)
@bot.command(pass_context=True)
async def edit_name(ctx, new_name: str, current_name: str):
'''
Edits the name of a location
IF A NAME HAS SPACES IN IT YOU NEED TO WRAP IT IN QUOTATION MARKS. ie "Cool Shop 123"
?edit_name [New Name] [Current Name]
'''
try:
loc_str = bot_commands.edit_name(new_name, current_name, discord_uuid=ctx.message.author.id)
await bot.say('{}, the following location has been updated: \n\n{}'.format(ctx.message.author.mention, loc_str))
except LocationLookUpError:
await bot.say('{}, you do not have a location called **{}**.'.format(
ctx.message.author.mention, current_name))
2018-08-01 01:16:14 +00:00
@commands.cooldown(5, 60, commands.BucketType.user)
@bot.command(pass_context=True)
async def delete_item(ctx, item: str, * args):
'''
Deletes an item listing from a shop inventory
?delete_name [Item] [Shop Name]
'''
shop = get_name(args)
try:
bot_commands.delete_item(item, shop, discord_uuid=ctx.message.author.id)
await bot.say('{}, **{}** has been removed from the inventory of **{}**.', item, shop)
except LocationLookUpError:
await bot.say('{}, you do not have a shop called **{}**.'.format(ctx.message.author.mention, shop))
except DeleteEntryError:
await bot.say('{}, **{}** does not sell **{}**.', shop, item)
# Helper Functions ************************************************************
def get_name(args):
if len(args) > 0:
name = ' '.join(args)
else:
name = None
return name
def get_nickname(discord_user):
if discord_user.nick is None:
2018-06-30 15:07:56 +00:00
name = discord_user.display_name
else:
2018-06-30 15:07:56 +00:00
name = discord_user.nick
if name == 'dootb.in ꙩ ⃤':
name = 'aeskdar'
return name
def get_args_dict(args):
if len(args) != 0:
2018-08-01 00:22:17 +00:00
return dict(zip_longest(*[iter(args)] * 2, fillvalue=" "))
else:
return {}
def update_user_names(bot_commands):
threading.Timer(600, update_user_names, [bot_commands]).start()
session = bot_commands.interface.database.Session()
print("Updating MC usernames...")
player_list = session.query(Player).all()
for player in player_list:
player.name = grab_playername(player.mc_uuid)
session.commit()
session.close()
2018-05-25 16:32:29 +00:00
# Bot Startup ******************************************************************
2018-07-21 18:28:31 +00:00
config = read_config()
2018-07-21 18:28:31 +00:00
TOKEN = config['Discord']['Token']
2018-06-30 15:53:27 +00:00
2018-07-21 18:28:31 +00:00
engine_arg = get_engine_arg(config)
2018-06-30 15:53:27 +00:00
bot_commands = Commands(engine_arg)
update_user_names(bot_commands)
2018-07-21 18:28:31 +00:00
bot.run(TOKEN)
2018-05-21 16:57:20 +00:00