如何根据 request.user 覆盖 django 管理表单外键

Question

---

admin.py

class PromoAdmin(admin.modelAdmin)
      list_display = ( 'name', 'id', 'category', 'promo_type', 'store', 'brand', 'date_start' )
      form = SampleForm

forms.py

class SampleForm(forms.ModelForm):
class Meta:
     model = Promo

def __init__(self, request *args, **kwargs):
    super(PromoAdminForm, self).__init__(*args, **kwargs)
    self.fields["store"].queryset = Store.objects.filter(onwer=request.user)

请求出错

Django 版本：1.3.1 异常类型：TypeError 异常值：
init() 至少需要 2 个参数（给定 1 个）

Answer 1

您不能在字段声明中使用 request.user 启动 store 字段。您可以尝试以下方法：

class MyAwesomeForm(forms.ModelForm):
    store = forms.ModelChoiceField(Store.objects)
    class Meta:
       model = Promo

def __init__(self, user, *args, **kwargs):
    super(MyAwesomeForm, self).__init__(*args, **kwargs)
    self.fields['store'].queryset = Store.objects.filter(owner=user)

在实例化表单时，您可以传递 request.user 对象。

myform = MyAwesomeForm(request.user)

如果你想在管理员中实现这一点，你可以试试this 为了在管理员中仅提供与登录用户相关的对象，提供了覆盖 ModelAdmin.queryset 函数的可能性：

class MyModelAdmin(admin.ModelAdmin):
    form = MyAwesomeAdminForm()
    def queryset(self, request):
        qs = super(MyModelAdmin, self).queryset(request)
        if request.user.is_superuser:
            return qs
        return qs.filter(store__owner=request.user)

class MyAwesomeAdminForm(forms.ModelForm):
    class Meta:
       model = Promo

请注意，store__owner 仅在您的促销模型中存储有外键字段时才有效：

class Promo(models.Model):
    store = models.ForeignKey(Store)

class Store(models.Model):
    owner = models.ForeignKey(User)

我认为也应该可以以某种方式将请求传递给表单的 init 方法。但是没有找到合适的方法来做。