Twitter API を使ってみる

	import urllib
	import hmac, sha
	import time, random
	
	# この関数の返り値を Authorization ヘッダーに設定する
	def getOAuthHeader(url,method,params):
		# 事前に確保した consumerKey, consumerSecret, accessToken, accessTokenSecret が設定されていること
		
		oAuthParams = {
			"oauth_consumer_key"     : consumerKey,
			"oauth_signature_method" : "HMAC-SHA1",
			"oauth_timestamp"        : str(int(time.time())),
			"oauth_nonce"            : str(random.getrandbits(64)),
			"oauth_version"          : "1.0",
			"oauth_token"            : accessToken
		}
		
		allParams=params.copy()
		allParams.update(oAuthParams)
		allParamList=[]
		for key in sorted(allParams):
			allParamList.append(key+"="+allParams[key])
		
		msg=method+"&"+urllib.quote(url,"")+"&"+urllib.quote("&".join(allParamList),"~")
		h=hmac.new(consumerSecret+"&"+accessTokenSecret,msg,sha)
		oAuthParams["oauth_signature"]=h.digest().encode("base64").strip()
		
		oAuthParamsList=[]
		for key in oAuthParams:
			oAuthParamsList.append(key+"=\""+urllib.quote(oAuthParams[key])+"\"")
		
		return "OAuth "+(", ".join(oAuthParamsList))

	import urllib, urllib2
	import mimetypes
	
	# 他の API でも使えるように汎用化した関数
	def getAPIRequest(url,method,authentication,params={}):
		multipart=False
		
		newParams={}
		for key in params:
			value=params[key]
			if value==None:
				pass
			elif isinstance(value,int):
				newParams[key]=str(value)
			elif isinstance(value,float):
				newParams[key]=str(value)
			elif isinstance(value,str):
				newParams[key]=urllib.quote(value,"~")
			elif isinstance(value,unicode):
				newParams[key]=urllib.quote(value.encode("utf-8"),"~")
			elif isinstance(value,tuple):
				newParams[key]=value
				multipart=True
		params=newParams
		
		url=url.encode("utf-8")
		
		if method=="GET":
			paramList=[]
			for key in params:
				paramList.append(key+"="+params[key])
			request=urllib2.Request(url+"?"+"&".join(paramList))
		else:
			request=urllib2.Request(url)
		
		if authentication:
			# OAuth 対応のために必要なのはここだけ
			if not multipart:
				header=getOAuthHeader(url,method,params)
			else:
				header=getOAuthHeader(url,method,{})
			request.add_header("Authorization",header)
		
		if method=="POST":
			if not multipart:
				data=[]
				for key in params:
					data.append(key+"="+params[key])
				request.add_data("&".join(data))
			else:
				BOUNDARY="---AntunBoundaryBoundaryBoundary"
				data=""
				
				for key in params:
					data+="--"+BOUNDARY+"\r\n"
					
					if not isinstance(params[key],tuple):
						data+="Content-Disposition: form-data; name=\""+str(key)+"\""+"\r\n"
						data+="\r\n"
						data+=str(params[key])+"\r\n"
					else:
						fileType=mimetypes.guess_type(params[key][0])[0]
						assert fileType!=None,"Could not determine file type"
						
						data+="Content-Disposition: form-data; name=\""+str(key)+"\"; filename=\""+str(params[key][0])+"\""+"\r\n"
						data+="Content-Type: "+str(fileType)+"\r\n"
						data+="\r\n"
						data+=str(params[key][1])+"\r\n"
				
				data+="--"+BOUNDARY+"--"+"\r\n"
				
				request.add_header("Content-Type","multipart/form-data; boundary="+BOUNDARY)
				request.add_header("Content-Length",str(len(data)))
				request.add_data(data)
		else:
			assert multipart==False
		
		return request
	
	# statuses/update 固有の処理はこんなにシンプル
	def updateStatuses(tweet):
		params={
			"status" : tweet,
		}
		request=getAPIRequest("https://api.twitter.com/1.1/statuses/update.json","POST",True,params)
		stream=urllib2.urlopen(request)
		stream.read()
		stream.close()

	import json
	
	def getUserStreams():
		request=getAPIRequest("https://userstream.twitter.com/1.1/user.json","GET",True,{})
		
		return urllib2.urlopen(request,timeout=90.0)
	
	stream=self.twitter.getUserStreams()
	
	response=u""
	# 1行目のフォロワーリストを読み飛ばし (User Stream の場合だけ)
	while True:
		c=stream.read(1)
		if c==u"\n":
			break
		response+=c
	
	while True:
		# readline() でも一応動きますが、仕様 (関数の実装) 上、正確なリアルタイムでレスポンスが得られなくなるので、read() で読んでます
		response=u""
		while True:
			c=stream.read(1)
			if c==u"\n":
				break
			response+=c
		response=response.strip()
		
		# 接続維持のために、時々空行が送られてくる
		if response=="":
			continue
		
		response=json.loads(unicode(response),"utf-8")
		print(json.dumps(response,indent=4,ensure_ascii=False))

	import json
	import urlparse
	import base64
	import hashlib
	import hmac
	import SocketServer
	import BaseHTTPServer
	
	class Server(SocketServer.ThreadingMixIn,BaseHTTPServer.HTTPServer):
		allow_reuse_address=True
	
	class Handler(BaseHTTPServer.BaseHTTPRequestHandler):
		protocol_version="HTTP/1.1"
		
		def do_GET(self):
			query=urlparse.urlparse(self.path).query
			query=dict(map(lambda arg:(arg.split("=",1)+[None])[:2],query.split("&")))
			
			if "crc_token" in query:
				crcToken=query.get("crc_token")
				
				# 事前に確保した consumerSecret が設定されていること
				
				response={
					"response_token" : "sha256="+
					    base64.b64encode(hmac.new(
					    consumerSecret.encode(),msg=crcToken.encode(),digestmod=hashlib.sha256
					    ).digest()).decode(),
				}
				response=json.dumps(response)
				
				self.send_response(200)
				self.send_header("Content-type","application/json")
				self.send_header("Content-Length",str(len(response)))
				self.end_headers()
				
				self.wfile.write(response)
				self.wfile.flush()
	
	server=Server(("localhost",8080),Handler)
	server.serve_forever()