Пример #1
0
import requests, json, os, sys
from requests_oauthlib import OAuth1
import resources as rs

apptoken = "g9uAl2oo4Y1lkioWPSxSVJcmP"
appsecret = "ctWopK2Mptgm2qCizYzN0sBZuhToZHwv6KOrBrekSQEpXwGU3M"
mytoken = os.getenv('twittertoken')
mysecret = os.getenv('twittersecret')
token = os.getenv('stoken')

url = 'https://api.twitter.com/1.1/statuses/update.json'
tweet = str(sys.argv[1])
auth = OAuth1(apptoken, appsecret, mytoken, mysecret)
data = {"status": tweet}

send_tweet = requests.post(url, auth=auth, data=data)

if send_tweet.status_code == requests.codes.ok:
    message = "Successfully tweeted {}".format(tweet)
    rs.msg_sean(message, token)
else:
    message = "Was not able to tweet {} \r\n Reason: {}".format(tweet, send_tweet.text)
    rs.msg_sean(message, token)

Пример #2
0
import requests, json, os, sys
from requests_oauthlib import OAuth1
import resources as rs

apptoken = "g9uAl2oo4Y1lkioWPSxSVJcmP"
appsecret = "ctWopK2Mptgm2qCizYzN0sBZuhToZHwv6KOrBrekSQEpXwGU3M"
mytoken = os.getenv('twittertoken')
mysecret = os.getenv('twittersecret')
token = os.getenv('stoken')

url = 'https://api.twitter.com/1.1/statuses/update.json'
tweet = str(sys.argv[1])
auth = OAuth1(apptoken, appsecret, mytoken, mysecret)
data = {"status": tweet}

send_tweet = requests.post(url, auth=auth, data=data)

if send_tweet.status_code == requests.codes.ok:
    message = "Successfully tweeted {}".format(tweet)
    rs.msg_sean(message, token)
else:
    message = "Was not able to tweet {} \r\n Reason: {}".format(
        tweet, send_tweet.text)
    rs.msg_sean(message, token)
Пример #3
0
                            query = cur.execute(sql, (team))
                            if query > 0:
                                team_row = cur.fetchone()
                                current_n = int(team_row["N_Number"])
                                if current_n != n_number:
                                    sql = "UPDATE 2n_Teams SET N_Number = %s WHERE Team = %s"
                                    cur.execute(sql, (n_number, team))
                                    message = team + ' n number changed from ' + str(team_row["N_Number"]) + ' to ' + str(n_number)
                                else:
                                    message = "Current N number is already that value. Nothing to change."
                            else:
                                sql = "INSERT INTO 2n_Teams (Team, N_Number) VALUES(%s, %s)"
                                cur.execute(sql, (team, n_number))
                                message = team + ' added with an n number of ' + str(n_number)
                            rs.post(event.channel_id, message, '2n bot', token, icon_emoji=':robot:')
                            rs.msg_sean(message + ' changed by ' + event.name, token)

                        if re.search(r'^!2n delete \S*$', event.text) and event.user_id in super_users.values():
                            team = re.search(r'^!2n delete \S*$', event.text).group().split(' ')[2]
                            sql = "SELECT * FROM 2n_Teams WHERE Team = '%s'" % team
                            query = cur.execute(sql)
                            if query > 0:
                                sql = "DELETE FROM 2n_Teams WHERE Team = '%s'" % team
                                cur.execute(sql)
                                sql = "DELETE FROM 2n_Nicks WHERE Team = '%s'" % team
                                cur.execute(sql)
                                msg = team + ' has been deleted and all associated nicknames removed.'
                                rs.post(event.channel_id, msg, '2n bot', token, icon_emoji=':robot:')
                                rs.msg_sean(msg + ' by ' + event.name, token)
                            else:
                                rs.post(event.channel_id, team + " doesn't exist.", '2n bot', token, icon_emoji=':robot:')
Пример #4
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)
Пример #5
0
def __main__():
    slack_token = str(sys.argv[1])
    channel = str(sys.argv[2])

    con = mdb.connect('localhost', os.getenv('dbuser1'), os.getenv('dbpass1'), 'rtm',
                      cursorclass=MySQLdb.cursors.DictCursor)
    cur = con.cursor()
    cur.execute("SELECT * FROM Monikers WHERE Network = ':league:'")
    full_list = cur.fetchall()
    leaguers = list(chunks(full_list, 35))
    bad_names = []
    for leaguer_list in leaguers:
        namelist = []
        for leaguer in leaguer_list:
            name = leaguer["Moniker"]
            namelist.append(name)
        namelist = ','.join(namelist)
        for summoner in summoner_get_by_name(namelist).items():
            summoner = summoner[1]
            sql = "SELECT * FROM League WHERE S_ID = '%s'"
            query = cur.execute(sql, (summoner["id"]))
            if query == 0:
                # add the user to the League table if they're not in it yet
                sql = "SELECT * FROM Monikers WHERE Network = ':league:' AND Moniker = %s"
                query = cur.execute(sql, (summoner["name"]))
                if query > 0:
                    user = cur.fetchone()
                    sql = "INSERT INTO League (Slack_Id, Name, S_ID, Level) VALUES(%s, %s, %s, %s)"
                    cur.execute(sql, (user["Slack_Id"], summoner["name"], summoner["id"], summoner["summonerLevel"]))
                else:
                    bad_names.append(summoner["name"])
            else:
                # update the user in the table if their level changes
                rank = summoner_get_current_rank(summoner["id"])
                rank_sql = "SELECT * FROM League WHERE S_ID = %s AND Rank = %s"
                rank_query = cur.execute(rank_sql, (summoner["id"], rank))
                sql = "SELECT * FROM League WHERE S_ID = '%s' AND Level = '%s'"
                query1 = cur.execute(sql, (summoner["id"], summoner["summonerLevel"]))
                # now check to make sure they're in the Moniker table
                # (riot api fuzzy matches names better than i can)
                sql = "SELECT * FROM Monikers WHERE Network = ':league:' AND Moniker = %s"
                query = cur.execute(sql, (summoner["name"]))
                if query > 0 and query1 == 0:
                    user = cur.fetchone()
                    sql = "UPDATE League SET Level = %s WHERE S_ID = %s"
                    cur.execute(sql, (summoner["summonerLevel"], summoner["id"]))
                    sql = "SELECT * FROM Users WHERE Slack_Id = %s"
                    cur.execute(sql, (user["Slack_Id"]))
                    frd_user = cur.fetchone()
                    message = "omg {}/{} just leveled up to {}".format(frd_user["Nick"], summoner["name"], summoner["summonerLevel"])
                    rs.post(channel, message, "LoL shades", slack_token, icon_emoji=':league:')

                if query > 0 and rank_query == 0:
                    user = cur.fetchone()
                    sql = "UPDATE League SET Rank = %s where S_ID = %s"
                    cur.execute(sql, (rank, summoner["id"]))
                    if rank != "Currently Unranked":
                        sql = "SELECT * FROM Users WHERE Slack_Id = %s"
                        cur.execute(sql, (user["Slack_Id"]))
                        frd_user = cur.fetchone()
                        message = "omg {}/{} just changed ranks to {}.".format(frd_user["Nick"], summoner["name"], rank)
                        rs.post(channel, message, "LoL shades", slack_token, icon_emoji=':league:')
                time.sleep(1)

    con.commit()
    if len(bad_names) > 0:
        rs.msg_sean("bad names in monikers for :league: :" + ", ".join(bad_names), slack_token)