from BotErrors import * from DatabaseModels import Player from Commands import Commands from discord.ext import commands from discord import Game from MinecraftAccountInfoGrabber import * from BotConfig import * import asyncio import logging logger = logging.getLogger(__name__) description = ''' Geoffrey (pronounced JOFF-ree) 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 stuff is. Please respect Geoffrey, the bot is very sensitive. If have a suggestion or if something is borked, you can PM my ding dong of a creator ZeroHD. *You must use ?register before adding things to Geoffrey* ''' bad_error_message = 'OOPSIE WOOPSIE!! Uwu We made a fucky wucky!! A wittle fucko boingo! The admins at our ' \ 'headquarters are working VEWY HAWD to fix this! (Error in command {}: {})' bot = commands.Bot(command_prefix=bot_config.prefix, description=description, case_insensitive=True) bot_commands = Commands() extensions = ['cogs.Add_Commands', 'cogs.Delete_Commands', 'cogs.Edit_Commands', 'cogs.Search_Commands', 'cogs.Admin_Commands'] @bot.event async def on_ready(): print('GeoffreyBot') print('Username: ' + bot.user.name) print('ID: ' + bot.user.id) logger.info("Geoffrey Online, ID: %s", bot.user.id) await bot.change_presence(game=Game(name=bot_config.status)) @bot.event async def on_command_error(error, ctx): if hasattr(ctx, 'cog'): if "Admin_Commands" in ctx.cog.__str__(): return 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) elif isinstance(error.original, NoPermissionError): error_str = 'You don\'t have permission for that cool command.' elif isinstance(error.original, UsernameLookupFailed): error_str = 'Your user name was not found, either Mojang is having a fucky wucky ' \ 'or your nickname is not set correctly. *stares at the Mods*' 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): error_str = 'Use a shorter name or a smaller value, dong ding.' elif isinstance(error.original, NotOnServerError): error_str = 'Command needs to be run on 24CC. Run this command there whoever you are.'.format() else: logger.error("Geoffrey encountered unhandled exception: %s", error) error_str = bad_error_message.format(ctx.invoked_with, error) await bot.send_message(ctx.message.channel, '{} **Error Running Command:** {}'.format(ctx.message.author.mention, error_str)) async def username_update(): session = None await bot.wait_until_ready() while not bot.is_closed: session = bot_commands.interface.database.Session() try: print("Updating MC usernames...") session = bot_commands.interface.database.Session() player_list = session.query(Player).all() for player in player_list: player.name = grab_playername(player.mc_uuid) session.commit() print("Done.") await asyncio.sleep(600) except UsernameLookupFailed: logger.info("Username lookup error.") print("Username lookup error, are Mojang's servers down?") session.rollback() finally: session.close() if session is not None: session.close() def start_bot(): for extension in extensions: try: bot.load_extension(extension) except Exception as e: logger.info('Failed to load extension {}, {}'.format(extension, e)) print('Failed to load extension {}, {}'.format(extension, e)) try: bot.loop.create_task(username_update()) logger.info('Logging into discord...') bot.run(bot_config.token) except TimeoutError: print("Disconnected, is Discord offline?") logger.info('Disconnected, is Discord offline?') finally: logger.info("Bot shutting down...") print("Bot shutting down...")