worker.py

import traceback
import os
import time
import re

from collections import defaultdict
from datetime import datetime, timedelta

#this package seems outdated, but it was super simple to set up
import tweepy

CONSUMER_KEY = os.environ.get('CONSUMER_KEY')
CONSUMER_SECRET = os.environ.get('CONSUMER_SECRET')
ACCESS_KEY = os.environ.get('ACCESS_KEY')
ACCESS_SECRET = os.environ.get('ACCESS_SECRET')

FOOTERS = ["#PULAKYELLING"]

ACCOUNTS = ["pulakm"]

#how often to check for new tweets (careful of twitter rate limiting)
DELAY = 12

#how long until we don't consider a tweet new?
#this is to deal with tweets bizarrely repeating
EXPIRED_TIME = timedelta(minutes=2)

URL_RE = ".+\..+"  # crude, but should handle most of what Pulak tweets

auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)
auth.set_access_token(ACCESS_KEY, ACCESS_SECRET)
api = tweepy.API(auth)


def send_tweet(message):
    final_message = message
    #add in footers (hashtags and users)
    for footer in FOOTERS:
        if len(final_message + " " + footer) <= 140:
            final_message += " " + footer
    try:
        api.update_status(final_message)
        print "Tweeted:"
        print final_message.encode('utf-8')
    except:
        #this will print to heroku logs
        print "Error sending tweet:"
        print final_message.encode('utf-8')
        traceback.print_exc()


def to_upper(message):
    upper_message = []
    for token in message.split(" "):
        if is_link(token):
            print "%s is a link" % token
            upper_message.append(token)
        else:
            upper_message.append(token.upper())
    return " ".join(upper_message)


def is_link(token):
    return True if re.match(URL_RE, token) else False

if __name__ == "__main__":
    last_tweets = defaultdict(str)  # keep track of last tweet from the stream
    while True:
        for user in ACCOUNTS:
            try:
                #get most recent tweet
                tweet = api.user_timeline(user)[0]
                if tweet.text != last_tweets[user]:
                    last_tweets[user] = tweet.text
                    if(datetime.now() - tweet.created_at < EXPIRED_TIME):
                        send_tweet(to_upper(tweet.text))  # Tweet! Tweet!
                    else:
                        print "Got tweet but it was too old for", user
                        print tweet.text.encode('utf-8')
                else:
                    print "Got same tweet for", user
            except IndexError:
                #no tweeets found for this user
                print "Error getting tweet for", user
                traceback.print_exc()
            finally:
                time.sleep(DELAY)  # Avoid Twitter rate limiting