时间戳

首页 > 解决方案 > 不为自定义用户模型保存配置文件数据

python - 不为自定义用户模型保存配置文件数据

问题描述

我是 python 新手，我正在尝试从用户创建模型扩展用户注册，我用我想要的字段创建了配置文件模型，并在保存用户时用信号保存了配置文件对象，配置文件创建成功并且与用户关联但未保存个人资料数据（在此示例中：职位、公司、电话）为空

在模型.py

class profile (models.Model):
    user = models.OneToOneField(User, on_delete = models.CASCADE)
    company = models.CharField(max_length=100)
    jobTitle = models.CharField(max_length = 100,default ='')
    phonenumber = PhoneNumberField(null=False, blank=False, unique=True, help_text='Contact phone number', default ='')


@receiver(post_save, sender = User)
def create_user_profile( sender, instance, created, **kwargs):
    if created:
        profile.objects.create(user = instance)
    instance.profile.save()

@receiver(post_save, sender=User)
def save_profile(sender, instance, **kwargs):
    instance.profile.save()

在forms.py

class UserRegisterForm(UserCreationForm):
    email = forms.EmailField(required= True)
    company = forms.CharField(required = True)
    phonenumber = PhoneNumberField(widget=forms.TextInput(attrs={'placeholder': ('Phone')}), label=("Phone number"), required=True)
    first_name = forms.CharField(required = True)
    jobTitle = forms.CharField(required = True)


    class Meta:
        model = User
        fields = [
            'username',
            'first_name',
            'last_name',
            'email',
            'password1',
            'password2',
            'company',
            'jobTitle'
            ]

    def save(self, commit=True):
        user = super(UserRegisterForm, self).save(commit=False)
        user.first_name = self.cleaned_data['first_name']
        user.last_name = self.cleaned_data['last_name']
        user.email = self.cleaned_data['email']

        if commit:
            user.save()


        return user

在views.py中

#User register
def signup(request):
    if request.method == 'POST':
        form = UserRegisterForm(request.POST)

        if form.is_valid():
            user = form.save()
            UserProfile = user.profile
            username = form.cleaned_data.get('username')
            company = form.cleaned_data.get('company')
            jobTitle = form.cleaned_data.get('jobTitle')
            phonenumber = form.cleaned_data.get('phonenumber')
            UserProfile.save()

            messages.success(request, f'Account is created for {username}!')
            return redirect('login')
    else:
            form = UserRegisterForm()
    return render(request, 'users/signup.html', {'form': form})

所以你能帮我理解是什么导致了这个问题吗？我希望在注册到配置文件模型时保存插入的值。

谢谢

标签： pythondjangopython-3.xregistration

解决方案

删除第二个@receiver(post_save, sender=User)，它没用（它总是在 #1 配置文件中完成）。

当您这样做时user = form.save()，会发出信号，创建一个空 profile的.

因此，就在 之后user = form.save()，获取通过信号创建的配置文件，例如：

profile = Profile.objects.get(user=user)
username = form.cleaned_data.get('username')
company = form.cleaned_data.get('company')
jobTitle = form.cleaned_data.get('jobTitle')
phonenumber = form.cleaned_data.get('phonenumber')
profile.save()

我的建议：阅读 Python 的 PEP8：类总是“UpCase”，变量总是小写。

推荐阅读