commit e09cf4cbfe80c8a230cc1118123a2f508cfa3c52 Author: Kenneth Jao Date: Tue Feb 28 00:54:45 2017 -0500 Initial diff --git a/breakbot/date.dat b/breakbot/date.dat new file mode 100644 index 0000000..2e07a9a Binary files /dev/null and b/breakbot/date.dat differ diff --git a/breakbot/slackbot.py b/breakbot/slackbot.py new file mode 100644 index 0000000..79e6f89 --- /dev/null +++ b/breakbot/slackbot.py @@ -0,0 +1,105 @@ +import time +import datetime +import pickle +from slackclient import SlackClient +from random import randint + +BOT_ID = 'U4AMH4C4C' +dt = datetime.datetime +f = open('date.dat','rb') +targetDate = pickle.load(f) +f.close() + +# constants +AT_BOT = "<@" + BOT_ID + ">" +EXAMPLE_COMMAND = "do" + +# instantiate Slack & Twilio clients +slack_client = SlackClient("xoxb-146731148148-cOhyqUViCY4wI2gWYMcoOkpW") + +def handle_command(command, channel): + global targetDate + global dt + response = "Not sure what you mean. Try @break help!" + if command.startswith("days left in hell"): + response = "You have *" + str((targetDate.date() - dt.now().date()).days) + "* days left." + + elif command.startswith("time left in hell"): + timeDelta = str(targetDate - dt.now()) + days, other = timeDelta.split(",") + hours, minutes, seconds = other.split(":") + response = "Only " + days + ","+hours+" hours, "+minutes+" minutes, and "+seconds.split(".")[0]+" seconds left!" + elif command.startswith("summary"): + timeDelta = str(targetDate - dt.now()) + days, other = timeDelta.split(",") + hours, minutes, seconds = other.split(":") + approx = [ + str((targetDate.date() - dt.now().date()).days) + " days, ", + str(int(hours) + int(days.replace(" days",""))*24) + " hours, ", + ] + approx.append(str(int(minutes)+int(approx[1].replace(" hours, ",""))*60)+" minutes, or ") + approx.append(str(int(minutes)+int(approx[2].replace(" minutes, or ",""))*60)+" seconds.") + + first = "The current target date is *" + targetDate.strftime("%B %d, %Y") + "*!" + second = "There are " + days + ","+hours+" hours, "+minutes+" minutes, and "+seconds.split(".")[0]+" seconds left!" + third = "Other representations are: " + "".join(approx); + response = first+"\n"+second+"\n"+third + elif command.startswith("set new time "): + newDate = command.lower().replace("set new time ","").split(" ") + try: + newDate = dt(int(newDate[2]),int(newDate[0]),int(newDate[1])) + if(newDate < dt.now()): + response = "Please enter in a date after today!" + else: + targetDate = newDate + f = open('date.dat', 'wb') + pickle.dump(targetDate,f) + f.close() + response = "Ok! Your new date is *" + targetDate.strftime("%B %d, %Y") + "*!" + except ValueError: + response = "That's an invalid date! The format is: *Day Month Year*" + + elif command.startswith("target date"): + response = "The current target date is *" + targetDate.strftime("%B %d, %Y") + "*!" + elif command.startswith("flip a coin"): + coin = ["Heads!", "Tails!"] + response = coin[randint(0,1)] + elif command.startswith("?") or command.lower().startswith("help"): + commands = [ + "Summary: Gives a summary of all information.", + "Days left in hell: Gives days left to date.", + "Time left in hell: Gives countdown left to date.", + "Set new time [Day Month Year]: Sets new target date.", + "Target date: Gives current target date.", + "Flip a coin: Gives heads or tails." + ] + response = "My current commands are:```"+'\n'.join(commands)+"```" + slack_client.api_call("chat.postMessage", channel=channel,text=response, as_user=True) +def parse_slack_output(slack_rtm_output): + """ + The Slack Real Time Messaging API is an events firehose. + this parsing function returns None unless a message is + directed at the Bot, based on its ID. + """ + output_list = slack_rtm_output + if output_list and len(output_list) > 0: + for output in output_list: + if output and 'text' in output and AT_BOT in output['text']: + # return text after the @ mention, whitespace removed + print (output) + return output['text'].split(AT_BOT)[1].strip().lower(), \ + output['channel'] + + return None, None + +if __name__ == "__main__": + READ_WEBSOCKET_DELAY = 1 # 1 second delay between reading from firehose + if slack_client.rtm_connect(): + print("Bot connected and running!") + while True: + command, channel = parse_slack_output(slack_client.rtm_read()) + if command and channel: + handle_command(command, channel) + time.sleep(READ_WEBSOCKET_DELAY) + else: + print("Connection failed. Invalid Slack token or bot ID?") diff --git a/breakbot/slackid.py b/breakbot/slackid.py new file mode 100644 index 0000000..79259fb --- /dev/null +++ b/breakbot/slackid.py @@ -0,0 +1,19 @@ +import os +from slackclient import SlackClient + + +BOT_NAME = 'break' + +slack_client = SlackClient("xoxb-146752022133-ChFrlR3ptyZEJ9uctmOEKL5Z") + + +if __name__ == "__main__": + api_call = slack_client.api_call("users.list") + if api_call.get('ok'): + # retrieve all users so we can find our bot + users = api_call.get('members') + for user in users: + if 'name' in user and user.get('name') == BOT_NAME: + print("Bot ID for '" + user['name'] + "' is " + user.get('id')) + else: + print("could not find bot user with the name " + BOT_NAME)