[python] Django ModelForm에서 ForeignKey 선택을 어떻게 필터링합니까?

이것은 간단하며 Django 1.4에서 작동합니다.

class ClientAdminForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(ClientAdminForm, self).__init__(*args, **kwargs)
        # access object through self.instance...
        self.fields['base_rate'].queryset = Rate.objects.filter(company=self.instance.company)

class ClientAdmin(admin.ModelAdmin):
    form = ClientAdminForm
    ....

Django는 이미 ModelAdmin에서이 내장 메소드를 문서에서 제공하므로 폼 클래스에서이를 지정할 필요는 없지만 ModelAdmin에서 직접 지정할 수 있습니다.

ModelAdmin.formfield_for_foreignkey(self, db_field, request, **kwargs)¶
'''The formfield_for_foreignkey method on a ModelAdmin allows you to
   override the default formfield for a foreign keys field. For example,
   to return a subset of objects for this foreign key field based on the
   user:'''

class MyModelAdmin(admin.ModelAdmin):
    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if db_field.name == "car":
            kwargs["queryset"] = Car.objects.filter(owner=request.user)
        return super(MyModelAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)

예를 들어 사용자가 액세스 할 수있는 프런트 엔드 관리 인터페이스를 만드는 등의 방법으로 ModelAdmin을 서브 클래 싱 한 다음 아래 방법을 변경하면됩니다. 최종 결과는 사용자와 관련된 콘텐츠 만 표시하고 수퍼 유저는 모든 것을 볼 수있는 사용자 인터페이스입니다.

네 가지 방법을 재정의했으며 처음 두 가지 방법으로 사용자가 아무것도 삭제할 수 없으며 관리 사이트에서 삭제 버튼도 제거합니다.

세 번째 재정의는 참조 (예 : ‘user’또는 ‘porcupine'(예시))가 포함 된 모든 쿼리를 필터링합니다.

마지막 재정의는 모델의 외래 키 필드를 필터링하여 기본 쿼리 집합과 동일한 선택 항목을 필터링합니다.

이러한 방식으로, 사용자가 자신의 객체를 망칠 수있는 관리하기 쉬운 전면 관리 사이트를 제시 할 수 있으며 위에서 언급 한 특정 ModelAdmin 필터를 입력 할 필요가 없습니다.

class FrontEndAdmin(models.ModelAdmin):
    def __init__(self, model, admin_site):
        self.model = model
        self.opts = model._meta
        self.admin_site = admin_site
        super(FrontEndAdmin, self).__init__(model, admin_site)

‘삭제’버튼을 제거하십시오.

    def get_actions(self, request):
        actions = super(FrontEndAdmin, self).get_actions(request)
        if 'delete_selected' in actions:
            del actions['delete_selected']
        return actions

삭제 권한을 방지합니다

    def has_delete_permission(self, request, obj=None):
        return False

관리 사이트에서 볼 수있는 개체를 필터링합니다.

    def get_queryset(self, request):
        if request.user.is_superuser:
            try:
                qs = self.model.objects.all()
            except AttributeError:
                qs = self.model._default_manager.get_queryset()
            return qs

        else:
            try:
                qs = self.model.objects.all()
            except AttributeError:
                qs = self.model._default_manager.get_queryset()

            if hasattr(self.model, ‘user’):
                return qs.filter(user=request.user)
            if hasattr(self.model, ‘porcupine’):
                return qs.filter(porcupine=request.user.porcupine)
            else:
                return qs

관리 사이트의 모든 외래 키 필드에 대한 선택을 필터링합니다.

    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if request.employee.is_superuser:
            return super(FrontEndAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)

        else:
            if hasattr(db_field.rel.to, 'user'):
                kwargs["queryset"] = db_field.rel.to.objects.filter(user=request.user)
            if hasattr(db_field.rel.to, 'porcupine'):
                kwargs["queryset"] = db_field.rel.to.objects.filter(porcupine=request.user.porcupine)
            return super(ModelAdminFront, self).formfield_for_foreignkey(db_field, request, **kwargs)

답변

class AddPhotoToProject(CreateView):
    """
    a view where a user can associate a photo with a project
    """
    model = Connection
    form_class = CreateConnectionForm


    def get_context_data(self, **kwargs):
        context = super(AddPhotoToProject, self).get_context_data(**kwargs)
        context['photo'] = self.kwargs['pk']
        context['form'].fields['project'].queryset = Project.objects.for_user(self.request.user)
        return context
    def form_valid(self, form):
        pobj = Photo.objects.get(pk=self.kwargs['pk'])
        obj = form.save(commit=False)
        obj.photo = pobj
        obj.save()

        return_json = {'success': True}

        if self.request.is_ajax():

            final_response = json.dumps(return_json)
            return HttpResponse(final_response)

        else:

            messages.success(self.request, 'photo was added to project!')
            return HttpResponseRedirect(reverse('MyPhotos'))

    context['form'].fields['project'].queryset = Project.objects.for_user(self.request.user)

formmodel.base_fields['myfield'].queryset = MyModel.objects.filter(...)

# models.py
class Company(models.Model):
    # ...
class Contract(models.Model):
    company = models.ForeignKey(Company)
    locations = models.ManyToManyField('Location')
class Location(models.Model):
    company = models.ForeignKey(Company)

# admin.py
class LocationInline(admin.TabularInline):
    model = Location
class ContractInline(admin.TabularInline):
    model = Contract
class CompanyAdmin(admin.ModelAdmin):
    inlines = (ContractInline, LocationInline)
    inline_filter = dict(Location__company='self')

class ChangeKeyValueForm(forms.ModelForm):
    _terminal_list = forms.ModelMultipleChoiceField(
queryset=Terminal.objects.all() )

    class Meta:
        model = ChangeKeyValue
        fields = ['_terminal_list', 'param_path', 'param_value', 'scheduled_time',  ]

class ChangeKeyValueAdmin(admin.ModelAdmin):
    form = ChangeKeyValueForm
    list_display = ('terminal','task_list', 'plugin','last_update_time')
    list_per_page =16

    def get_form(self, request, obj = None, **kwargs):
        form = super(ChangeKeyValueAdmin, self).get_form(request, **kwargs)
        qs, filterargs = Terminal.get_list(request)
        form.base_fields['_terminal_list'].queryset = qs
        return form