def set_oauth(self,
key,
secret,
acs_token='',
acs_token_secret='',
oauth_verifier=''):
'''
Set OAuth parameters
'''
self._oauth = AppEngineOAuth(key,
secret,
acs_token,
acs_token_secret,
oauth_verifier)
class AppEngineSina(object):
def __init__(self, key , name='', pswd=''):
'''
Note: Some actions require password or OAuth.
'''
self._api_url = 'http://api.t.sina.com.cn'
self._name = name
self._app_key = key
self._oauth = None
self._content = None
self._headers = {}
if pswd != '':
auth = base64.encodestring(name + ':' + pswd)[:-1]
self._headers['Authorization'] = 'Basic ' + auth
def getContent(self):
'''
return the content
'''
return self._content
# ========= Status Methods ===========
def show_status(self, id):
'''
Returns a single status, specified by the id parameter below.
The status's author will be returned inline.
Sucess => Retrun 200 status and set response to self.content/
Fialed => Return other HTTP status
'''
status =self._get('/statuses/show/id.json',
{'id': id,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def update_status(self,
status,
lat=0,
long=0):
'''
Post a msg
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
status=urllib2.quote(status)
return self._post('/statuses/update.json',
{'status': status,
'lat': lat,
'long': long,
'source':self._app_key})
def del_status(self,id):
'''
Destroy a msg
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/statuses/destroy/id.json',
{'id': id,
'source':self._app_key})
def rt_status(self,
id,
status=''):
'''
Repost a msg
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
status=urllib2.quote(status)
return self._post('/statuses/repost.json',
{'id': id,
'status': status,
'source':self._app_key})
def update_comment(self,
id,
comment,
cid=0):
'''
Post a comment
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
comment=urllib2.quote(comment)
return self._post('/statuses/comment.json',
{'id': id,
'comment': comment,
'cid':cid,
'source':self._app_key})
def del_comment(self,id):
'''
Destroy a comment
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/statuses/comment_destroy/id.json',
{'id': id,
'source':self._app_key})
def reply_comment(self,
id,
cid,
comment):
'''
Reply a comment
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
comment=urllib2.quote(comment)
return self._post('/statuses/reply.json',
{'id': id,
'cid':cid,
'comment':comment,
'source':self._app_key})
# ========= Timeline Methods =========
def public_timeline(self):
'''
get and save public Timeline
Sucess => Retrun 200 status and set response to self.content/
Fialed => Return error HTTP status
'''
status = self._get('/statuses/public_timeline.json', {'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def friends_timeline(self,
since_id=0,
max_id=0,
count=20,
page=1):
'''
get and save friends Timeline
Sucess => Retrun 200 status and set response to self.content/
Fialed => Return error HTTP status
'''
status = self._get('/statuses/friends_timeline.json',
{'since_id':since_id,
'max_id':max_id,
'count':count,
'page':page,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def user_timeline(self,
id,
screen_name='',
user_id=0,
since_id=0,
max_id=0,
count=20,
page=1):
'''
get and save user Timeline
Sucess => Retrun 200 status and set response to self.content/
Fialed => Return error HTTP status
'''
status = self._get('/statuses/user_timeline.json',
{'id':id,
'user_id':user_id,
'screen_name':screen_name,
'since_id':since_id,
'max_id':max_id,
'count':count,
'page':page,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def mentions(self,
since_id=0,
max_id=0,
count=20,
page=1):
'''
get and save Mention timeline
Sucess => Retrun 200 status and set response to self.content/
Fialed => Return error HTTP status
'''
status = self._get('/statuses/mentions.json',
{'since_id':since_id,
'max_id':max_id,
'count':count,
'page':page,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def comments(self,
since_id=0,
max_id=0,
count=20,
page=1):
'''
get and save comments timeline
Sucess => Retrun 200 status and set response to self.content/
Fialed => Return error HTTP status
'''
status = self._get('/statuses/comments_timeline.json',
{'since_id':since_id,
'max_id':max_id,
'count':count,
'page':page,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def comments_by_me(self,
since_id=0,
max_id=0,
count=20,
page=1):
'''
get and save comments timeline by me
Sucess => Retrun 200 status and set response to self.content/
Fialed => Return error HTTP status
'''
status = self._get('/statuses/comments_by_me.json',
{'since_id':since_id,
'max_id':max_id,
'count':count,
'page':page,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def comments_for_status(self,
id,
count=20,
page=1):
'''
get and save comments for status
Sucess => Retrun 200 status and set response to self.content/
Fialed => Return error HTTP status
'''
status = self._get('/statuses/comments.json',
{'id':id,
'count':count,
'page':page,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
# ========= User Methods =========
def show_user(self,
id,
user_id=0,
screen_name=''):
'''
Get and save user info
Sucess => Retrun 200 status and set content to self._content
Fialed => Return error HTTP status
'''
status = self._get('/users/show.json',
{'id':id,
'user_id':user_id,
'screen_name':screen_name,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def friends(self,
id,
user_id=0,
screen_name='',
cursor=-1,
count=20):
'''
Get and save friends from user
Sucess => Retrun 200 status and set content to self._content
Fialed => Return error HTTP status
'''
status = self._get('/statuses/friends.json',
{'id':id,
'user_id':user_id,
'screen_name':screen_name,
'cursor':cursor,
'count':count,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def followers(self,
id,
user_id=0,
screen_name='',
cursor=-1,
count=20):
'''
Get and save followers from user
Sucess => Retrun 200 status and set content to self._content
Fialed => Return error HTTP status
'''
status = self._get('/statuses/followers.json',
{'id':id,
'user_id':user_id,
'screen_name':screen_name,
'cursor':cursor,
'count':count,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
# ========= Direct Message Methods =========
def direct_messages(self,
since_id=0,
max_id=0,
count=20,
page=1):
'''
get and save Direct Messages timeline
Sucess => Retrun 200 status and set content to self._content
Fialed => Return other HTTP status
'''
status = self._get('/direct_messages.json',
{'since_id':since_id,
'max_id':max_id,
'count':count,
'page':page,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def send_direct_messages(self,
since_id=0,
max_id=0,
count=20,
page=1):
'''
get and save direct messages timeline you have send
Sucess => Retrun 200 status and set content to self._content
Fialed => Return other HTTP status
'''
status = self._get('/direct_messages/sent.json',
{'since_id':since_id,
'max_id':max_id,
'count':count,
'page':page,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def new_direct_messages(self,
screen_name,
text,
user_id=0):
'''
Sent a Direct Messages
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
text=urllib2.quote(text)
return self._post('/direct_messages/new.json',
{'user_id':user_id,
'screen_name':screen_name,
'text':text,
'source':self._app_key})
def destroy_direct_messages(self,id):
'''
Destroy a Direct Messages
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/direct_messages/destroy/id.json',
{'id':id,
'source':self._app_key})
# ========= Friendships methods =========
def follow_user(self,
id,
user_id=0,
screen_name=''):
'''
Follow user
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/friendships/create/id.json',
{'id':id,
'user_id':user_id,
'screen_name':screen_name,
'source':self._app_key})
def unfollow_user(self,
id,
user_id=0,
screen_name=''):
'''
Unfollow user
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/friendships/destroy/id.json',
{'id':id,
'user_id':user_id,
'screen_name':screen_name,
'source':self._app_key})
def isfollow_user(self,
target_screen_name,
target_id=0,
source_screen_name='',
source_id=0):
'''
the existence of friendship between two users
Sucess => Retrun 200 status and set content to self._content
Fialed => Return other HTTP status
'''
param={'target_id':target_id,
'target_screen_name':target_screen_name,
'source':self._app_key}
if(source_id<>0 or source_screen_name<>''):
param['source_id']=source_id
param['source_screen_name']=source_screen_name
status = self._get('/friendships/show.json',param)
if(status==200):
self._content=self.last_response.content
return status
# ========= Social Graph methods ===========
def friends_ids(self,
id,
user_id=0,
screen_name='',
cursor=-1,
count=20):
'''
Get and save a list of id from the target user's frineds
Sucess => Retrun 200 status and set content to self._content
Fialed => Return other HTTP status
'''
status = self._get('/friends/ids.json',
{'id':id,
'user_id':user_id,
'screen_name':screen_name,
'cursor':cursor,
'count':count,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def followers_ids(self,
id,
user_id=0,
screen_name='',
cursor=-1,
count=20):
'''
Get and save a list of id from the target user's followers
Sucess => Retrun 200 status and set content to self._content
Fialed => Return other HTTP status
'''
status = self._get('/followers/ids.json',
{'id':id,
'user_id':user_id,
'screen_name':screen_name,
'cursor':cursor,
'count':count,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
# ========= Account methods ==========
def verify_credentials(self):
'''
Verify Credentials and save info
Sucess => Retrun 200 status and set content to self._content
Fialed => Return other HTTP status
'''
status = self._get('/account/verify_credentials.json', {'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def rate_limit_status(self):
'''
Get and save rate limit status
Sucess => Retrun 200 status and set content to self._content
Fialed => Return other HTTP status
'''
status = self._get('/account/rate_limit_status.json', {'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def end_session(self):
'''
End session of the user
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/account/end_session.json', {'source':self._app_key})
def register(self,
nick,
gender,
password,
email):
'''
Register a new account
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/account/register.json',
{'nick':nick,
'gender':gender,
'password':password,
'email':email,
'source':self._app_key})
def update_profile(self, param):
'''
param is a dict with the following key:
name, gender, province, city, description
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
param['source']=self._app_key
return self._post('/account/update_profile.json',param)
# ========= Favorites methods =========
def favorites(self, page=1):
'''
Get and save favorites timeline
Sucess => Retrun 200 status and set content to self._content
Fialed => Return other HTTP status
'''
status = self._get('/favorites.json',
{'page':page,
'source':self._app_key})
if(status==200):
self._content=self.last_response.content
return status
def create_favorite(self,id):
'''
Create favorite
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/favorites/create.json',
{'id':id,
'source':self._app_key})
def destroy_favorite(self,id):
'''
Destroy favorite
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/favorites/destroy/id.json',
{'id':id,
'source':self._app_key})
# ========= OAuth methods ==========
def set_oauth(self,
key,
secret,
acs_token='',
acs_token_secret='',
oauth_verifier=''):
'''
Set OAuth parameters
'''
self._oauth = AppEngineOAuth(key,
secret,
acs_token,
acs_token_secret,
oauth_verifier)
def request_token(self):
'''
Get request token, request token secret
'''
dic = self._oauth.request_token(self._api_url + '/oauth/request_token')
return dic
def set_authorize(self,
req_token,
userId,
pw):
'''
Verifier request token
Return request_token, oauth_verifier.
'''
res = self._oauth.set_authorize(self._api_url + '/oauth/authorize',
req_token,
userId,
pw)
return res
def access_tokens(self,
req_token,
req_token_secret,
oauth_verifier):
'''
Exchange request token for access token
'''
return self._oauth.access_tokens(self._api_url + '/oauth/access_token',
req_token,
req_token_secret,
oauth_verifier)
# ========= Private methods =========
def _post(self, path, params):
url = self._api_url + path
if self._oauth != None:
params = self._oauth.get_oauth_params(url, params, 'POST')
res = urlfetch.fetch(url=url,
payload=urllib.urlencode(params),
method='POST',
headers=self._headers)
self.last_response = res
return res.status_code
def _get(self, path, params):
url = self._api_url + path
if self._oauth != None:
params = self._oauth.get_oauth_params(url, params, 'GET')
url += '?' + urllib.urlencode(params)
res = urlfetch.fetch(url=url, method='GET', headers=self._headers)
self.last_response = res
return res.status_code
def set_oauth(self, key, secret, acs_token="", acs_token_secret=""):
"""
Set OAuth parameters
"""
self._oauth = AppEngineOAuth(key, secret, acs_token, acs_token_secret)
class AppEngineTwitter(object):
def __init__(self, tw_name='', tw_pswd=''):
'''
Note: Some actions require password or OAuth.
'''
self._oauth_api_url = 'https://twitter.com'
self._api_url='http://api.twitter.com/1'
self._search_url = 'http://search.twitter.com'
self.tw_name = tw_name
self._oauth = None
self._headers = {}
if tw_pswd != '':
auth = base64.encodestring(tw_name + ':' + tw_pswd)[:-1]
self._headers['Authorization'] = 'Basic ' + auth
# ========= Status Methods ===========
def update(self, msg, rid='0'):
'''
Post a tweet
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
if(rid=='0'):
return self._post('/statuses/update.json', {'status': msg})
else:
return self._post('/statuses/update.json', {'status': msg, 'in_reply_to_status_id': long(rid)})
def delMsg(self, msgid):
'''
delete a msg
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
url="/statuses/destroy/%s.json"%(msgid)
return self._post(url, {})
def getContent(self):
'''
return the content
'''
return self.content
# ========= Timeline Methods =========
def user_timeline(self, page=0,user_name=''):
'''
get user Timeline
Sucess => Return contents of user timeline /
Fialed => Return error HTTP status
'''
param={}
if(user_name<>''):
param['screen_name']=user_name
if(page==0):
param['count']=10
else:
param['page']=page
status = self._get('/statuses/user_timeline.json', param)
if(status==200):
self.content=self.last_response.content
return status
def home_timeline(self, msgid=0, page=0):
'''
save and get home timeline
Sucess => Return contents of home timeline
Failed => Return other HTTP status
'''
if(msgid==0):
if(page==0):
status = self._get('/statuses/home_timeline.json', {"count":10})
elif(page==-1):
status = self._get('/statuses/home_timeline.json', {"count":3})
else:
status = self._get('/statuses/home_timeline.json', {"page":page})
else:
status = self._get('/statuses/home_timeline.json', {"since_id":msgid})
if status == 200:
self.content=self.last_response.content
return status
def friends_timeline(self , msgid=0):
'''
save and get friends timeline
Sucess => Return contents of friends timeline
Failed => Return other HTTP status
'''
if(msgid==0):
count=10
status = self._get('/statuses/friends_timeline.json', {"count":count})
elif(msgid==1):
count=20
status = self._get('/statuses/friends_timeline.json', {"count":count})
else:
status = self._get('/statuses/friends_timeline.json', {"since_id":msgid})
if status == 200:
self.content=self.last_response.content
return status
def mentions(self, msgid=0,page=0):
'''
get and save Mention messages
Sucess => Return content of mentions
Fialed => Return other HTTP status
'''
if(msgid==0):
if(page==0):
status = self._get('/statuses/mentions.json', {"count":10})
else:
status = self._get('/statuses/mentions.json', {"page":page})
else:
status = self._get('/statuses/mentions.json', {"page":page})
if(status==200):
self.content=self.last_response.content
return status
def list_timeline(self, params, msgid=0,page=0):
'''
get and save list messages
Sucess => Return content of list timeline
Fialed => Return other HTTP status
'''
list_url='/lists/statuses.json'
if(msgid==0):
if(page==0):
params['per_page']=10
elif(page==-1):
params['per_page']=3
else:
params['page']=page
else:
params['since_id']=msgid
status = self._get(list_url, params)
if(status==200):
self.content=self.last_response.content
return status
def getRt2Me(self,page=0):
if(page==0):
count=10
status = self._get('/statuses/retweeted_to_me.json', {"count":count})
else:
status = self._get('/statuses/retweeted_to_me.json', {"page":page})
if status == 200:
self.content=self.last_response.content
return status
def getListInfo(self,target_name,list_id):
url='/%s/lists/%s.json'%(target_name,list_id)
status = self._get(url,{})
if status == 200:
self.content=self.last_response.content
return status
# ========= Friendship Methods =========
def follow(self, target_name):
'''
follow a user
Sucess => Return 200 /
Fialed => Return other HTTP status
'''
return self._post('/friendships/create.json', {'screen_name': target_name})
def unfollow(self, target_name):
'''
unfollow a user
Sucess => Return 200 /
Fialed => Return other HTTP status
'''
return self._post('/friendships/destroy.json', {'screen_name': target_name})
def is_following(self, target_name):
'''
Yes => Return True / No => Return False /
Fialed => Return HTTP status except 200
'''
if self.tw_name == '':
# With OAuth, screen_name is not required.
self.verify()
user_info = simplejson.loads(self.last_response.content)
self.tw_name = user_info['screen_name']
status = self._get('/friendships/exists.json',
{'user_a':target_name , 'user_b': self.tw_name})
if status == 200:
return (self.last_response.content == 'true')
else:
return status
def block(self, target_name):
'''
block a user
Sucess => Return 200 /
Fialed => Return other HTTP status
'''
return self._post('/blocks/create.json', {'screen_name': target_name})
# ========= Direct Message Methods =========
def getDMessage(self,dmid=0,page=0):
'''
get and save Direct Messages
Sucess => Return content of Direct Message
Fialed => Return other HTTP status
'''
if(dmid==0):
if(page==0):
status = self._get('/direct_messages.json', {"count":10})
else:
status = self._get('/direct_messages.json', {"page":page})
elif(dmid==1):
status = self._get('/direct_messages.json', {"count":1})
else:
status = self._get('/direct_messages.json', {"since_id":dmid})
if(status==200):
self.content=self.last_response.content
return status
def sentDMessage(self, name ,text):
'''
Sent a Direct Messages
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
return self._post('/direct_messages/new.json', {'screen_name': name,'text': text})
# ========= Favorite Methods ========
def addFav(self, msgid):
'''
add a msg as fav
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
url="/favorites/create/%s.json"%(msgid)
status=self._post(url, {})
return status
def unFav(self, msgid):
'''
destroy a favorite tweets
Sucess => Retrun 200
Fialed => Return other HTTP status
'''
url="/favorites/destroy/%s.json"%(msgid)
status=self._post(url, {})
return status
def getFav(self , page=0):
'''
get and save your fav messages
Sucess => Return 200 / Fialed => Return other HTTP status
'''
if(page==0):
status = self._get('/favorites.json', {"count":10})
else:
status = self._get('/favorites.json', {"page":page})
if(status==200):
self.content=self.last_response.content
return status
def rtMsg(self,msgid):
'''
get and save the messages
Sucess => Return 200
Fialed => Return other HTTP status
'''
url="/statuses/show/%s.json"%(msgid)
status = self._get(url, {})
if(status==200):
self.content=self.last_response.content
return status
def newRTMsg(self,msgid):
'''
get and save the messages
Sucess => Return 200
Fialed => Return other HTTP status
'''
url="/statuses/retweet/%s.json"%(msgid)
status = self._post(url, {})
if(status==200):
self.content=self.last_response.content
return status
def getRtMsg(self):
'''
return the message
'''
return self.rtmsg
# ========= User Methods =========
def getName(self):
self.verify()
user_info = simplejson.loads(self.last_response.content)
return user_info['screen_name']
def getUserInfo(self,name):
'''
return the info of the user
'''
status = self._get('/users/show.json', {"screen_name":name})
if(status==200):
self.content=self.last_response.content
return status
def verify(self):
'''
Verify user_name and password, and get user info
Sucess => Return 200 / Fialed => Return other HTTP status
'''
status = self._get('/account/verify_credentials.json', {})
if(status==200):
self.content=self.last_response.content
return status
# ========= List Methods =========
def getLists(self,name,cursor):
url="/%s/lists.json"%(name)
status = self._get(url, {'cursor':cursor})
if(status==200):
self.content=self.last_response.content
return status
def getSubscriptions(self,name,cursor):
url="/%s/lists/subscriptions.json"%(name)
status = self._get(url, {'cursor':cursor})
if(status==200):
self.content=self.last_response.content
return status
def getMemberships(self,name,cursor):
url="/%s/lists/memberships.json"%(name)
status = self._get(url, {'cursor':cursor})
if(status==200):
self.content=self.last_response.content
return status
def searchTweet(self,word,page):
params={}
params['q'] = wordhandler.getString(unicode(word.encode("utf-8"),'utf-8')).encode("utf-8")
if(page<>0):
params['page'] = page
status = self._search('/search.json', params)
if(status==200):
self.content=self.last_response.content
return status
# ========= OAuth methods ==========
# (See http://0-oo.net/sbox/python-box/appengine-oauth )
def set_oauth(self, key, secret, acs_token='', acs_token_secret=''):
'''
Set OAuth parameters
'''
self._oauth = AppEngineOAuth(key, secret, acs_token, acs_token_secret)
def prepare_oauth_login(self):
'''
Get request token, request token secret and login URL
'''
dic = self._oauth.prepare_login(self._oauth_api_url + '/oauth/request_token/')
dic['url'] = self._oauth_api_url + '/oauth/authorize?' + dic['params']
return dic
def exchange_oauth_tokens(self, req_token, req_token_secret):
'''
Exchange request token for access token
'''
return self._oauth.exchange_tokens(self._oauth_api_url + '/oauth/access_token/',
req_token,
req_token_secret)
# ========= Private methods =========
def _post(self, path, params):
url = self._api_url + path
if self._oauth != None:
params = self._oauth.get_oauth_params(url, params, 'POST')
#self.smessage=type(params["status"])
#smessage=urllib.urlencode(params)
res = urlfetch.fetch(url=url,
payload=urllib.urlencode(params),
method='POST',
headers=self._headers)
self.last_response = res
return res.status_code
def _get(self, path, params):
url = self._api_url + path
if self._oauth != None:
params = self._oauth.get_oauth_params(url, params, 'GET')
url += '?' + urllib.urlencode(params)
res = urlfetch.fetch(url=url, method='GET', headers=self._headers)
self.last_response = res
return res.status_code
def _search(self, path, params):
'''
FYI http://apiwiki.twitter.com/Rate-limiting (Especially 503 error)
'''
url = self._search_url + path + '?' + urllib.urlencode(params)
res = urlfetch.fetch(url=url, method='GET')
self.last_response = res
return res.status_code
if res.status_code == 200:
return simplejson.loads(res.content)['results']
elif res.status_code == 503:
err_msg = 'Rate Limiting: Retry After ' + res.headers['Retry-After']
else:
err_msg = 'Error: HTTP Status is ' + str(res.status_code)
raise Exception('Twitter Search API ' + err_msg)
class AppEngineTwitter(object):
def __init__(self, tw_name="", tw_pswd=""):
"""
Note: Some actions require password or OAuth.
"""
self._api_url = "https://twitter.com"
# self._api_url='http://api.twitter.com/1'
self._search_url = "http://www.hellocode.us/twip/"
self.tw_name = tw_name
self._oauth = None
self._headers = {}
if tw_pswd != "":
auth = base64.encodestring(tw_name + ":" + tw_pswd)[:-1]
self._headers["Authorization"] = "Basic " + auth
# ========= Status Methods ===========
def update(self, msg, rid="0"):
"""
Post a tweet
Sucess => Retrun 200
Fialed => Return other HTTP status
"""
if rid == "0":
return self._post("/statuses/update.json", {"status": msg})
else:
return self._post("/statuses/update.json", {"status": msg, "in_reply_to_status_id": long(rid)})
def delMsg(self, msgid):
"""
delete a msg
Sucess => Retrun 200
Fialed => Return other HTTP status
"""
url = "/statuses/destroy/%s.json" % (msgid)
return self._post(url, {})
def getContent(self):
"""
return the content
"""
return self.content
# ========= Timeline Methods =========
def user_timeline(self, page=0, user_name=""):
"""
get user Timeline
Sucess => Return contents of user timeline /
Fialed => Return error HTTP status
"""
param = {}
if user_name <> "":
param["screen_name"] = user_name
if page == 0:
param["count"] = 10
else:
param["page"] = page
status = self._get("/statuses/user_timeline.json", param)
if status == 200:
self.content = self.last_response.content
return status
def home_timeline(self, msgid=0, page=0):
"""
save and get home timeline
Sucess => Return contents of home timeline
Failed => Return other HTTP status
"""
if msgid == 0:
if page == 0:
status = self._get("/statuses/home_timeline.json", {"count": 10})
elif page == -1:
status = self._get("/statuses/home_timeline.json", {"count": 3})
else:
status = self._get("/statuses/home_timeline.json", {"page": page})
else:
status = self._get("/statuses/home_timeline.json", {"since_id": msgid})
if status == 200:
self.content = self.last_response.content
return status
def friends_timeline(self, msgid=0):
"""
save and get friends timeline
Sucess => Return contents of friends timeline
Failed => Return other HTTP status
"""
if msgid == 0:
count = 10
status = self._get("/statuses/friends_timeline.json", {"count": count})
elif msgid == 1:
count = 20
status = self._get("/statuses/friends_timeline.json", {"count": count})
else:
status = self._get("/statuses/friends_timeline.json", {"since_id": msgid})
if status == 200:
self.content = self.last_response.content
return status
def mentions(self, msgid=0, page=0):
"""
get and save Mention messages
Sucess => Return content of mentions
Fialed => Return other HTTP status
"""
if msgid == 0:
if page == 0:
status = self._get("/statuses/mentions.json", {"count": 10})
else:
status = self._get("/statuses/mentions.json", {"page": page})
else:
status = self._get("/statuses/mentions.json", {"page": page})
if status == 200:
self.content = self.last_response.content
return status
def list_timeline(self, list, msgid=0, page=0):
"""
get and save list messages
Sucess => Return content of list timeline
Fialed => Return other HTTP status
"""
list_url = "/" + list + "/statuses.json"
if msgid == 0:
if page == 0:
status = self._get(list_url, {"per_page": 10})
elif page == -1:
status = self._get(list_url, {"per_page": 3})
else:
status = self._get(list_url, {"page": page})
else:
status = self._get(list_url, {"since_id": msgid})
if status == 200:
self.content = self.last_response.content
return status
def getRt2Me(self, page=0):
if page == 0:
count = 10
status = self._get("/statuses/retweeted_to_me.json", {"count": count})
else:
status = self._get("/statuses/retweeted_to_me.json", {"page": page})
if status == 200:
self.content = self.last_response.content
return status
def getListInfo(self, target_name, list_id):
url = "/%s/lists/%s.json" % (target_name, list_id)
status = self._get(url, {})
if status == 200:
self.content = self.last_response.content
return status
# ========= Friendship Methods =========
def follow(self, target_name):
"""
follow a user
Sucess => Return 200 /
Fialed => Return other HTTP status
"""
return self._post("/friendships/create.json", {"screen_name": target_name})
def unfollow(self, target_name):
"""
unfollow a user
Sucess => Return 200 /
Fialed => Return other HTTP status
"""
return self._post("/friendships/destroy.json", {"screen_name": target_name})
def is_following(self, target_name):
"""
Yes => Return True / No => Return False /
Fialed => Return HTTP status except 200
"""
if self.tw_name == "":
# With OAuth, screen_name is not required.
self.verify()
user_info = simplejson.loads(self.last_response.content)
self.tw_name = user_info["screen_name"]
status = self._get("/friendships/exists.json", {"user_a": target_name, "user_b": self.tw_name})
if status == 200:
return self.last_response.content == "true"
else:
return status
def block(self, target_name):
"""
block a user
Sucess => Return 200 /
Fialed => Return other HTTP status
"""
return self._post("/blocks/create.json", {"screen_name": target_name})
# ========= Direct Message Methods =========
def getDMessage(self, dmid=0, page=0):
"""
get and save Direct Messages
Sucess => Return content of Direct Message
Fialed => Return other HTTP status
"""
if dmid == 0:
if page == 0:
status = self._get("/direct_messages.json", {"count": 10})
else:
status = self._get("/direct_messages.json", {"page": page})
elif dmid == 1:
status = self._get("/direct_messages.json", {"count": 1})
else:
status = self._get("/direct_messages.json", {"since_id": dmid})
if status == 200:
self.content = self.last_response.content
return status
def sentDMessage(self, name, text):
"""
Sent a Direct Messages
Sucess => Retrun 200
Fialed => Return other HTTP status
"""
return self._post("/direct_messages/new.json", {"screen_name": name, "text": text})
# ========= Favorite Methods ========
def addFav(self, msgid):
"""
add a msg as fav
Sucess => Retrun 200
Fialed => Return other HTTP status
"""
url = "/favorites/create/%s.json" % (msgid)
status = self._post(url, {})
return status
def unFav(self, msgid):
"""
destroy a favorite tweets
Sucess => Retrun 200
Fialed => Return other HTTP status
"""
url = "/favorites/destroy/%s.json" % (msgid)
status = self._post(url, {})
return status
def getFav(self, page=0):
"""
get and save your fav messages
Sucess => Return 200 / Fialed => Return other HTTP status
"""
if page == 0:
status = self._get("/favorites.json", {"count": 10})
else:
status = self._get("/favorites.json", {"page": page})
if status == 200:
self.content = self.last_response.content
return status
def rtMsg(self, msgid):
"""
get and save the messages
Sucess => Return 200
Fialed => Return other HTTP status
"""
url = "/statuses/show/%s.json" % (msgid)
status = self._get(url, {})
if status == 200:
self.content = self.last_response.content
return status
def newRTMsg(self, msgid):
"""
get and save the messages
Sucess => Return 200
Fialed => Return other HTTP status
"""
url = "/statuses/retweet/%s.json" % (msgid)
status = self._post(url, {})
if status == 200:
self.content = self.last_response.content
return status
def getRtMsg(self):
"""
return the message
"""
return self.rtmsg
# ========= User Methods =========
def getName(self):
self.verify()
user_info = simplejson.loads(self.last_response.content)
return user_info["screen_name"]
def getUserInfo(self, name):
"""
return the info of the user
"""
status = self._get("/users/show.json", {"screen_name": name})
if status == 200:
self.content = self.last_response.content
return status
def verify(self):
"""
Verify user_name and password, and get user info
Sucess => Return 200 / Fialed => Return other HTTP status
"""
status = self._get("/account/verify_credentials.json", {})
if status == 200:
self.content = self.last_response.content
return status
# ========= List Methods =========
def getLists(self, name, cursor):
url = "/%s/lists.json" % (name)
status = self._get(url, {"cursor": cursor})
if status == 200:
self.content = self.last_response.content
return status
def getSubscriptions(self, name, cursor):
url = "/%s/lists/subscriptions.json" % (name)
status = self._get(url, {"cursor": cursor})
if status == 200:
self.content = self.last_response.content
return status
def getMemberships(self, name, cursor):
url = "/%s/lists/memberships.json" % (name)
status = self._get(url, {"cursor": cursor})
if status == 200:
self.content = self.last_response.content
return status
def searchTweet(self, word, page):
params = {}
params["q"] = wordhandler.getString(unicode(word.encode("utf-8"), "utf-8")).encode("utf-8")
if page <> 0:
params["page"] = page
status = self._search("/search.json", params)
if status == 200:
self.content = self.last_response.content
return status
# ========= OAuth methods ==========
# (See http://0-oo.net/sbox/python-box/appengine-oauth )
def set_oauth(self, key, secret, acs_token="", acs_token_secret=""):
"""
Set OAuth parameters
"""
self._oauth = AppEngineOAuth(key, secret, acs_token, acs_token_secret)
def prepare_oauth_login(self):
"""
Get request token, request token secret and login URL
"""
dic = self._oauth.prepare_login(self._api_url + "/oauth/request_token/")
dic["url"] = self._api_url + "/oauth/authorize?" + dic["params"]
return dic
def exchange_oauth_tokens(self, req_token, req_token_secret):
"""
Exchange request token for access token
"""
return self._oauth.exchange_tokens(self._api_url + "/oauth/access_token/", req_token, req_token_secret)
# ========= Private methods =========
def _post(self, path, params):
url = self._api_url + path
if self._oauth != None:
params = self._oauth.get_oauth_params(url, params, "POST")
# self.smessage=type(params["status"])
# smessage=urllib.urlencode(params)
res = urlfetch.fetch(url=url, payload=urllib.urlencode(params), method="POST", headers=self._headers)
self.last_response = res
return res.status_code
def _get(self, path, params):
url = self._api_url + path
if self._oauth != None:
params = self._oauth.get_oauth_params(url, params, "GET")
url += "?" + urllib.urlencode(params)
res = urlfetch.fetch(url=url, method="GET", headers=self._headers)
self.last_response = res
return res.status_code
def _search(self, path, params):
"""
FYI http://apiwiki.twitter.com/Rate-limiting (Especially 503 error)
"""
url = self._search_url + path + "?" + urllib.urlencode(params)
res = urlfetch.fetch(url=url, method="GET")
self.last_response = res
return res.status_code
if res.status_code == 200:
return simplejson.loads(res.content)["results"]
elif res.status_code == 503:
err_msg = "Rate Limiting: Retry After " + res.headers["Retry-After"]
else:
err_msg = "Error: HTTP Status is " + str(res.status_code)
raise Exception("Twitter Search API " + err_msg)
