Как ограничить выбор вариантов ForeignKey для поля идентификатора Django raw

метод ниже работает для меня, но это queryset, который влияет на каждого администратора, который должен использовать модель клиента. Но если у вас есть другой администратор, например счет, который требует другого queryset, вы можете немного поэкспериментировать с прокси-сервером модели.

модель

class Customer(models.Model):
    name = models.CharField(max_length=100)
    is_active = models.BooleanField()

class Order(models.Model):
    cust = models.ForeignKey(Customer)

Admin

class CustomerAdmin(admin.ModelAdmin):         
    def queryset(self, request):
        qs = super(CustomerAdmin, self).queryset(request)           
        return qs.filter(is_active=1)

class OrderAdmin():     
    raw_id_fields = ('cust', )    

Я нахожу данное решение (настройка ModelAdmin queryset) немного слишком ограничительный, для реалистичных проектов.

то, что я делаю, обычно следующее:

• создайте пользовательский фильтр в my ModelAdmin (например, подклассы admin.SimpleListFilter см. doc)

• создать подкласс виджет ForeignKeyRawIdWidget следующим образом:

  class CustomRawIdWidget(ForeignKeyRawIdWidget):
  
      def url_parameters(self):
          """
          activate one or more filters by default
          """
  
          res = super(CustomRawIdWidget, self).url_parameters()
  
          res["<filter_name>__exact"] = "<filter_value>"
  
          return res
  

  обратите внимание, что единственное, что делает пользовательский виджет, - это "предварительный выбор" фильтра, который turn, отвечает за" ограничение " queryset

• используйте пользовательский виджет:

  class MyForm(forms.ModelForm):
  
      myfield = forms.ModelChoiceField(queryset=MyModel.objects.all(),
          ...
          widget=CustomRawIdWidget(
               MyRelationModel._meta.get_field('myfield').rel,
               admin.site))
  

одним из слабых мест этого подхода является то, что фильтр, выбранный виджетом, не препятствует выбору другого экземпляра из этой модели. Если это необходимо, я переопределяю метод ModelAdmin.save_model(...) (см. doc), чтобы проверить, что связанные экземпляры являются только разрешенными.

Я нахожу этот подход немного более сложный, но гораздо более гибкий, чем ограничение набора запросов для всего ModelAdmin.

Если вам нужно отфильтровать всплывающее окно raw_id list_view на основе экземпляра модели, вы можете использовать пример ниже:

1. Написать пользовательский виджет

class RawIdWidget(widgets.ForeignKeyRawIdWidget):

    def url_parameters(self):
        res = super(RawIdWidget, self).url_parameters()
        object = self.attrs.get('object', None)
        if object:
            # Filter variants by product_id
            res['product_id'] = object.variant.product_id
        return res

2. Передать экземпляр на форму init

class ModelForm(forms.ModelForm):

    def __init__(self, *args, **kwargs):
        super(ModelForm, self).__init__(*args, **kwargs)
        obj = kwargs.get('instance', None)
        if obj and obj.pk is not None:
            self.fields['variant'].widget = RawIdWidget(
                rel=obj._meta.get_field('variant').rel,
                admin_site=admin.site,
                # Pass the object to attrs
                attrs={'object': obj}
            )

Я создал генетическое решение для решения проблемы пользовательских параметров для передачи во всплывающее окно. Вам просто нужно скопировать этот код в свой проект:

from django.contrib.admin import widgets

class GenericRawIdWidget(widgets.ForeignKeyRawIdWidget):
    url_params = []

    def __init__(self, rel, admin_site, attrs=None, \
        using=None, url_params=[]):
        super(GenericRawIdWidget, self).__init__(
            rel, admin_site, attrs=attrs, using=using)
        self.url_params = url_params

    def url_parameters(self):
        """
        activate one or more filters by default
        """
        res = super(GenericRawIdWidget, self).url_parameters()
        res.update(**self.url_params)

        return res

тогда вы можете использовать следующее:

field.widget = GenericRawIdWidget(YOURMODEL._meta.get_field('YOUR_RELATION').rel,
            admin.site, url_params={"<YOURMODEL>__id__exact":     object_id})

я использовал его таким образом:

class ANSRuleInline(admin.TabularInline):
    model = ANSRule 
    form = ANSRuleInlineForm
    extra = 1
    raw_id_fields = ('parent',)

    def __init__(self, *args, **kwargs):
        super (ANSRuleInline,self ).__init__(*args,**kwargs)

    def formfield_for_dbfield(self, db_field, **kwargs):
        formfield = super(ANSRuleInline, self).formfield_for_dbfield(db_field, **kwargs)
        request = kwargs.get("request", None)
        object_id = self.get_object(request)

        if db_field.name == 'parent':
            formfield.widget = GenericRawIdWidget(ANSRule._meta.get_field('parent').rel,
                admin.site, url_params={"pathology__id__exact": object_id})

        return formfield

    def get_object(self, request):
        object_id = request.META['PATH_INFO'].strip('/').split('/')[-1]
        try:
            object_id = int(object_id)
        except ValueError:
            return None
        return object_id

когда я использую GenericRawIdWidget, Я передаю дикт url_params, который будет использоваться на url.