django - 如何从现场添加经理

Question

我想要做的是，当某个模型使用我的字段时，它会自动将自定义管理器添加到该模型。

据我所知，contibute_to_class提供了这样的功能

class MyCustomField(CharField):
    def contribute_to_class(self, cls, name):
        super(MyCustomField, self).contribute_to_class(cls, name)
        setattr(cls, 'custom_manager', CustomManager())

问题是，在我的自定义管理器中，我使用 self.model._default_manager 对默认管理器进行查询，但是当我尝试这样做时，django 说 AttributeError: 'NoneType' object has no attribute '_default_manager'

如果我不使用contribute_to_class并在我的模型类旁边编写自定义管理器，它会按预期工作。可能是什么问题？

Answer 1

管理器，就像字段一样，有一个contribute_to_class方法，如果你不调用它，它们将无法正确设置。正确的调用方法是使用 Model.add_to_class:

class MyCustomField(CharField):
    def contribute_to_class(self, cls, name):
        super(MyCustomField, self).contribute_to_class(cls, name)
        cls.add_to_class('custom_manager', CustomManager())