python调用新浪微博API实践

plantegg

　　因为最近接触到调用新浪微博开放接口的项目，所以就想试试用python调用微博API。
　　SDK下载地址：http://open.weibo.com/wiki/SDK  代码不多十几K，完全可以看懂。
　　有微博账号可以新建一个APP，然后就可以得到app key和app secret，这个是APP获得OAuth2.0授权所必须的。
　　了解OAuth2可以查看链接新浪微博的说明。 OAuth2授权参数除了需要app key和app secret还需要网站回调地址redirect_uri，并且这个回调地址不允许是局域网的（神马localhost，127.0.0.1好像都不行），这个着实让我着急了半天。我使用API也不是网站调用，于是查了很多。看到有人写可以用这个地址替代，https://api.weibo.com/oauth2/default.html，我试了一下果然可以，对于屌丝来说是个好消息。
　　下面先来个简单的程序，感受一下：
　　设置好以下参数
　　



import sys
import weibo
import webbrowser
APP_KEY = ''
MY_APP_SECRET = ''
REDIRECT_URL = 'https://api.weibo.com/oauth2/default.html'
　　
　　获得微博授权URL，如第2行，用默认浏览器打开后会要求登陆微博，用需要授权的账号登陆，如下图



1 api = weibo.APIClient(app_key=APP_KEY,app_secret=MY_APP_SECRET,redirect_uri=REDIRECT_URL)
2 authorize_url = api.get_authorize_url()
3 print(authorize_url)
4 webbrowser.open_new(authorize_url)

　　登陆后会调转到一个连接https://api.weibo.com/oauth2/default.html?code=92cc6accecfb5b2176adf58f4c
　　关键就是code值，这个是认证的关键。手动输入code值模拟认证



1 request = api.request_access_token(code, REDIRECT_URL)
2 access_token = request.access_token
3 expires_in = request.expires_in
4 api.set_access_token(access_token, expires_in)
5 api.statuses.update.post(status=u'Test OAuth 2.0 Send a Weibo!')
access_token就是获得的token，expires_in是授权的过期时间 （UNIX时间）

用set_access_token保存授权。往下就可以调用微博接口了。测试发了一条微博


　　
　　但是这样的手动输入code方式，不适合程序的调用，是否可以不用打开链接的方式来请求登陆获取授权，经多方查找和参考，将程序改进如下，可以实现自动获取code并保存，方便程序服务调用。


accessWeibo


# -*- coding: utf-8 -*-  
#/usr/bin/env python  
#access to SinaWeibo By sinaweibopy
#实现微博自动登录，token自动生成，保存及更新
#适合于后端服务调用

from weibo import APIClient  
import pymongo  
import sys, os, urllib, urllib2  
from http_helper import *  
from retry import *  
try:  
import json  
except ImportError:  
import simplejson as json  
# setting sys encoding to utf-8  
default_encoding = 'utf-8'  
if sys.getdefaultencoding() != default_encoding:  
reload(sys)  
sys.setdefaultencoding(default_encoding)  
# weibo api访问配置  
APP_KEY = ''      # app key  
APP_SECRET = ''   # app secret  
REDIRECT_URL = 'https://api.weibo.com/oauth2/default.html' # callback url 授权回调页,与OAuth2.0 授权设置的一致  
USERID = ''       # 登陆的微博用户名，必须是OAuth2.0 设置的测试账号                     
USERPASSWD = ''   # 用户密码  

client = APIClient(app_key=APP_KEY, app_secret=APP_SECRET, redirect_uri=REDIRECT_URL)  
def make_access_token():  
#请求access token  
params = urllib.urlencode({
'action':'submit',
'withOfficalFlag':'0',
'ticket':'',
'isLoginSina':'',  
'response_type':'code',
'regCallback':'',
'redirect_uri':REDIRECT_URL,
'client_id':APP_KEY,
'state':'',
'from':'',
'userId':USERID,
'passwd':USERPASSWD,
})  
login_url = 'https://api.weibo.com/oauth2/authorize'  
url = client.get_authorize_url()  
content = urllib2.urlopen(url)  
if content:  
headers = { 'Referer' : url }  
request = urllib2.Request(login_url, params, headers)  
opener = get_opener(False)  
urllib2.install_opener(opener)  
try:  
f = opener.open(request)  
return_redirect_uri = f.url               
except urllib2.HTTPError, e:  
return_redirect_uri = e.geturl()  
# 取到返回的code  
code = return_redirect_uri.split('=')[1]  
#得到token  
token = client.request_access_token(code,REDIRECT_URL)  
save_access_token(token)  
def save_access_token(token):  
#将access token保存到MongoDB数据库
mongoCon=pymongo.Connection(host="127.0.0.1",port=27017)
db= mongoCon.weibo
t={
"access_token":token['access_token'],
"expires_in":str(token['expires_in']),
"date":time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
}
db.token.insert(t,safe=True)  
#Decorator 目的是当调用make_access_token()后再执行一次apply_access_token()
@retry(1)  
def apply_access_token():  
#从MongoDB读取及设置access token
try:  
mongoCon=pymongo.Connection(host="127.0.0.1",port=27017)
db= mongoCon.weibo
if db.token.count()>0:
tokenInfos=db.token.find().sort([("_id",pymongo.DESCENDING)]).limit(1)
else:  
make_access_token()  
return False  
for  tokenInfo in tokenInfos:
access_token=tokenInfo["access_token"]
expires_in=tokenInfo["expires_in"]
try:  
client.set_access_token(access_token, expires_in)  
except StandardError, e:  
if hasattr(e, 'error'):   
if e.error == 'expired_token':  
# token过期重新生成  
                    make_access_token()
return False  
else:  
pass  
except:  
make_access_token()
return False  
return True  
if __name__ == "__main__":  
apply_access_token()  
# 以下为访问微博api的应用逻辑  
# 以发布文字微博接口为例
client.statuses.update.post(status='Test OAuth 2.0 Send a Weibo!')

retry.py


import math
import time
# Retry decorator with exponential backoff
def retry(tries, delay=1, backoff=2):
"""Retries a function or method until it returns True.
delay sets the initial delay, and backoff sets how much the delay should
lengthen after each failure. backoff must be greater than 1, or else it
isn't really a backoff. tries must be at least 0, and delay greater than
0."""
if backoff  decorated function
return deco_retry  # @retry(arg[, ...]) -> true decorator

http_helper.py


# -*- coding: utf-8 -*-
#/usr/bin/env python
import urllib2,cookielib

class SmartRedirectHandler(urllib2.HTTPRedirectHandler):
def http_error_301(cls, req, fp, code, msg, headers):
result = urllib2.HTTPRedirectHandler.http_error_301(cls, req, fp, code, msg, headers)
result.status = code
print headers
return result
def http_error_302(cls, req, fp, code, msg, headers):
result = urllib2.HTTPRedirectHandler.http_error_302(cls, req, fp, code, msg, headers)
result.status = code
print headers
return result
def get_cookie():
cookies = cookielib.CookieJar()
return urllib2.HTTPCookieProcessor(cookies)
def get_opener(proxy=False):
rv=urllib2.build_opener(get_cookie(), SmartRedirectHandler())
rv.addheaders = [('User-agent', 'Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1)')]
return rv