diff --git a/app/dragon-bot.py b/app/dragon-bot.py index 005d4138..74e16354 100644 --- a/app/dragon-bot.py +++ b/app/dragon-bot.py @@ -14,15 +14,16 @@ tokens = { 'test': 'MzQ1MjkwMTI5OTQ4Mjc4Nzg0.DG5IBw._9umb82PrL22bPe7GjmHClU-NtU', 'prod': 'MzM5NDQ2NTgwMTU3MzQ5ODg4.DG5K5Q.2kIonA_XHLXU4_Sq4O63OzCb0Jc' } -token = os.getenv('DRAGON_ENV') +dragon_environment = os.getenv('DRAGON_ENV') +debug = dragon_environment == 'test' @client.event async def on_ready(): print("\n********************************") - print("\nDRAGON BOT RUNNING IN {} MODE".format(token.upper())) + print("\nDRAGON BOT RUNNING IN {} MODE".format(dragon_environment.upper())) print("\n********************************") - if token == 'test': + if debug: print("\nPress control+c to exit the bot") print("Followed by control+d or by typing") print("'exit' to exit the docker container") @@ -146,9 +147,10 @@ async def on_message(message): await client.send_message(message.channel, help_methods.get_help_message('define')) if message.content.startswith('!help'): - help_message = help_methods.parse_message(message.content) - await client.send_message(message.channel, help_message) - + await client.send_message( + message.channel, + help_methods.parse_message(message.content) + ) if 'autis' in message.content or message.content.startswith('!triggered'): await client.send_message(message.channel, 'https://i.imgur.com/g6yOJjp.gif') @@ -266,4 +268,4 @@ async def on_message(message): await client.send_message(message.channel, "There arent {} lines of output yet".format(num_lines)) -client.run(tokens[token]) \ No newline at end of file +client.run(tokens[dragon_environment]) \ No newline at end of file