django django-allauth在信号中保存来自社交登录的extra_data

时间:2016-11-18 19:40:54

标签: python django django-allauth

设置

我正在使用Django 1.8.15和django-allauth 0.28.0

描述

我想要做的很容易解释: 如果用户通过社交媒体(Facebook,谷歌+或推特)登录,我想从extra_data检索sociallogin以获取头像的网址和其他信息以将其存储在我的{{1}中},这是我Profile模型的One-to-one relation

我的方法

1)

首先,我添加了一个Userhere的类,在用户通过社交媒体登录后调用该类。

models.py

pre_social_login

settings.py

class SocialAdapter(DefaultSocialAccountAdapter):
def pre_social_login(self, request, sociallogin):
    """Get called after a social login. check for data and save what you want."""
    user = User.objects.get(id=request.user.id)  # Error: user not available
    profile = Profile(user=user)
    picture = sociallogin.account.extra_data.get('picture', None)
    if picture:
        # ... code to save picture to profile 

我想从用户那里获取实例,但似乎它还没有创建。所以我尝试了以下内容:

2)

如果添加了新用户,我还有另一个信号可以创建个人资料:

models.py

SOCIALACCOUNT_ADAPTER = 'profile.models.SocialAdapter'

我添加了以下内容:

@receiver(post_save, sender=settings.AUTH_USER_MODEL)
def create_profile_for_new_user(sender, created, instance, **kwargs):
    """Signal, that creates a new profile for a new user."""
    if created:
        profile = Profile(user=instance)
        # following code was pasted here
        profile.save()

但是data = SocialAccount.objects.filter(user=instance) # do more with data 它总是空的[]

我有理解这个问题。我的意思是,如果用户通过社交媒体登录,我无法从data(案例1)访问用户,因为它尚未创建,那么当我尝试User时应创建SocialAccount在案例2中得到它?!你还有其他想法可以解决这个问题吗?或者我在这里遗漏了什么?

提前致谢

1 个答案:

答案 0 :(得分:2)

经过几个小时的绝望后得到了它。

我刚刚使用了另一个信号user_signed_up,而不是来自socialaccount的信号

from allauth.account.signals import user_signed_up

@receiver(user_signed_up)
def retrieve_social_data(request, user, **kwargs):
    """Signal, that gets extra data from sociallogin and put it to profile."""
    # get the profile where i want to store the extra_data
    profile = Profile(user=user)
    # in this signal I can retrieve the obj from SocialAccount
    data = SocialAccount.objects.filter(user=user, provider='facebook')
    # check if the user has signed up via social media
    if data:
        picture = data[0].get_avatar_url()
        if picture:
            # custom def to save the pic in the profile
            save_image_from_url(model=profile, url=picture)
        profile.save()

http://django-allauth.readthedocs.io/en/latest/signals.html#allauth-account