From 2fc4f4f7ab3f6cf31175ae7921946ca8457cb341 Mon Sep 17 00:00:00 2001 From: Luke Robles Date: Wed, 9 Aug 2017 13:56:11 -0700 Subject: [PATCH] Condensing the help method into just two lines!, fixes #8 --- app/dragon-bot.py | 21 ++------------------- app/help_methods.py | 11 +++++++++++ 2 files changed, 13 insertions(+), 19 deletions(-) diff --git a/app/dragon-bot.py b/app/dragon-bot.py index 89a92c5f..7c153d27 100644 --- a/app/dragon-bot.py +++ b/app/dragon-bot.py @@ -131,25 +131,8 @@ async def on_message(message): await client.send_message(message.channel, "`{}`\n```{}```".format(' '.join(word), definition)) if message.content.startswith('!help'): - if len(message.content.split()) > 1: - method = message.content.split()[1] - try: - explanation = help_methods.get_help_message(method) - except KeyError: - await client.send_message( - message.channel, - "I can't help you with that" - ) - return - await client.send_message( - message.channel, - "```{}```".format(explanation) - ) - else: - await client.send_message( - message.channel, - help_methods.get_help_message('show_all') - ) + help_message = help_methods.parse_message(message.content) + await client.send_message(message.channel, help_message) if 'autis' in message.content or message.content.startswith('!triggered'): diff --git a/app/help_methods.py b/app/help_methods.py index a01faf5d..dd039af5 100644 --- a/app/help_methods.py +++ b/app/help_methods.py @@ -45,3 +45,14 @@ def get_help_message(method): else: message = ' '.join(supported_methods[method]) return message + +def parse_message(message): + if len(message.split()) > 1: + method = message.split()[1] + try: + explanation = get_help_message(method) + except KeyError: + return "I can't help you with that" + return "```{}```".format(explanation) + else: + return get_help_message('show_all')