Initial

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?")

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)