github2/github2/request.py

import sys
import httplib
import simplejson
from urlparse import urlparse
from urllib import urlencode

URL_PREFIX = "http://github.com/api/v2/json"

class GithubError(Exception):
    """An error occured when making a request to the Github API."""

class GithubRequest(object):
    url_prefix = URL_PREFIX
    GithubError = GithubError

    connector_for_scheme = {
        "http": httplib.HTTPConnection,
        "https": httplib.HTTPSConnection,
    }

    def __init__(self, username, api_token, url_prefix=None):
        self.username = username
        self.api_token = api_token
        self.url_prefix = url_prefix or self.url_prefix
    
    def encode_authentication_data(self, extra_post_data):
        post_data = {"user": self.username,
                     "token": self.api_token}
        post_data.update(extra_post_data) 
        return urlencode(post_data)

    def get(self, *path_components):
        path_components = filter(None, path_components)
        return self.make_request("/".join(path_components))

    def post(self, *path_components, **extra_post_data):
        path_components = filter(None, path_components)
        return self.make_request("/".join(path_components), extra_post_data)

    def make_request(self, path, extra_post_data=None):
        extra_post_data = extra_post_data or {}
        url = "/".join([self.url_prefix, path])
        print("URL: %s" % url)
        resource = urlparse(url)
        post_data = self.encode_authentication_data(extra_post_data)
        print("POST_DATA: %s" % post_data)
        connector = self.connector_for_scheme[resource.scheme]
        headers = self.http_headers
        headers["Accept"] = "text/html"
        headers["Content-Length"] = str(len(post_data))
        connection = connector(resource.hostname, resource.port)
        connection.request("POST", resource.path, post_data, headers)
        response = connection.getresponse()
        response_text = response.read()
        print(response_text)
        json = simplejson.loads(response_text)
        if json.get("error"):
            raise self.GithubError(json["error"][0]["error"])

        return json

    @property
    def http_headers(self):
        return {"User-Agent": "pygithub2 v1",
                "Accept-Encoding": "application/json"}