延長django模型的問題

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。