instareader.py/reader.py

#
# class to authenticate to google and get items from
# google reader
#
#
import urllib
import urllib2
import re
import xml.dom.minidom
import sys

class GoogleReader:
    ''' class for connecting to google'''
    def __init__(self, login, password):
        self.login = login
        self.password = password
        self.auth_url = "https://www.google.com/accounts/ClientLogin"
        self.re_auth = re.compile("SID=")
        self.sid = self.authenticate(self.login,self.password)
        if self.sid == -1:
            print "Authentication unsuccessful. Exiting."
            sys.exit(-1)

        self.header = {}
        self.header = self.create_header(self.header,self.sid)
        self.items = []
        
    def authenticate(self,login,password):
        ''' method to authenticate to google'''
        parameters = {  'Email' : login,
                        'Passwd' : password,
                        'accountType' : 'HOSTED_OR_GOOGLE',
                        'service' : 'reader',
                        'source' : 'googlereader2instapaper',
                        'continue': 'http://www.google.com'
                     }
        headerdata = urllib.urlencode(parameters)
        try:
            request = urllib2.Request(self.auth_url, headerdata)
            response = urllib2.urlopen(request).read().split("\n")
            for r in response:
                if self.re_auth.match(r):
                    return r.split("=")[1]
        except IOError, e:
            print "Authenticate error %s" % e
            return -1
    
    def create_header(self, header, sid):
        ''' Method to create the header which is used afterwards for authentication '''
        header = {'User-agent' : 'python'}
        header['Cookie'] = 'Name=SID;SID=%s;Domain=.google.com;Path=/;Expires=160000000000' % sid
        return header
        
    def get_starred_items(self,count=60,header=None):
        ''' method to get starred items from google reader 
            returns a list of hashmaps of the form
            item = { 
                    'title' : "foo",
        	        'url' : "bla"
        	        'item' : "baz"
        	        'feed' : "bar"
        	       }
        '''
        if header is None:
            header = self.header

        print count
        params = {
                    'n' : count,
                 }

        starred_base_url = "http://www.google.com/reader/atom/user/-/state/com.google/starred"
        starred_url = starred_base_url + '?' + urllib.urlencode(params)
        try:
            request = urllib2.Request(starred_url, None, header)
            response = urllib2.urlopen(request).read()
            dom = xml.dom.minidom.parseString(response)
            try:
                entries = dom.getElementsByTagName("entry")
                for e in entries:
                    item = { 'title' : "", 'url' : "" }
                    title =  e.getElementsByTagName("title")[0].firstChild.data.encode("utf-8")
                    item['title'] = title
                    links = e.getElementsByTagName("link")
                    item['url'] = links[0].getAttribute("href")
                    item_id = e.getElementsByTagName("id")
                    item['item'] = item_id[0].firstChild.data.encode("utf-8")
                    source = e.getElementsByTagName("source")
                    item['feed'] = source[0].getAttribute("gr:stream-id")
                    self.items.append(item)
                return self.items
            except ExpatError, er:
                print er
                return -2
        except IOError, e:
            print e
            return -2


    def remove_starred_item(self,item,feed,header=None):
        """ Method to remove the starred status from a reader item
        """
        if header is None:
            header = self.header
        # Get edit token
        token = self.get_edit_token()
        post_args = {
                        'client' : 'python',
                        'r'      : 'user/-/state/com.google/starred',
                        'async'  : 'true',
                        's'      : feed,
                        'i'      : item,
                        'T'      : token
                    }
        # Basic edit url
        edit_base_url = "http://www.google.com/reader/api/0/edit-tag"
        edit_params = urllib.urlencode(post_args)
        #edit_url = edit_base_url + '?' + edit_params
        try:
            request = urllib2.Request(edit_base_url, edit_params, header)
            response = urllib2.urlopen(request).read()
            return response
        except IOError, e:
            print "Remove starred item error: %s" % e
            return -3

    def get_edit_token(self,header=None):
        """method for get_edit_token"""
        token_url = "http://www.google.com/reader/api/0/token"
        if header is None:
            header = self.header

        try:
            request = urllib2.Request(token_url, None, header)
            response = urllib2.urlopen(request).read()
            return response
        except IOError, e:
            print "Get token error: %s" % e
            return -4

    def get_subscription_list(self, header):
        ''' Generic Method for getting data from google reader
        '''
        url = 'http://www.google.com/reader/api/0/subscription/list'
        # retrieve data here
        try:
            request = urllib2.Request(url, None, header)
            response = urllib2.urlopen(request).read()
            return response
        except IOError, e:
            print e
            return -5