tumblrのOAuthトークンを取得するだけのスクリプト

api.py

# -*- coding: utf-8 -*-
import urllib, urllib2

from oauth import OAuth

class Api():
    def __init__(self,  atoken="", atokensecret="", ckey="", csecret="", hostname=""):
        self.oauth = OAuth(ckey, csecret, atoken, atokensecret)
        self.site = "http://api.tumblr.com/v2/blog/%s/" % (hostname)
    
    def initializer(self):
        ''' アクセストークン, シークレットを作る '''
        request_url = "http://www.tumblr.com/oauth/request_token"
        auth_url = "http://www.tumblr.com/oauth/authorize"
        accesstoken_url = "http://www.tumblr.com/oauth/access_token"
        return self.oauth.oauth_initializer(request_url, auth_url, accesstoken_url)
    
    def execute(self, url, method, params={}, file_param=()):
        ''' リクエスト処理の実行 '''
        # パラメータを成形して辞書に
        for key, val in params.items():
            if isinstance(val, unicode): val = val.encode("utf-8")
            params[key] = urllib.quote(str(val), "")
        if file_param:
            request = self.oauth.make_request_raw(url, method, params, file_param)
        else:
            request = self.oauth.make_request(url, method, params)
        try:
            response = urllib2.urlopen(request)
            data = response.read()
            return data
        except urllib2.URLError, e:
            return e.read()

    def post_text(self, body, title=""):
        url = self.site + "post"
        return self.execute(url,"POST", {"type":"text",
                                         "title":title,
                                         "body":body})
    
    def post_photo(self, source="", caption="", link="", image_obj=None, file_name=""):
        url = self.site + "post"
        if image_obj:
            if isinstance(file_name, unicode): file_name = file_name.encode("utf-8")
            data = ("data", image_obj, file_name)
        else:
            data = ()
        return self.execute(url, "POST", {"type":"photo",
                                          "caption":caption,
                                          "link":link,
                                          "source":source,
                                          },
                            data
                            )

get_token.py

# -*- coding: utf-8 -*-
from api import Api

if __name__ == "__main__":
    api = Api(atoken="", atokensecret="", ckey="ckey", csecret="csecret")
    print api.initializer()

oauth.py

# -*- coding: utf-8 -*-
import hashlib
import hmac
import mimetypes
import random
import time
import urllib, urllib2
import urlparse

def make_raw_request(params, file_param=()):
    BOUNDARY = 'BOXNYABOXNYA'
    body = []
    for name, value in params.items():
        body.append('--%s' % BOUNDARY)
        body.append('Content-Disposition: form-data; name="%s"' % name)
        body.append('Content-Type: text/plain; charset=utf-8')
        body.append('')
        body.append(urllib.unquote(value))
    if file_param:
        name, value, filename = file_param
        body.append('--%s' % BOUNDARY)
        body.append('Content-Disposition: form-data; name="%s"; filename="%s"' % (name, filename))
        body.append('Content-Type: %s' % mimetypes.guess_type(filename)[0])
        body.append('')
        body.append(value)
    
    body.append('--%s--' % BOUNDARY)
    body.append('')
    body = '\r\n'.join(body)
    header = {'Content-Type': 'multipart/form-data; boundary=%s' % BOUNDARY,
               'Content-Length': len(body)}
    return body, header

class OAuth():
    def __init__(self, ckey, csecret, atoken="", atokensecret=""):
        self.ckey = ckey
        self.csecret = csecret
        self.atoken = atoken
        self.atokensecret = atokensecret
    
    def make_signature(self, params, url, method, secret=""):
        ''' シグネチャを計算する '''
        pstr = "&".join(["%s=%s" % kv for kv in sorted(params.items())])
        msg = "%s&%s&%s" % (method, urllib.quote(url, ""), urllib.quote(pstr, ""))
        h = hmac.new("%s&%s" % (self.csecret, secret), msg, hashlib.sha1)
        return h.digest().encode("base64").strip()

    def init_params(self, extra_params={}):
        params = {
            "oauth_consumer_key":self.ckey,
            "oauth_signature_method": "HMAC-SHA1",
            "oauth_timestamp": str(int(time.time())),
            "oauth_nonce": str(random.getrandbits(64)),
            "oauth_version": "1.0"
            }
        params.update(extra_params)
        return params
    
    def make_request(self, url, method, extra_params={}):
        ''' OAuthのリクエストを作る '''
        params = self.init_params(extra_params)
        params["oauth_token"] = self.atoken
        params["oauth_signature"] = self.make_signature(params, url, method, self.atokensecret)
        for key in extra_params.keys():
            del params[key]
        
        if method == "GET":
            url = "%s?%s" % (url, urllib.urlencode(extra_params))
        request = urllib2.Request(url)
        
        if method == "POST":
            request.add_data("&".join(['%s=%s' % kv for kv in extra_params.items()]))
        
        oauth_header = "OAuth %s" % (", ".join(['%s="%s"' % (key, urllib.quote(val)) for key, val in params.items()]))
        request.add_header("Authorization", oauth_header)
        return request
    
    def make_request_raw(self, url, method, extra_params={}, file_param=()):
        params = self.init_params(extra_params)
        params["oauth_token"] = self.atoken
        params["oauth_signature"] = self.make_signature(params, url, method, self.atokensecret)
        body, header = make_raw_request(params, file_param)
        request = urllib2.Request(url, body, header)
        return request
        
    def oauth_initializer(self, request_url, auth_url ,accesstoken_url):
        # Get Token
        print "---* Getting a request token:"
        params = self.init_params()
        params["oauth_signature"] = self.make_signature(params, request_url, "GET")
        request = urllib2.Request("%s?%s" % (request_url, urllib.urlencode(params)))
        ret = urlparse.parse_qs(urllib2.urlopen(request).read())
        token = ret["oauth_token"][0]
        token_secret = ret["oauth_token_secret"][0]
        # Get PIN
        print "---* Please access to following URL, and authorize Boxnya.\n"
        print "%s?%s=%s\n" % (auth_url, "oauth_token", token)
        print "---* Next, please input a PIN that is given by tumblr.com."
        print "->"
        pin = raw_input()
        print "---* Getting an access token:",
        # Get Access Token
        params = self.init_params({"oauth_verifier":pin, "oauth_token":token})
        params["oauth_signature"] = self.make_signature(params, accesstoken_url, "GET", token_secret)
        request = urllib2.Request("%s?%s" % (accesstoken_url, urllib.urlencode(params)))
        fin = urlparse.parse_qs(urllib2.urlopen(request).read())
        print "\n---> Authorized."
        return {"atokensecret":fin["oauth_token_secret"][0], "atoken":fin["oauth_token"][0]}