merakibot.py

import os
import time
import re
from slackclient import SlackClient
from client import Client
import json

# instantiate Slack client
slack_client = SlackClient(os.environ.get('MERAKI_BOT_TOKEN'))

# merakibot's user ID in Slack: vlaue is assigned after the bot starts.
merakibot_id = None

# constants
RTM_READ_DELAY = 1 # 1 second delay between reading from RTM
GET_CLIENT = "get_client"
MENTION_REGEX = "^<@(|[WU].+?)>(.*)"

def parse_bot_commands(slack_events):
    """Parses a list of events coming from the Slack RTM API to find
    bot commands. If a bot command is found, this function returns a tuple
    of command and channel. If its not found, then this function returns None,
    None"""
    
    for event in slack_events:
        if event["type"] == "message" and not "subtype" in event:
            user_id, message = parse_direct_mention(event["text"])
            if user_id == merakibot_id:
                return message, event["channel"]
    return None, None

def parse_direct_mention(message_text):
    """Finds a direct mention (a mention that is at the beginning) in message
    text and returns the user ID which was mentioned. if there is no direct
    mention returns None"""
    
    matches = re.search(MENTION_REGEX, message_text)
    """The first group contains the username, the second group contains the
    remaining message"""
    return (matches.group(1), matches.group(2).strip()) if matches else (None, None)
 
def handle_command(command, channel):
    """Executes bot command if the command is known"""

    # Default response is help text for the user
    default_response = "Not sure what you mean. Try *{}*.".format(GET_CLIENT)

    # Finds and executes the given command, filling in response
    response = None

    # This is where you implement commands
    if command.startswith(GET_CLIENT):
        client = Client(command[11:]).clientinfo()
        response = format(client)

    # Sends the response back to the channel
    slack_client.api_call(
        "chat.postMessage",
        channel=channel,
        text=response or default_response
    )

if __name__ == "__main__":
    if slack_client.rtm_connect(with_team_state=False):
        print("Meraki Bot connected and running!")
        # Read bot's user ID by calling Web API method `auth.test`
        merakibot_id = slack_client.api_call("auth.test")["user_id"]
        while True:
            command, channel = parse_bot_commands(slack_client.rtm_read())
            if command:
                handle_command(command, channel)
            time.sleep(RTM_READ_DELAY)
    else:
        print("Connection failed. Exception traceback printed above.")