Example #1
0
def __main__():
    colors = [F.GREEN, F.YELLOW, F.BLUE, F.MAGENTA, F.CYAN]
    super_users = {'slewis':'U029RFUDH'}
    username = '******' #username for the bot
    token = os.getenv('stoken')
    dats_the_sound_of_da_police = False # enables police bot for @channels or @groups
    kill_me = False
    old_minute = 1000

    con = mdb.connect('localhost', os.getenv('dbuser1'), os.getenv('dbpass1'), 'rtm',
                      cursorclass=MySQLdb.cursors.DictCursor)
    cur = con.cursor()
    con2 = mdb.connect('localhost', os.getenv('dbuser2'), os.getenv('dbpass2'), 'rtm',
                      cursorclass=MySQLdb.cursors.DictCursor)
    cr = con2.cursor()

    print "Connecting"
    open_connection = rs.renew(token)
    if 'hello' in rs.to_u(open_connection.recv()):
        print 'Successfully Connected'
        open_connection.recv()
    icolor = 0
    while kill_me == False:
        try:
            # make sure we're connected to the db
            try:
                cur.fetchone()
                cr.fetchone()
            except:
                con = mdb.connect('localhost', os.getenv('dbuser1'), os.getenv('dbpass1'), 'rtm', cursorclass=mdb.cursors.DictCursor)
                cur = con.cursor()
                con2 = mdb.connect('localhost', os.getenv('dbuser2'), os.getenv('dbpass2'), 'rtm',
                                  cursorclass=MySQLdb.cursors.DictCursor)
                cr = con2.cursor()

            rightnow = datetime.datetime.now()
            hour = rightnow.hour
            minute = rightnow.minute
            is_weekday = 'True' if rightnow.isoweekday() < 6 else 'False'
            if old_minute != minute:
                sql = "SELECT * FROM Tasks WHERE (HOUR = %s OR HOUR = 247) AND Min = %s"
                query = cur.execute(sql, (hour, minute))
                if query > 0:
                    rows = cur.fetchall()
                    for row in rows:
                        if row["Weekday"] == is_weekday or row["Weekday"] == "All":
                            try:
                                exec(row["Command"])
                            except Exception, e:
                                rs.msg_sean("Scheduled problem: {} error: {}".format(row["Command"],str(e)), token)
            old_minute = minute
            try:
                event_json = json.loads(open_connection.recv())
            except:
                rs.msg_sean(str(open_connection.recv()), token)

            event = SlackEvent(event_json, cur, token)
            if 'user' in event_json.keys():
                event_json['user'] = event.name
            if 'channel' in event_json.keys():
                event_json['channel'] = event.channel_name

            if icolor >= len(colors):
                icolor = 0
            if ('message' not in event.etype) and event.subtype != "bot_message":
                print rs.colorize(event_json, icolor)
                icolor = icolor + 1
            else:

                if event.etype == 'message':
                    if 'gif me ' in event.text:
                        sql = "SELECT * FROM Gif_Whitelist WHERE Slack_Id = %s AND Channel = %s"
                        query = cur.execute(sql, (event.user_id, event.channel_id))
                        if query > 0 or (event.channel_id == rs.get_im(event.user_id, token)):
                            user_authed = True
                        else:
                            user_authed = False

                        if (user_authed == True) or (event.user_id in super_users.values()):
                            tag = event.text.replace('gif me ', '').replace(' ', '+')
                            subprocess.Popen(["python", rs.get_shades_path() + "gifme.py", tag, event.channel_id], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
                            print tag + ' ' + event.channel_id

                    if event.user_id in super_users.values():

                        if 'gif bless ' in event.text:
                            to_bless = event.text.replace('gif bless ', '')
                            sql = "SELECT * FROM Users WHERE Name LIKE %s OR Nick LIKE %s"
                            query = cur.execute(sql, (rs.like(to_bless), rs.like(to_bless)))
                            if query > 0:
                                row = cur.fetchone()
                                slackid = row["Slack_Id"]
                                to_bless = row["Name"]
                                sql = "SELECT * FROM Gif_Whitelist WHERE Slack_Id = %s AND Channel = %s"
                                query = cur.execute(sql, (slackid, event.channel_id))
                                if query > 0:
                                    message = "%s has already been blessed in this channel." % to_bless
                                else:
                                    sql = "INSERT INTO Gif_Whitelist (Slack_Id, Channel) VALUES(%s, %s)"
                                    cur.execute(sql, (slackid, event.channel_id))
                                    message = to_bless + " has been blessed with gif me powers in this channel."
                            else:
                                message = "Sorry brochinski, I don't know who " + to_bless + " is."
                            rs.post(event.channel_id, message, "based GIF GODS", token, icon_emoji=":fire:")

                        if 'gif revoke ' in event.text:
                            to_revoke = event.text.replace('gif revoke ', '')
                            sql = "SELECT * FROM Users WHERE Name LIKE %s or Nick LIKE %s"
                            query = cur.execute(sql, (rs.like(to_revoke), rs.like(to_revoke)))
                            if query > 0:
                                row = cur.fetchone()
                                slackid = row["Slack_Id"]
                                to_revoke = row["Name"]
                                sql = "SELECT * FROM Gif_Whitelist WHERE Slack_Id = %s AND Channel = %s"
                                query = cur.execute(sql, (slackid, event.channel_id))
                                if query > 0:
                                    sql = "DELETE FROM Gif_Whitelist WHERE Slack_Id = %s AND Channel = %s"
                                    cur.execute(sql, (slackid, event.channel_id))
                                    message = to_revoke + ' has been damned by the gif gods.'
                                else:
                                    message = to_revoke + " ain't blessed, yo"
                            else:
                                message = "Sorry brochinski, I don't know who " + to_revoke + " is."
                            rs.post(event.channel_id, message, "based GIF GODS", token, icon_emoji=":fire:")

                        if "!shutdown" in event.text:
                            rs.post(event.channel_id, 'going down', 'shades', token)
                            kill_me = True

                        if "!restart" in event.text:
                            rs.post(event.channel_id, 'restarting and updating, be back shortly', 'shades', token)
                            subprocess.Popen(["bash", rs.get_shades_path() + "restart.sh"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
                            kill_me = True

                        if "shades channel_police " in event.text:
                            if 'off' in event.text:
                                dats_the_sound_of_da_police == False
                                rs.msg_sean('Channel police has been turned off', token)
                            if 'on' in event.text:
                                dats_the_sound_of_da_police == True
                                rs.msg_sean('Channel police is back on baby!', token)

                        if "dbdump " in event.text:
                            try:
                                db = event.text.replace("dbdump ", "")
                                # getting extra quotes if we cur.execut(sql, (db)) on this one
                                sql = "SELECT * FROM {}".format(db)
                                query = cur.execute(sql)
                                if query > 0:
                                    temp_arr = []
                                    rows = cur.fetchall()
                                    for row in rows:
                                        temp_arr.append(row)
                                    data = str(temp_arr)
                                    message = data
                                else:
                                    message = "yo, I don't have {} in my db".format(db)
                            except Exception, e:
                                message = "problem with {}: \r {}".format(db, e)
                            rs.msg_sean(message, token)

                        if "shades tweet: " in event.text:
                            tweet = event.text.replace("shades tweet: ", "")
                            tweet = "{}".format(tweet)
                            subprocess.Popen(["python", rs.get_shades_path() + "tweet.py", tweet], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
                            message = "Sending tweet :bird:"
                            rs.post(event.channel_id, message, "tweetyshadesofgrey", token, icon_emoji=":bird:")

                        if "!schedule " in event.text:
                            try:
                                commands = event.text.replace("!schedule ", "")
                                weekday = re.search(r'(?<=w\:)\S+', commands).group()
                                hour = re.search(r'(?<=h\:)\d+', commands).group()
                                minute = re.search(r'(?<=m\:)\d+', commands).group()
                                command = re.search(r'(?<=cmd\:).*', commands).group()

                                sql = "INSERT INTO Tasks (Command, Weekday, Hour, Min) VALUES(%s, %s, %s, %s)"
                                cur.execute(sql, (command, weekday, hour, minute))
                                message = "Successfully scheduled task"
                            except Exception, problem:
                                message = "Couldn't schedule because: {}".format(problem)
                            rs.post(event.channel_id, message, "shades mcstatus", token)

                        if "reload resources" in event.text:
                            try:
                                reload(rs)
                                message = "Successfully reimported our resources file"
                            except:
                                message = "Was unable to reimport resources.py"
                            rs.msg_sean(message, token)

                        if re.search(r'^!e', event.text) != None:
                            code = event.text.replace("DOT", ".")
                            if re.search(r'^!ev ', event.text) != None:
                                code = code.replace("!ev ", "")
                                try:
                                    message = str(eval(code))
                                    rs.post(event.channel_id, message, 'shades mcPythonInterpreter', token)
                                except Exception, e:
                                    message = str(e)
                                    rs.post(event.channel_id, message, 'shades mcPythonInterpreter', token)

                            if re.search(r'^!ex ', event.text) != None:
                                code = code.replace("!ex ", "")
                                print code
                                try:
                                    exec(code)
                                    message = "success"
                                except Exception, e:
                                    message = str(e)
                                rs.post(event.channel_id, message, 'shades mcPythonInterpreter', token)
Example #2
0
            if (("gist.github" not in str(event_json)) and ("message_changed" not in str(event_json)) and ("presence_change" not in str(event_json))):
                try:
                    print rs.colorize(event_json, icolor, highlight=True)
                    icolor = icolor + 1
                except:
                    pass
                #       s_message = 'We got a problem sir, \r\n' + str(event)
                #       rs.msg_sean(s_message, token)
        except Exception, e:
            try:
                open_connection.close()
            except:
                pass
            connected = False
            while connected == False:
                try:
                    print "Disconnected, reconnecting. \r\n {}".format(str(e))
                    rs.msg_sean("Oops i died \r\n {}".format(str(e)), token)
                    open_connection = rs.renew(token)
                    if 'hello' in rs.to_u(open_connection.recv()):
                        print "Successfully reconnected."
                        connected = True
                except:
                    print "Failed to reconnect, retrying"
                    time.sleep(1)
        con.commit()
        time.sleep(.001)

__main__()