django rest_auth使用UserProfile扩展User

时间:2015-03-09 10:43:01

标签: django-models django-forms django-rest-framework

我在aljuth和rest_auth上使用django 1.7.5。

使用新字段创建自定义SignupForm成功,但无法在注册时获取值。

company_name是被忽略的字段,存储名字和姓氏......

1)rest_auth / forms.py

class SLCustomSignupForm(forms.Form):
""" in settings ACCOUNT_SIGNUP_FORM_CLASS='rest_auth.forms.SLCustomSignupForm' """

first_name = forms.CharField(max_length=30, label='Nome')
last_name = forms.CharField(max_length=30, label='Cognome')
company_name = forms.CharField(max_length=30, label='Azienda')

def signup(self, request, user):
    """
    Invoked at signup time to complete the signup of the user.
    """
    user.first_name = self.cleaned_data['first_name']
    user.last_name = self.cleaned_data['last_name']
    user.profilo_utente.company_name = self.cleaned_data['company_name']
    user.save()

在rest_auth / models.py

from django.contrib.auth.models import User
from django.db import models
from allauth.account.models import EmailAddress


class ProfiloUtente(models.Model):
     user = models.OneToOneField(User, related_name='profilo_utente')
     company_name = models.CharField(max_length=100)

def __unicode__(self):
    return "{}'s profile".format(self.user.username)

class Meta:
    db_table = 'profilo_utente'

def account_verified(self):
    if self.user.is_authenticated:
        result = EmailAddress.objects.filter(email=self.user.email)
        if len(result):
            return result[0].verified
    return False

User.profilo_utente = property(lambda u: ProfiloUtente.objects.get_or_create(user=u)[0])

我已经使用ProfiloUtente修改了提供的UserProfile模型,只是为了了解管理这些数据的位置和人员。

从shell访问:

    from django.contrib.auth.models import User
a = User.objects.get(username="myuser")
a.profilo_utente.company_name                  # see current value

1 个答案:

答案 0 :(得分:0)

如果在user.save()调用之前它们不存在的信号中创建了配置文件,那么您尝试在保存主对象之前添加到关系。

user.first_name = self.cleaned_data['first_name']
user.last_name = self.cleaned_data['last_name']
user.save()     
user.profilo_utente.company_name = self.cleaned_data['company_name']
user.profilo_utente.save()