]> git.openstreetmap.org Git - osqa.git/blobdiff - forum_modules/oauthauth/consumer.py
Merge pull request #49 from udacity/login_logout_links
[osqa.git] / forum_modules / oauthauth / consumer.py
old mode 100755 (executable)
new mode 100644 (file)
index e3cbda6..ee9bf97
-import urllib\r
-import urllib2\r
-import httplib\r
-import time\r
-\r
-from forum.authentication.base import AuthenticationConsumer, InvalidAuthentication\r
-from django.utils.translation import ugettext as _\r
-\r
-from lib import oauth\r
-\r
-class OAuthAbstractAuthConsumer(AuthenticationConsumer):\r
-\r
-    def __init__(self, consumer_key, consumer_secret, server_url, request_token_url, access_token_url, authorization_url):\r
-        self.consumer_secret = consumer_secret\r
-        self.consumer_key = consumer_key\r
-\r
-        self.consumer = oauth.OAuthConsumer(consumer_key, consumer_secret)\r
-        self.signature_method = oauth.OAuthSignatureMethod_HMAC_SHA1()\r
-\r
-        self.server_url = server_url\r
-        self.request_token_url = request_token_url\r
-        self.access_token_url = access_token_url\r
-        self.authorization_url = authorization_url\r
-\r
-    def prepare_authentication_request(self, request, redirect_to):\r
-        request_token = self.fetch_request_token()\r
-        request.session['unauthed_token'] = request_token.to_string()\r
-        return self.authorize_token_url(request_token)\r
-\r
-    def process_authentication_request(self, request):\r
-        unauthed_token = request.session.get('unauthed_token', None)\r
-        if not unauthed_token:\r
-             raise InvalidAuthentication(_('Error, the oauth token is not on the server'))\r
-\r
-        token = oauth.OAuthToken.from_string(unauthed_token)\r
-\r
-        if token.key != request.GET.get('oauth_token', 'no-token'):\r
-            raise InvalidAuthentication(_("Something went wrong! Auth tokens do not match"))\r
-\r
-        access_token = self.fetch_access_token(token)\r
-\r
-        return access_token.to_string()\r
-\r
-    def get_user_data(self, key):\r
-        #token = oauth.OAuthToken.from_string(access_token)\r
-        return {}\r
-        \r
-    def fetch_request_token(self):\r
-        oauth_request = oauth.OAuthRequest.from_consumer_and_token(self.consumer, http_url=self.request_token_url)\r
-        oauth_request.sign_request(self.signature_method, self.consumer, None)\r
-        params = oauth_request.parameters\r
-        data = urllib.urlencode(params)\r
-        full_url='%s?%s'%(self.request_token_url, data)\r
-        response = urllib2.urlopen(full_url)\r
-        return oauth.OAuthToken.from_string(response.read())\r
-\r
-    def authorize_token_url(self, token, callback_url=None):\r
-        oauth_request = oauth.OAuthRequest.from_token_and_callback(token=token,\\r
-                        callback=callback_url, http_url=self.authorization_url)\r
-        params = oauth_request.parameters\r
-        data = urllib.urlencode(params)\r
-        full_url='%s?%s'%(self.authorization_url, data)\r
-        return full_url\r
-\r
-    def fetch_access_token(self, token):\r
-        oauth_request = oauth.OAuthRequest.from_consumer_and_token(self.consumer, token=token, http_url=self.access_token_url)\r
-        oauth_request.sign_request(self.signature_method, self.consumer, token)\r
-        params = oauth_request.parameters\r
-        data = urllib.urlencode(params)\r
-        full_url='%s?%s'%(self.access_token_url, data)\r
-        response = urllib2.urlopen(full_url)\r
-        return oauth.OAuthToken.from_string(response.read())\r
-\r
-    def fetch_data(self, token, http_url, parameters=None):\r
-        access_token = oauth.OAuthToken.from_string(token)\r
-        oauth_request = oauth.OAuthRequest.from_consumer_and_token(\r
-            self.consumer, token=access_token, http_method="GET",\r
-            http_url=http_url, parameters=parameters,\r
-        )\r
-        oauth_request.sign_request(self.signature_method, self.consumer, access_token)\r
-\r
-        url = oauth_request.to_url()\r
-        connection = httplib.HTTPSConnection(self.server_url)\r
-        connection.request(oauth_request.http_method, url)\r
-\r
-        return connection.getresponse().read()\r
-\r
+import urllib
+import urllib2
+import httplib
+import time
+
+from forum.settings import APP_URL
+from forum.authentication.base import AuthenticationConsumer, InvalidAuthentication
+from django.utils.translation import ugettext as _
+from django.core.urlresolvers import reverse
+
+from settings import TWITTER_AUTO_CALLBACK_REDIRECT
+from lib import oauth2
+
+class OAuthAbstractAuthConsumer(AuthenticationConsumer):
+
+    def __init__(self, consumer_key, consumer_secret, server_url, request_token_url, access_token_url, authorization_url):
+        self.consumer_secret = consumer_secret
+        self.consumer_key = consumer_key
+
+        self.consumer = oauth2.Consumer(consumer_key, consumer_secret)
+        self.signature_method = oauth2.SignatureMethod_HMAC_SHA1()
+
+        self.server_url = server_url
+        self.request_token_url = request_token_url
+        self.access_token_url = access_token_url
+        self.authorization_url = authorization_url
+
+    def prepare_authentication_request(self, request, redirect_to):
+        request_token = self.fetch_request_token()
+        request.session['unauthed_token'] = request_token.to_string()
+        return self.authorize_token_url(request_token)
+
+    def process_authentication_request(self, request):
+        unauthed_token = request.session.get('unauthed_token', None)
+        if not unauthed_token:
+             raise InvalidAuthentication(_('Error, the oauth token is not on the server'))
+
+        token = oauth2.Token.from_string(unauthed_token)
+
+        if token.key != request.GET.get('oauth_token', 'no-token'):
+            raise InvalidAuthentication(_("Something went wrong! Auth tokens do not match"))
+
+        access_token = self.fetch_access_token(token, request.GET.get('oauth_verifier', '')) 
+
+        return access_token.to_string()
+
+    def get_user_data(self, key):
+        #token = oauth.OAuthToken.from_string(access_token)
+        return {}
+        
+    def fetch_request_token(self):
+        parameters = {}
+        # If the installation is configured to automatically redirect to the Twitter provider done page -- do it.
+        if bool(TWITTER_AUTO_CALLBACK_REDIRECT):
+            callback_url = '%s%s' % (APP_URL, reverse('auth_provider_done', prefix='/', kwargs={'provider' : 'twitter'}))
+            # Pass
+            parameters.update({
+                'oauth_callback' : callback_url,
+            })
+
+        oauth_request = oauth2.Request.from_consumer_and_token(self.consumer, http_url=self.request_token_url, parameters=parameters)
+        oauth_request.sign_request(self.signature_method, self.consumer, None)
+        params = oauth_request
+        data = urllib.urlencode(params)
+        full_url='%s?%s'%(self.request_token_url, data)
+        response = urllib2.urlopen(full_url)
+        return oauth2.Token.from_string(response.read())
+
+    def authorize_token_url(self, token, callback_url=None):
+        oauth_request = oauth2.Request.from_token_and_callback(token=token,\
+                        callback=callback_url, http_url=self.authorization_url)
+        params = oauth_request
+        data = urllib.urlencode(params)
+        full_url='%s?%s'%(self.authorization_url, data)
+        return full_url
+
+    def fetch_access_token(self, token, oauth_verifier): 
+        oauth_request = oauth2.Request.from_consumer_and_token(self.consumer, token=token, http_url=self.access_token_url)
+        oauth_request['oauth_verifier'] = oauth_verifier 
+        oauth_request.sign_request(self.signature_method, self.consumer, token)
+        params = oauth_request
+        data = urllib.urlencode(params)
+        full_url='%s?%s'%(self.access_token_url, data)
+        response = urllib2.urlopen(full_url)
+        return oauth2.Token.from_string(response.read())
+
+    def fetch_data(self, token, http_url, parameters=None):
+        access_token = oauth2.Token.from_string(token)
+        oauth_request = oauth2.Request.from_consumer_and_token(
+            self.consumer, token=access_token, http_method="GET",
+            http_url=http_url, parameters=parameters,
+        )
+        oauth_request.sign_request(self.signature_method, self.consumer, access_token)
+
+        url = oauth_request.to_url()
+        connection = httplib.HTTPSConnection(self.server_url)
+        connection.request("GET", url)
+
+        return connection.getresponse().read()
+