我开始使用django。 在我的项目中,我正在使用MPTT。
我有一个相当简单的模型。
我将根添加到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:'Some 产品类别1'})
当然这是不可能的,但它确实说明了我想要的东西。
我希望能够过滤TreeChoiseField,只显示一些基于模型属性的分支(在本例中为root)。
由于
答案 0 :(得分:0)
经过多次争抢,我找到了解决方案。 我不认为这是最好的方法,但据我所见,它有效。
我重载了mptt TreeNodeChoiceField并复制了TreeForeignKeyFiltered
在产品中,我更改了
的foreingkeycategory = 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传递给foreingkey,它将过滤管理表单中的可能性。
同样,我不知道这是否是最好的方法。当然它可以更干净......