如何根据模型的类型为每个不同的子类分配不同的模板

Question

我在我的Django项目中有一些模型是基模型类的子类。例子:

class A(Model):
    pass
class B(A):
    pass
class C(A):
    pass

例如，在网上购物中，我将有Ebook, CD, Ball。我可以用OneToOneField创建OneToOne关系，但我认为在这种情况下，用户必须填写两个表单，首先他可能填写产品字段，然后填写电子书字段之类的表单。我还想在数据库中显示产品，无论是电子书或电影或CD播放器或任何其他类型，在产品列表中列出。所以我决定用这种方式创建模型。问题是在view部分，我必须决定我应该填写哪个模板。下面是我决定解决这个问题的方法。但这个解决方案的问题是，我认为随着产品数量的增长，它并不是真正有用的，这将成为一场噩梦。我想知道如何改进这个解决方案。

我是python和Django的新手。我找了很多，但什么也没找到。

class Product(models.Model):
    product_type = models.CharField(max_length=10, default=self.__class__.__name__, editable=False)
    def getType(self):
        return self.product_type
    def __init__(self, *args, **kwargs):
        models.Model.__init__(self, *args, **kwargs)

class Book(Product):
    def __init__(self, *args, **kwargs):
        Product.__init__(self, *args, **kwargs)
        self.product_type == self.__class__.__name__
@login_required
def edit(request, product_id):
    try:
        product = Product.objects.get(pk=service_id)
    except Product.DoesNotExist():
        return HttpResponseNotFound()
    if product.getType() == 'Ebook':
        product = product.ebook
        formObject = EbookForm
        template = 'products/edit_ebook.html'
    elif product.getType() == 'CD':
        product = product.host
        formObject = CdForm
        template = 'products/edit_cd.html'
    elif product.getType() == 'Dress':
        formObject = DressForm
        template = "products/edit_dress.html"
    if request.method == 'POST':
        form = formObject(request.POST, instance=product)
        if form.is_valid():
            form.save()
            return HttpResponseRedirect(reverse('product_index'))
    else:
        form = formObject(instance=product)
    return render(request, template, {'form' : form })

Answer 1

如果您将模板名称存储在模型中，则可以摆脱所有if语句，因此您可以仅调用render(request, product.template, {'form': form})。

在你的Product模型上，你会有一个名为template的CharField，你会用'book'， 'ebook'等填充它。

这背后的原理被称为"用多态性替换条件"。

您还应该考虑使用抽象基模型，这对性能更好。