django model select_related 或 prefetch_related 子 model

Question

新手，我有兩個型號，如下

class ReceipeMaster(models.Model):
    receipe_type = models.CharField(max_length=50, choices=TYPE_OPTIONS, default=TYPE_OPTIONS[0])
    item = models.ForeignKey(Item, on_delete=models.CASCADE, related_name='receipe')
    units = models.IntegerField(default=1)
    status = models.BooleanField(default=True)

class ReceipeDetail(models.Model):
    master = models.ForeignKey(ReceipeMaster, on_delete=models.CASCADE, related_name='items')
    item_type = models.ForeignKey(Item_type, null=True, on_delete=models.PROTECT)
    item = models.ForeignKey(Item, on_delete=models.PROTECT)
    quantity = models.IntegerField()

我有一個詳細的視圖DetailView

class ReceipeDetailView(PermissionRequiredMixin, DetailView):
    permission_required = 'item_management.view_receipemaster'
    model = ReceipeMaster
    template_name = 'receipes/show.html'
    context_object_name = 'receipe'

我想在ReceipeMaster model 和ReceipeDetail select_related上prefetch_related相關或選擇相關。 只需在兩個模型上預取即可。

問候

Answer 1

要自定義如何在通用視圖中獲取查詢集，需要覆蓋視圖的get_queryset方法。 此外，如果想在預取的 object 上使用select_related ，則應該使用Prefetch objects [Django docs]並指定它們的queryset 。 由於這是一個DetailView並因此處理一個 object 您將需要覆蓋get_object ：

from django.db.models import Prefetch
from django.http import Http404
from django.utils.translation import gettext as _

class ReceipeDetailView(PermissionRequiredMixin, DetailView):
    permission_required = 'item_management.view_receipemaster'
    model = ReceipeMaster
    template_name = 'receipes/show.html'
    context_object_name = 'receipe'
    
    def get_object(self, queryset=None):
        # Use a custom queryset if provided; this is required for subclasses
        # like DateDetailView
        if queryset is None:
            queryset = self.get_queryset()

        # Next, try looking up by primary key.
        pk = self.kwargs.get(self.pk_url_kwarg)
        slug = self.kwargs.get(self.slug_url_kwarg)
        if pk is not None:
            queryset = queryset.filter(pk=pk)

        # Next, try looking up by slug.
        if slug is not None and (pk is None or self.query_pk_and_slug):
            slug_field = self.get_slug_field()
            queryset = queryset.filter(**{slug_field: slug})

        # If none of those are defined, it's an error.
        if pk is None and slug is None:
            raise AttributeError(
                "Generic detail view %s must be called with either an object "
                "pk or a slug in the URLconf." % self.__class__.__name__
            )
        queryset = queryset.select_related('item').prefetch_related(
            Prefetch(
                'items',
                 queryset=ReceipeDetail.objects.select_related('item_type', 'item')
            )
        )
        try:
            # Get the single item from the filtered queryset
            obj = queryset.get()
        except queryset.model.DoesNotExist:
            raise Http404(_("No %(verbose_name)s found matching the query") %
                          {'verbose_name': queryset.model._meta.verbose_name})
        return obj

Answer 2

您可以完全覆蓋get_queryset方法，而無需調用super() ：

def get_queryset(self):
    queryset = ReceipeMaster.objects.select_related('item').prefetch_related(
        Prefetch(
            'items',
            queryset=ReceipeDetail.objects.select_related('item_type', 'item'),
        )
    )
    return queryset

然后get_object方法將過濾查詢集以獲取queryset 。 在這種情況下這不會成為問題，因為filter()和select_related()鏈接的順序並不重要（ docs ）。