X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/a9eef437702d5df7a2f97010e6798c689371808c..dbc38cef60897028891425c921dd0eca8c20a2e1:/forum_modules/facebookauth/authentication.py diff --git a/forum_modules/facebookauth/authentication.py b/forum_modules/facebookauth/authentication.py old mode 100755 new mode 100644 index f2c5b6b..8562bad --- a/forum_modules/facebookauth/authentication.py +++ b/forum_modules/facebookauth/authentication.py @@ -1,85 +1,91 @@ -import hashlib -from time import time -from datetime import datetime -from urllib import urlopen, urlencode -from forum.authentication.base import AuthenticationConsumer, ConsumerTemplateContext, InvalidAuthentication -from django.utils.translation import ugettext as _ - -import settings - -try: - from json import load as load_json -except: - from django.utils.simplejson import JSONDecoder - - def load_json(json): - decoder = JSONDecoder() - return decoder.decode(json.read()) - -class FacebookAuthConsumer(AuthenticationConsumer): - - def process_authentication_request(self, request): - API_KEY = settings.FB_API_KEY - - if API_KEY in request.COOKIES: - if self.check_cookies_signature(request.COOKIES): - if self.check_session_expiry(request.COOKIES): - return request.COOKIES[API_KEY + '_user'] - else: - raise InvalidAuthentication(_('Sorry, your Facebook session has expired, please try again')) - else: - raise InvalidAuthentication(_('The authentication with Facebook connect failed due to an invalid signature')) - else: - raise InvalidAuthentication(_('The authentication with Facebook connect failed, cannot find authentication tokens')) - - def generate_signature(self, values): - keys = [] - - for key in sorted(values.keys()): - keys.append(key) - - signature = ''.join(['%s=%s' % (key, values[key]) for key in keys]) + settings.FB_APP_SECRET - return hashlib.md5(signature).hexdigest() - - def check_session_expiry(self, cookies): - return datetime.fromtimestamp(float(cookies[settings.FB_API_KEY+'_expires'])) > datetime.now() - - def check_cookies_signature(self, cookies): - API_KEY = settings.FB_API_KEY - - values = {} - - for key in cookies.keys(): - if (key.startswith(API_KEY + '_')): - values[key.replace(API_KEY + '_', '')] = cookies[key] - - return self.generate_signature(values) == cookies[API_KEY] - - def get_user_data(self, key): - request_data = { - 'method': 'Users.getInfo', - 'api_key': settings.FB_API_KEY, - 'call_id': time(), - 'v': '1.0', - 'uids': key, - 'fields': 'name,first_name,last_name,email', - 'format': 'json', - } - - request_data['sig'] = self.generate_signature(request_data) - fb_response = load_json(urlopen(settings.REST_SERVER, urlencode(request_data)))[0] - - return { - 'username': fb_response['first_name'] + ' ' + fb_response['last_name'], - 'email': fb_response['email'] - } - -class FacebookAuthContext(ConsumerTemplateContext): - mode = 'BIGICON' - type = 'CUSTOM' - weight = 100 - human_name = 'Facebook' - code_template = 'modules/facebookauth/button.html' - extra_css = ["http://www.facebook.com/css/connect/connect_button.css"] - - API_KEY = settings.FB_API_KEY \ No newline at end of file +# -*- coding: utf-8 -*- + +import cgi +import logging + +from urllib import urlopen, urlencode +from forum.authentication.base import AuthenticationConsumer, ConsumerTemplateContext, InvalidAuthentication + +from django.conf import settings as django_settings +from django.utils.encoding import smart_unicode +from django.utils.translation import ugettext as _ + +import settings + +try: + from json import load as load_json +except Exception: + from django.utils.simplejson import JSONDecoder + + def load_json(json): + decoder = JSONDecoder() + return decoder.decode(json.read()) + +class FacebookAuthConsumer(AuthenticationConsumer): + + def prepare_authentication_request(self, request, redirect_to): + args = dict( + client_id=settings.FB_API_KEY, + redirect_uri="%s%s" % (django_settings.APP_URL, redirect_to), + scope="email" + ) + + facebook_api_authentication_url = "https://graph.facebook.com/oauth/authorize?" + urlencode(args) + + return facebook_api_authentication_url + + def process_authentication_request(self, request): + try: + args = dict(client_id=settings.FB_API_KEY, redirect_uri="%s%s" % (django_settings.APP_URL, request.path)) + + args["client_secret"] = settings.FB_APP_SECRET #facebook APP Secret + + args["code"] = request.GET.get("code", None) + response = cgi.parse_qs(urlopen("https://graph.facebook.com/oauth/access_token?" + urlencode(args)).read()) + access_token = response["access_token"][-1] + + + user_data = self.get_user_data(access_token) + assoc_key = user_data["id"] + + # Store the access token in cookie + request.session["access_token"] = access_token + request.session["assoc_key"] = assoc_key + + # Return the association key + return assoc_key + except Exception, e: + logging.error("Problem during facebook authentication: %s" % e) + raise InvalidAuthentication(_("Something wrond happened during Facebook authentication, administrators will be notified")) + + def get_user_data(self, access_token): + profile = load_json(urlopen("https://graph.facebook.com/me?" + urlencode(dict(access_token=access_token)))) + + name = profile["name"] + + # Check whether the length if the email is greater than 75, if it is -- just replace the email + # with a blank string variable, otherwise we're going to have trouble with the Django model. + email = smart_unicode(profile['email']) + if len(email) > 75: + email = '' + + # If the name is longer than 30 characters - leave it blank + if len(name) > 30: + name = '' + + # Return the user data. + return { + 'id' : profile['id'], + 'username': name, + 'email': email, + } + +class FacebookAuthContext(ConsumerTemplateContext): + mode = 'BIGICON' + type = 'CUSTOM' + weight = 100 + human_name = 'Facebook' + code_template = 'modules/facebookauth/button.html' + extra_css = [] + + API_KEY = settings.FB_API_KEY