Question

我有一个网站，您可以注册成为使用该服务的人（客户）或提供该服务的人（工作人员）。我在models.py中创建了两个配置文件来表示每个配置文件。它们现在大部分都非常相似。当您转到这两个表单时，两个表单都会正确显示，如果您以客户身份注册并按提交，一切顺利，新用户将显示在http://127.0.0.1:8000/admin/的“客户资料”中。但是，如果您尝试注册为工作人员，则会出现以下错误：

Exception Type: RelatedObjectDoesNotExist
Exception Value:    
User has no workerprofile.

我不明白这一点，因为您将在下面的代码中看到我使用customerprofile并且它运行正常，如果我使用workerprofile它会崩溃。

Views.py：

def signup_as_worker(request):
    if request.method == 'POST':
        form = WorkerSignUpForm(request.POST)
        if form.is_valid():
            user = form.save()
            user.refresh_from_db()  # load the profile instance created by the signal
            user.workerprofile.birth_date = form.cleaned_data.get('birth_date')
            user.workerprofile.university = form.cleaned_data.get('university')
            user.save()  # explicitly save custom fields not in User model
            raw_password = form.cleaned_data.get('password1')
            user = authenticate(username=user.username, password=raw_password)
            login(request, user)  # login user after signup
            return redirect('home')
    else:
        form = WorkerSignUpForm()
    return render(request, 'core/signup_as_worker.html', {'form': form})

def signup_as_customer(request):
    if request.method == 'POST':
        form = CustomerSignUpForm(request.POST)
        if form.is_valid():
            user = form.save()
            user.refresh_from_db()  # load the profile instance created by the signal
            user.customerprofile.birth_date = form.cleaned_data.get('birth_date')
            user.customerprofile.university = form.cleaned_data.get('university')
            user.save()  # explicitly save custom fields not in User model
            raw_password = form.cleaned_data.get('password1')
            user = authenticate(username=user.username, password=raw_password)
            login(request, user)  # login user after signup
            return redirect('home')
    else:
        form = CustomerSignUpForm()
    return render(request, 'core/signup_as_customer.html', {'form': form})

forms.py：

class WorkerSignUpForm(UserCreationForm):
    #birth_date and university fields need to be declared seperately because they are not apart of User:
    birth_date = forms.DateField(help_text='Required. Format: YYYY-MM-DD')
    university = forms.CharField()


    class Meta:
        model = User
        fields = ('username',
                  'email',
                  'first_name',
                  'last_name',
                  'birth_date',
                  'university',
                  'password1',
                  'password2', )


class CustomerSignUpForm(UserCreationForm):
    #birth_date and university fields need to be declared seperately because they are not apart of User:
    birth_date = forms.DateField(help_text='Required. Format: YYYY-MM-DD')
    university = forms.CharField()


    class Meta:
        model = User
        fields = ('username',
                  'email',
                  'first_name',
                  'last_name',
                  'birth_date',
                  'university',
                  'password1',
                  'password2', )

models.py：

from django.db import models
from django.contrib.auth.models import User
from django.db.models.signals import post_save
from django.dispatch import receiver

class WorkerProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    university = models.CharField(max_length=30, blank=True)
    birth_date = models.DateField(null=True, blank=True)

    role = models.CharField(max_length = 10, default = 'USER')


    def __str__(self):
        return self.user.username

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

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




class CustomerProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    university = models.CharField(max_length=30, blank=True)
    birth_date = models.DateField(null=True, blank=True)

    role = models.CharField(max_length = 10, default = 'CUSTOMER')
    needLaundryDone = models.BooleanField(default = False)

    def __str__(self):
        return self.user.username

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

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

我不明白问题是什么。

Answer 1

两个模型的信号处理程序方法名称相同。您实际上正在重新定义方法，因此只调用第二组方法。将您的CustomerProfile处理程序重命名为create_customer_profile和save_customer_profile。

扩展django模型的问题

1 个答案: