6

我的很多模型都有“公司”模型的外键。每个登录用户都可以是一个或多个公司的一部分(用户 m2m 公司,不为空)。

我希望当前管理员用户在管理员索引页面或基本标题上打开“公司护目镜”,即选择列表,他们可以在其中切换“当前”公司。这样做应该自动应用“公司等于”过滤器 - 对于具有公司外键的模型 - 除了任何其他过滤器。

实现这一目标的最佳方法是什么?

注意:这意味着作为管理界面的舒适功能,在此阶段不需要对模型进行实际保护(客户端视图确实需要,但我可以使用自定义管理器并通过 request.user 进行查找)。

我目前的想法是:

  1. 在会话中存储当前公司。

  2. 使用中间件从会话中查找当前公司,并将公司附加到所有相关链接:

    a) change_list: (?/&)"company__eq=42"

    b) change_view "add?company=42" 用于具有公司外键的模型。

    这可能需要对 URL 进行反向或模式匹配以找出它们的模型并检查它是否存在外键(或者我可能会事先准备该列表以提高性能)。

  3. 在每个 ModelAdmin 表单中包含外键字段,但通过 CSS 将其隐藏,以便 change_view add ("new") 包含来自链接的预设外键值而不提及它。

你觉得这是一种可行的方法吗?

如果http://code.djangoproject.com/ticket/10761实现了,我想我可以指定一个自定义查询集,它从 request.session 读取当前公司并完成它。也许更好地快速跟踪(=制作并提交补丁)该票?

编辑:或者也许只是在每个需要它/具有外键的 ModelAdmin 上重新定义 queryset() 方法?

4

2 回答 2

4

我的投票是 overriding ModelAdmin.queryset,因为您可以方便地访问那里的请求。覆盖save_model第 3 点。

 class MyModelAdmin(admin.ModelAdmin):
    def queryset(self, request):
        qs = super(MyModelAdmin, self).queryset(request)
        if request.session.get('company_goggles'):
             return qs.filter(company=request.session['company_goggles'])
        return qs

如果您有很多模型,我会将其子类ModelAdmin化为类似的东西GogglesAdmin并定义一个字段/默认值以从中提取字段名称以及公司的预保存自动注入。

class CompanyGogglesAdmin(admin.ModelAdmin):
    def queryset(self, request):
        qs = super(CompanyGoggleAdmin, self).queryset(request)
        if request.session.get('company_goggles'):
            return qs.filter(**{ getattr(self, 'company_field', 'company') : 
                          request.session['company_goggles'] })

顺便说一句,我真的很喜欢这个“公司护目镜”的术语。

于 2011-03-17T20:41:21.830 回答
1

回答最后一个问题:如果您只想显示查询集中的某些项目,您可以覆盖ModelAdmin'queryset()方法而不会出现问题。例如,如果您要在当前会话中设置公司。您还可以覆盖该save_model()方法以company ForeignKey在保存表单时始终指向用户的公司:

class MyAdmin(admin.ModelAdmin):
    def queryset(self, request):
        company = request.session.get('company', None)
        qs = self.model._default_manager.get_query_set()
        if not company is None:
            qs = qs.filter(company=company)
        return qs

    def save_model(self, request, obj, form, change):
        company = request.session.get('company', None):
        instance = form.save(commit=False)
        if not change or not instance.company:
            instance.company = company
        instance.save()
        form.save_m2m()
        return instance
于 2011-03-17T20:42:10.367 回答