gistfile1.py

import json

import requests
from oauth_hook import OAuthHook


class NetDNAOAuthHook(OAuthHook):

    def __init__(self, consumer_key, consumer_secret, token=None,
                 token_secret=None, header_auth=True, **kwargs):
        super(NetDNAOAuthHook, self).__init__(token, token_secret,
                                              consumer_key, consumer_secret,
                                              header_auth)


class NetDNA(object):

    def __init__(self, company_alias, key, secret,
                 server='rws.netdna.com', **kwargs):
        self.company_alias = company_alias
        self.server = server
        self.client = requests.session(
                        hooks={
                          'pre_request': NetDNAOAuthHook(key, secret, **kwargs)
                        }
                      )

    def _get_url(self, uri):
        return "https://%s/%s%s" % (
                                     self.server,
                                     self.company_alias,
                                     uri
                                   )

    def _response_as_json(self, method, uri, **kwargs):
        print "Making %s request to %s\n" % (method.upper(), self._get_url(uri))
        response = getattr(self.client, method)(self._get_url(uri), **kwargs)
        response_json = json.loads(response.content)
        if response.status_code != 200:
            raise Exception("%d: %s" % (
                             response.status_code,
                             response_json['error']['message'])
                           )
        return response_json

    def get(self, uri, **kwargs):
        return self._response_as_json("get", uri, **kwargs)

    def post(self, uri, data={}, **kwargs):
        return self._response_as_json("post", uri, data=data, **kwargs)

    def put(self, uri, **kwargs):
        return self._response_as_json("put", uri, **kwargs)

    def delete(self, uri, **kwargs):
        return self._response_as_json("delete", uri, **kwargs)

if __name__ == "__main__":
    from pprint import pprint

    api = NetDNA("dlitvak",
                 "0de95ee2fa4138252c1e431c177ac04904feb56f7",
                 "35f1656f51a0e8940baff71410f")

    pprint(api.get("/account.json"))