108 lines
4.3 KiB
Python
108 lines
4.3 KiB
Python
import logging
|
|
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import redirect
|
|
from django.utils.http import urlquote
|
|
|
|
from social.pipeline.social_auth import associate_by_email
|
|
from social.apps.django_app.default.models import UserSocialAuth
|
|
from social.apps.django_app.middleware import SocialAuthExceptionMiddleware
|
|
from social.exceptions import (AuthAlreadyAssociated,SocialAuthBaseException)
|
|
from social.utils import social_logger
|
|
|
|
ANONYMOUS_AVATAR = '/static/images/header/avatar.png'
|
|
(NO_AVATAR, GRAVATAR, TWITTER, FACEBOOK, PRIVATETAR) = (0, 1, 2, 3, 4)
|
|
AVATARS = (NO_AVATAR, GRAVATAR, TWITTER, FACEBOOK, PRIVATETAR)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def selectively_associate_by_email(backend, details, user=None, *args, **kwargs):
|
|
"""
|
|
Associate current auth with a user with the same email address in the DB.
|
|
This pipeline entry is not 100% secure unless you know that the providers
|
|
enabled enforce email verification on their side, otherwise a user can
|
|
attempt to take over another user account by using the same (not validated)
|
|
email address on some provider.
|
|
|
|
Not using Facebook or Twitter to authenticate a user.
|
|
"""
|
|
if backend.name in ('twitter', 'facebook'):
|
|
return None
|
|
return associate_by_email(backend, details, user=None, *args, **kwargs)
|
|
|
|
def facebook_extra_values( user, extra_data):
|
|
try:
|
|
facebook_id = extra_data.get('id')
|
|
user.profile.facebook_id = facebook_id
|
|
if user.profile.avatar_source is None or user.profile.avatar_source is PRIVATETAR:
|
|
user.profile.avatar_source = FACEBOOK
|
|
user.profile.save()
|
|
return True
|
|
except Exception,e:
|
|
logger.error(e)
|
|
return False
|
|
|
|
def twitter_extra_values( user, extra_data):
|
|
try:
|
|
twitter_id = extra_data.get('screen_name')
|
|
profile_image_url = extra_data.get('profile_image_url_https')
|
|
user.profile.twitter_id = twitter_id
|
|
if user.profile.avatar_source is None or user.profile.avatar_source in (TWITTER, PRIVATETAR):
|
|
user.profile.pic_url = profile_image_url
|
|
if user.profile.avatar_source is None or user.profile.avatar_source is PRIVATETAR:
|
|
user.profile.avatar_source = TWITTER
|
|
user.profile.save()
|
|
return True
|
|
except Exception,e:
|
|
logger.error(e)
|
|
return False
|
|
|
|
def deliver_extra_data(backend, user, social, *args, **kwargs):
|
|
if backend.name is 'twitter':
|
|
twitter_extra_values( user, social.extra_data)
|
|
if backend.name is 'facebook':
|
|
facebook_extra_values( user, social.extra_data)
|
|
|
|
# following is needed because of length limitations in a unique constrain for MySQL
|
|
def chop_username(username, *args, **kwargs):
|
|
if username and len(username)>222:
|
|
return {'username':username[0:222]}
|
|
|
|
def selective_social_user(backend, uid, user=None, *args, **kwargs):
|
|
provider = backend.name
|
|
social = backend.strategy.storage.user.get_social_auth(provider, uid)
|
|
if social:
|
|
if user and social.user != user:
|
|
if backend.name not in ('twitter', 'facebook'):
|
|
msg = 'This {0} account is already in use.'.format(provider)
|
|
raise AuthAlreadyAssociated(backend, msg)
|
|
elif not user:
|
|
user = social.user
|
|
return {'social': social,
|
|
'user': user,
|
|
'is_new': user is None,
|
|
'new_association': False}
|
|
|
|
# http://stackoverflow.com/a/19361220
|
|
# adapting https://github.com/omab/python-social-auth/blob/v0.2.10/social/apps/django_app/middleware.py#L25
|
|
|
|
class SocialAuthExceptionMiddlewareWithoutMessages(SocialAuthExceptionMiddleware):
|
|
"""
|
|
a modification of SocialAuthExceptionMiddleware to pass backend and message without
|
|
attempting django.messages
|
|
"""
|
|
def process_exception(self, request, exception):
|
|
|
|
if isinstance(exception, SocialAuthBaseException):
|
|
backend = getattr(request, 'backend', None)
|
|
backend_name = getattr(backend, 'name', 'unknown-backend')
|
|
|
|
message = self.get_message(request, exception)
|
|
social_logger.error(message)
|
|
|
|
url = self.get_redirect_uri(request, exception)
|
|
url += ('?' in url and '&' or '?') + \
|
|
'message={0}&backend={1}'.format(urlquote(message),
|
|
backend_name)
|
|
return redirect(url)
|