Question

基本上，我试图通过使用您可以在下面看到的OneToOne模型来获取当前用户类型，并显示不同的模板＆＃34;部件＆＃34;特定于每个型号。我研究了一下，发现了关于type（）和hasattr（）的问题。我想知道是否有任何方法可以在模板中使用它们，或者你有更好的建议吗？谢谢！

models.py

class Type1(models.Model):
user            = models.OneToOneField(User)
company_name    = models.CharField(max_length=100)

    def __unicode__(self):
        return self.company_name

class Type2(models.Model):
user            = models.OneToOneField(User)
first_name      = models.CharField(max_length=30)
second_name     = models.CharField(max_length=30)

    def __unicode__(self):
        return self.first_name

views.py

def Type1Registration(request):
if request.user.is_authenticated():
    return HttpResponseRedirect('/')
if request.method == 'POST':
    form = Type1Form(request.POST)
    if form.is_valid():
        user = User.objects.create_user(username=form.cleaned_data['username'], 
            email=form.cleaned_data['email'], password=form.cleaned_data['password'])
        user.save()
        type1 = Type1(user=user, company_name=form.cleaned_data['company_name'])
        type1.save()
        return HttpResponseRedirect('/')
    else:
        return render(request, 'type1_register.html', {'form': form})
else:
    form = Type1Form()
    context = {'form': form}
    return render(request, 'type1_register.html', context)

def Type2Registration(request):
if request.user.is_authenticated():
    return HttpResponseRedirect('/')
if request.method == 'POST':
    form = Type2Form(request.POST)
    if form.is_valid():
        user = User.objects.create_user(username=form.cleaned_data['username'], 
            email=form.cleaned_data['email'], password=form.cleaned_data['password'])
        user.save()
        type2 = Type2(user=user, first_name=form.cleaned_data['first_name'],
            second_name=form.cleaned_data['second_name'])
        type2.save()
        return HttpResponseRedirect('/')
    else:
        return render(request, 'type2_register.html', {'form': form})
else:
    form = Type2Form()
    context = {'form': form}
    return render(request, 'type2_register.html', context)

Answer 1

虽然你有两种不同的型号。我在这里要考虑的是制作一个模型并按照这样做：

class UserComplete(AbstractUser):
    first_name      = models.CharField(max_length=30)
    second_name     = models.CharField(max_length=30)
    company_name    = models.CharField(max_length=100)
    company         = models.BooleanField(default=False)

    def get_name(self):
        "Returns the person's or business name."
        if self.company:
            return self.company_name
        else:
            return self.first_name

我要提到的一点是，我不完全确定你想要对这些模型做什么，所以我不确定这对你来说确实是最好的解决方案。我还使用了AbstractUser，这可能是你不熟悉的。基本上，它扩展了User类，在我看来，这是扩展User模型的最佳方法。有关此内容的更多信息，请访问：https://docs.djangoproject.com/en/1.10/topics/auth/customizing/#specifying-custom-user-model

然后你有两个选择。对于表单，您可以创建两个单独的表单或一个。这又取决于你想要对模型做什么，以及你想要编辑你的东西的地方，如果必须要改变的话。

以下是两种形式，如果你想这样：

class UserCreateForm(UserCreationForm):    
    class Meta:
        model = UserComplete
        fields = ("email", "username", "password", "first_name", "second_name")

class BusinessCreateForm(UserCreationForm):  
    class Meta:
        model = UserComplete
        fields = ("email", "username", "password", "company_name")

或只有一个：

class CreateForm(UserCreationForm):    
    company_name = forms.CharField(required=False)  
    first_name = forms.CharField(required=False)  
    second_name = forms.CharField(required=False)

    class Meta:
        model = UserComplete
        fields = ("email", "username", "password", "company_name", "first_name", "second_name", "company")

您必须在模板中手动呈现表单，而不是显示您不想要的表单。如果您需要更多相关信息，请阅读此处：https://docs.djangoproject.com/en/1.10/topics/forms/#rendering-fields-manually。使用第一个表单选项，您必须单独处理每个表单，就像在views.py中一样。使用第二个选项，您可以渲染一个表单并使用company中的布尔值来确定需要处理哪些字段。

然而，我想说这完全取决于你正在建设什么。使用第二个选项，您可以在模板中执行{% if request.user.company %}之类的操作。您的原始代码没有该选项。虽然，有一个解决方法。您可以执行：{% if request.user.type1.company_name %}它将检查该值是否存在。在我看来，这有点脏，但它会起作用。 注意：如果值为空（没有名称的商家），则返回false。

如果您需要第二种形式的views.py示例，请告诉我。希望这可以帮助。玩得开心！：）

P.S。我会替换它：

if request.user.is_authenticated():
    return HttpResponseRedirect('/')

login_required。在此处详细了解：https://docs.djangoproject.com/en/1.10/topics/auth/default/#django.contrib.auth.decorators.login_required

Django 1.10 - 每个OneToOne模型的不同模板

1 个答案: