python调用新浪微博API实践
因为最近接触到调用新浪微博开放接口的项目，所以就想试试用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 <= 1: raise ValueError("backoff must be greater than 1") tries = math.floor(tries) if tries < 0: raise ValueError("tries must be 0 or greater") if delay <= 0: raise ValueError("delay must be greater than 0") def deco_retry(f): def f_retry(*args, **kwargs): mtries, mdelay = tries, delay # make mutable rv = f(*args, **kwargs) # first attempt while mtries > 0: if rv == True or type(rv) == str: # Done on success .. return rv mtries -= 1 # consume an attempt time.sleep(mdelay) # wait... mdelay *= backoff # make future wait longer rv = f(*args, **kwargs) # Try again return False # Ran out of tries :-( return f_retry # true decorator -> 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
