在 Django admin 中限制 ManyToMany MultipleSelect 的查询集

Question

我有一个看起来像这样的模型：

class Event(models.Model):
    event_dates = ManyToManyField("EventDate")
    #...

class EventDate(models.Model):
    event_date = DateField()
    #...

但是，在 EventAdmin 中显示 event_dates 的 django admin MultipleSelect 表单字段中，我想将查询集限制为不是过去的 event_dates。

查询集类似于：

event_date_queryset = EventDate.objects.filter(event_date__gte = datetime.date.today()) 

但是我在哪里可以设置这个查询集，以便在字段中只显示非过去日期？

（我目前没有EventAdmin 的自定义表单，但很乐意添加一个。）

Answer 1

你可以试试：

event_dates = models.ManyToManyField("EventDate", limit_choices_to={'event_date__gte': date.today()})

取自https://docs.djangoproject.com/en/dev/ref/models/fields/#django.db.models.ForeignKey.limit_choices_to

但随后仅显示未来的日期，即使过去的某些日期仍与 Event 相关联。

如果您还想要之前与Event 关联的所有日期，您可以按以下方式操作ModelForm

from datetime import date

from django.contrib import admin
from django import forms
from django.db.models import Q

from models import Event, EventDate

class EventAdminForm(forms.ModelForm):
    class Meta:
        model = Event

    def __init__(self, *args, **kwargs):
        super(EventAdminForm, self).__init__(*args, **kwargs)
        if 'event_dates' in self.initial:
            self.fields['event_dates'].queryset = EventDate.objects.filter(Q(pk__in=self.initial['event_dates']) | Q(event_date__gte=date.today()))
        else:
            self.fields['event_dates'].queryset = EventDate.objects.filter(event_date__gte=date.today())

class EventAdmin(admin.ModelAdmin):
    form = EventAdminForm
    filter_horizontal = ['event_dates']

Answer 2

最简单的方法是在管理员中完成

你的模型

class Event(models.Model):
    event_dates = ManyToManyField("EventDate")
    #...

class EventDate(models.Model):
    event_date = DateField()
    #...

现在在您的管理文件中

这有一个额外的好处，它只返回您正在使用反向查找event__id=event_id查看的当前事件的事件日期

class EventAdmin(admin.ModelAdmin):
    def get_field_queryset(self, db, db_field, request):
        """
        If the ModelAdmin specifies ordering, the queryset should respect that
        ordering.  Otherwise don't specify the queryset, let the field decide
        (returns None in that case).
        """
        if db_field.name == 'event_dates':
            event_id = int(request.resolver_match.args[0])

            return db_field.remote_field.model._default_manager.filter(
                              event__id=event_id,
                              event_date__gte = datetime.date.today()
            )

        super().get_field_queryset(db, db_field, request)