basic auth -- that was easy :)

This commit is contained in:
Kenneth Reitz
2011-07-23 23:18:17 -04:00
parent a3f9e54a00
commit 97d85351bb
2 changed files with 22 additions and 5 deletions
+16 -4
View File
@@ -48,14 +48,28 @@ class GithubCore(object):
return (settings.base_url + resource) return (settings.base_url + resource)
def _requests_pre_hook(*args, **kwargs):
return args, kwargs
def _get_http_resource(self, endpoint, params=None): def _get_http_resource(self, endpoint, params=None):
url = self._generate_url(endpoint) url = self._generate_url(endpoint)
r = requests.get(url, params=params)
args, kwargs = self._requests_pre_hook(url, params=params)
r = requests.get(*args, **kwargs)
r.raise_for_status() r.raise_for_status()
return r return r
def _patch_http_resource(self, endpoint, params=None):
url = self._generate_url(endpoint)
r = requests.patch(url, params=params)
r.raise_for_status()
return r
def _get_resource(self, resource, obj, **kwargs): def _get_resource(self, resource, obj, **kwargs):
@@ -94,15 +108,13 @@ class Github(GithubCore):
def get_user(self, username): def get_user(self, username):
# return 'kennethreitz'
"""Get a single user.""" """Get a single user."""
return self._get_resource(('users', username), User) return self._get_resource(('users', username), User)
# return User()
def get_me(self): def get_me(self):
"""Get the authenticated user.""" """Get the authenticated user."""
return self._get_resource(('users'), User) return self._get_resource(('user'), User)
+6 -1
View File
@@ -22,9 +22,14 @@ def no_auth():
return gh return gh
def basic_auth(): def basic_auth(username, password):
"""Returns an authenticated Github object, via HTTP Basic.""" """Returns an authenticated Github object, via HTTP Basic."""
def enable_auth(*args, **kwargs):
kwargs['auth'] = (username, password)
return args, kwargs
gh = Github() gh = Github()
gh._requests_pre_hook = enable_auth
return gh return gh