mptt TreeNodeChoiceField 已过滤答案

【问题标题】：mptt TreeNodeChoiceField filteredmptt TreeNodeChoiceField 已过滤
【发布时间】：2013-01-31 15:58:20
【问题描述】：

我开始使用 django。在我的项目中，我使用的是 MPTT。

我有一个相当简单的模型。

类别：（树）
Product1：该类别的引用和外键。

我将根添加到 Category 模型中，只是为了简化对某一特定产品的基本类别的搜索

models.py

class Category(MPTTModel):
    name = models.CharField(max_length=200)
    parent = TreeForeignKey('self', null=True, blank=True, related_name='children')

    @property
    def root(self):
      return self.get_root().name

class Product1(models.Model):
    ref = models.CharField(max_length=200, blank=True) #YVX2311
    category = TreeForeignKey(Category)

admin.py

admin.site.register(Category, MPTTModelAdmin)
admin.site.register(Product)

我想要修改管理表单，以便能够说出类似的话。

models.ForeignKey(Category, limit_choices_to = {root : '一些产品类别1'})

当然这是不可能的，但它确实说明了我想要什么。

我希望能够根据模型上的属性（在本例中为根）过滤 TreeChoiseField 以仅显示一些分支。

谢谢

【问题讨论】：

标签： django mptt

【解决方案1】：

经过一番折腾，我找到了解决办法。我认为这不是最好的方法，但据我所知，它有效。

我重载了 mptt TreeNodeChoiceField 并复制了 TreeForeignKeyFiltered

在产品中，我将外键更改为

category = TreeForeignKeyFiltered(Category, root = 'MotherBoard')

我将 root 关键字添加到 TreeForeingKeyFiltered

class TreeForeignKeyFiltered(models.ForeignKey):
    """
    Extends the foreign key, but uses mptt's ``TreeNodeChoiceField`` as
    the default form field.

    This is useful if you are creating models that need automatically
    generated ModelForms to use the correct widgets.
    """

    def __init__(self, *args, **kwargs):
        self._root = kwargs.pop('root', None)
        super(TreeForeignKeyFiltered, self).__init__(*args, **kwargs)

    def formfield(self, **kwargs):
        """
        Use MPTT's ``TreeNodeChoiceField``
        """
        kwargs.setdefault('root', self._root)
        kwargs.setdefault('form_class', TreeNodeChoiceFieldFiltered)
        return super(TreeForeignKeyFiltered, self).formfield(**kwargs)

最后我将 TreeNodeChoiceFieldFiltered 添加为

class TreeNodeChoiceFieldFiltered(TreeNodeChoiceField):
    def __init__(self, queryset, *args, **kwargs):
        _root = kwargs.pop('root', None)

        if _root is not None:
            pid = None
            for i in queryset:
                if i.root.name == _root:
                    pid = i.tree_id
                    break

            if pid is not None:
                queryset = queryset.filter(tree_id = pid) 

        super(TreeNodeChoiceFieldFiltered, self).__init__(queryset, *args, **kwargs)

现在我可以将关键字 root 传递给将过滤管理表单中的可能性的外键。

同样，我不知道这是否是最好的方法。当然可以更干净....

【讨论】：