forum_modules/facebookauth/authentication.py

   1 import hashlib
   2 from time import time
   3 from datetime import datetime
   4 from urllib import urlopen,  urlencode
   5 from urlparse import parse_qs
   6 from forum.authentication.base import AuthenticationConsumer, ConsumerTemplateContext, InvalidAuthentication
   7 from django.utils.translation import ugettext as _
   8 from django.utils.encoding import smart_unicode
   9
  10 import settings
  11
  12 try:
  13     from json import load as load_json
  14 except:
  15     from django.utils.simplejson import JSONDecoder
  16
  17     def load_json(json):
  18         decoder = JSONDecoder()
  19         return decoder.decode(json.read())
  20
  21 class FacebookAuthConsumer(AuthenticationConsumer):
  22
  23     def process_authentication_request(self, request):
  24         API_KEY = str(settings.FB_API_KEY)
  25
  26         # Check if the Facebook cookie has been received.
  27         if 'fbs_%s' % API_KEY in request.COOKIES:
  28             fbs_cookie = request.COOKIES['fbs_%s' % API_KEY]
  29             parsed_fbs = parse_qs(smart_unicode(fbs_cookie))
  30             self.parsed_fbs = parsed_fbs
  31
  32             # Check if the session hasn't expired.
  33             if self.check_session_expiry(request.COOKIES):
  34                 return parsed_fbs['uid'][0]
  35             else:
  36                 raise InvalidAuthentication(_('Sorry, your Facebook session has expired, please try again'))
  37         else:
  38             raise InvalidAuthentication(_('The authentication with Facebook connect failed, cannot find authentication tokens'))
  39     def check_session_expiry(self, cookies):
  40         return datetime.fromtimestamp(float(self.parsed_fbs['expires'][0])) > datetime.now()
  41
  42     def get_user_data(self, cookies):
  43         API_KEY = str(settings.FB_API_KEY)
  44         fbs_cookie = cookies['fbs_%s' % API_KEY]
  45         parsed_fbs = parse_qs(smart_unicode(fbs_cookie))
  46
  47         # Communicate with the access token to the Facebook oauth interface.
  48         json = load_json(urlopen('https://graph.facebook.com/me?access_token=%s' % parsed_fbs['access_token'][0]))
  49
  50         first_name = smart_unicode(json['first_name'])
  51         last_name = smart_unicode(json['last_name'])
  52         full_name = '%s %s' % (first_name, last_name)
  53
  54         # There is a limit in the Django user model for the username length (no more than 30 characaters)
  55         if len(full_name) <= 30:
  56             username = full_name
  57         # If the full name is too long use only the first
  58         elif len(first_name) <= 30:
  59             username = first_name
  60         # If it's also that long -- only the last
  61         elif len(last_name) <= 30:
  62             username = last_name
  63         # If the real name of the user is indeed that weird, let him choose something on his own =)
  64         else:
  65             username = ''
  66
  67         # Check whether the length if the email is greater than 75, if it is -- just replace the email
  68         # with a blank string variable, otherwise we're going to have trouble with the Django model.
  69         email = smart_unicode(json['email'])
  70         if len(email) > 75:
  71             email = ''
  72
  73         # Return the user data.
  74         return {
  75             'username': username,
  76             'email': email,
  77         }
  78
  79 class FacebookAuthContext(ConsumerTemplateContext):
  80     mode = 'BIGICON'
  81     type = 'CUSTOM'
  82     weight = 100
  83     human_name = 'Facebook'
  84     code_template = 'modules/facebookauth/button.html'
  85     extra_css = ["http://www.facebook.com/css/connect/connect_button.css"]
  86
  87     API_KEY = settings.FB_API_KEY