RobBot.py

from twitter.api import Twitter, TwitterError
from twitter.oauth import OAuth, write_token_file, read_token_file
from twitter.oauth_dance import oauth_dance

import os
import time
import sys
import json
import hmac
import urllib
import pycurl
import hashlib
import StringIO
import re
import logging

def format_response_to_json(data):
    match = re.compile("\{(.+)\}$").search(data)
    return json.loads("{"+match.group(1)+"}")

if __name__ == '__main__':
    
    #Configure Logger
    logging.basicConfig(filename='RobBot.log',level=logging.DEBUG,format='%(asctime)s %(message)s', datefmt='%m/%d/%Y %I:%M:%S %p')
    
    
    #Load keystore file from the HOME directory
    keystore_file = os.environ.get('HOME', '') + os.sep + '.rob_bot_keystore'
    
    #Read the API keys and other settings into variables
    OAUTH_TOKEN,OAUTH_TOKEN_SECRET,TWITTER_CONSUMER_KEY,TWITTER_CONSUMER_SECRET,AI_SECRET_KEY,AI_API_KEY,TWITTER_NAME,CHAT_BOT_ID,SNOOZE_TIME= file.readlines(open(keystore_file))
    
    if float(SNOOZE_TIME) < float(10):
        e= Exception("The SNOOZE_TIME IN YOUR keystore file has to be at least 10!!!") 
        logging.warning("Error: %s" % e)
        raise e
    
    twitter = Twitter(domain='search.twitter.com')
    twitter.uriparts=()

    last_id_replied = ''

    poster = Twitter(
        auth=OAuth(
            OAUTH_TOKEN.rstrip(), OAUTH_TOKEN_SECRET.rstrip(), TWITTER_CONSUMER_KEY.rstrip(), TWITTER_CONSUMER_SECRET.rstrip()),
        secure=True,
        api_version='1',
        domain='api.twitter.com')

    #Infinite loop to monitor the twitter account
    while True:
        try:
            results = twitter.search(q=TWITTER_NAME, since_id=last_id_replied)['results']
        
            if not results:
                logging.info("No tweets to "+TWITTER_NAME+" this time...")

            for result in results:
                question = result['text'].replace(TWITTER_NAME, '')
                asker = result['from_user']
                id = str(result['id'])
               
                logging.info(" Person who asked: " + asker + " and his question is: " + question)
                
                #Building json needed for the chatbot API
                messageJSON =json.dumps({
                                        'message': {
                                            'message': "'"+question.strip()+"'",
                                            'chatBotID': CHAT_BOT_ID,
                                            'timestamp': time.time()
                                        },
                                        'user': 
                                        {
                                            'externalID': "'"+asker+"'"
                                        }
                                        })
                #Hash needed for the chatbot API
                hash = hmac.new(AI_SECRET_KEY.rstrip(),messageJSON, hashlib.sha256).hexdigest()
                messageJSONencoded = urllib.quote_plus(messageJSON)
                url = "?apiKey="+AI_API_KEY.rstrip()+"&hash="+hash+"&message="+messageJSONencoded
                
                #Connect to the chatbot API and get response 
                conn = pycurl.Curl()
                conn.setopt(pycurl.TIMEOUT, 3)
                conn.setopt(pycurl.URL, 'http://www.personalityforge.com/api/chat/'+url)
                b = StringIO.StringIO()
                conn.setopt(pycurl.WRITEFUNCTION, b.write)
                conn.perform()
                conn.close()
                data = format_response_to_json(b.getvalue())
                msg = '@%s %s (%s)' % (asker, data['message']['message'], id[-4:])
                logging.info("Response: ")
                last_id_replied = id
                poster.statuses.update(status=msg)
        except Exception,e:
            logging.warning("Error: %s" % e)
        print "Running...(See log for more info)"
        logging.info("Snooze for "+SNOOZE_TIME+" seconds")
        time.sleep(float(SNOOZE_TIME))