0

我正在使用 python-social-auth 将用户登录到我的网站，效果很好，但我想使用自定义用户模型，它不仅可以保存有关用户的基本信息，还可以获取他们的个人资料图片。

这是我的用户模型

def get_upload_file_name(instance, filename):
    return "%s_%s" % (str(time()).replace('.', '_'), filename)

class UserProfile(models.Model):
    user = models.OneToOneField(User, unique=True)
    name = models.CharField(max_length=250, null=True, blank=True)
    profile_image = models.ImageField(upload_to = get_upload_file_name, null=True, blank=True)

    def __str__(self):
    return u'%s profile' % self.user.username

这是管道功能

def user_details(strategy, details, response, user=None, *args, **kwargs):
if user:
    if kwargs['is_new']:
        attrs = {'user': user}
        if strategy.backend.name == 'facebook':
            fb = {
                    'name': response['first_name']
                }
            new_user = dict(attrs.items() + fb.items())
            UserProfile.objects.create(
                **new_user
            )

        elif strategy.backend.name == 'google-oauth2':
            new_user = dict(attrs.items())
            UserProfile.objects.create(
                **new_user
            )

        elif strategy.backend.name == 'twitter':
            new_user = dict(attrs.items())
            UserProfile.objects.create(
                **new_user
            )

这是获取用户个人资料图像的另一个功能

def save_profile_picture(strategy, user, response, details, is_new=False,
                         *args, **kwargs):

    if is_new and strategy.backend.name == 'facebook':
        url = 'http://graph.facebook.com/{0}/picture'.format(response['id'])

        try:
            response = request('GET', url, params={'type': 'large'})
            response.raise_for_status()
        except HTTPError:
            pass
        else:
            S_user = setattr(UserProfile, "profile_image", "{0}_social.jpg".format(user.username), ContentFile(response.content))
            S_user.save()

我只是先在 facebook 上尝试，但我似乎无法在数据库中填充名称字段，而且我还必须登录两次才能将其保存到默认的 social-auth 表中。这两个函数都已添加到 settings.py 文件中，我还想知道它们在提示中的位置是否重要，因为它们位于底部，即身份验证过程的最后一部分？

4

1 回答 1

0

我想通了，因为我使用的是 python3，所以我应该在我的 dict 值上使用 list()，如下所示： attrs = dict(list(attrs.items()) + list(fb_data.items()))

另外，不要将图像保存在数据库中，最好只保存 url，节省大量空间

于 2014-11-07T14:48:46.980 回答