問題擴展django用戶模型

Question

我有一個網站，用戶應該能夠註冊爲「工人」或「客戶」，一個超級模型類型的網站。我創建了兩個模型WorkerProfile和CustomerProfile以及這兩種表單，但是每次我提交客戶或工作者表單時，新用戶都會在http://127.0.0.1:8000/admin/的數據庫中同時存入客戶個人資料和工作人員個人資料，我如何防止這種情況發生發生了什麼？

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_customer_profile(sender, instance, created, **kwargs): 
    if created: 
     CustomerProfile.objects.create(user=instance) 

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

forms.py：

from django import forms 
from django.contrib.auth.forms import UserCreationForm 
from django.contrib.auth.models import User 

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',) 

views.py：

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

signup.html：

{% extends 'core/base.html' %} 

{% block head %} 
    <title> Sign Up</title> 
{% endblock %} 
{% block body %} 


    <h3>Sign Up As Worker</h3> 
    <form method="post"> 
    {% csrf_token %} 
    {{ form_worker.as_p }} 
    <button type="submit">Sign up</button> 
    </form> 

    <h3>Sign Up As Customer</h3> 
    <form method="post"> 
    {% csrf_token %} 
    {{ form_customer.as_p }} 
    <button type="submit">Sign up</button> 
    </form> 
{% endblock %} 

Answer 1

請勿在此處使用信號。它們都會在用戶保存時觸發，並且每個都創建相關的對象。

您應該刪除這些信號，而是在視圖中執行此操作。在每個表單的is_valid塊中，您只能創建所需的特定對象。

if form_worker.is_valid(): 
    user = form_worker.save() 
    worker = WorkerProfile(user=user) 
    worker.birth_date = form_worker.cleaned_data.get('birth_date') 
    worker.university = form_worker.cleaned_data.get('university') 
    worker.save() 
    raw_password = form_worker.cleaned_data.get('password1') 
    ... 
elif form_customer.is_valid(): 
    user = form_customer.save() 
    customer = CustomerProfile(user=user) 
    ... 

Answer 2

這是正常的：

user = models.OneToOneField(User, on_delete=models.CASCADE) 

如果調用此兩款車型，所以在保存過程中（），它會創建兩個。

我建議你的是修改你的模型。你爲什麼不創建一個模型'Profil'，並把一個布爾字段'is_customer'True/False？