Question

有人能给我一个扩展ForeignKey模型字段的例子吗？我试过这样：

class ForeignKeyField(forms.ModelChoiceField):
    def __init__(self, *args, **kwargs):
        super(ForeignKeyField, self).__init__(Chain.objects.all(), *args, **kwargs)
    def clean(self, value):
        return Chain.objects.get(pk=value)      

class CustomForeignKey(models.ForeignKey):
    description = "key from ndb"
    __metaclass__ = models.SubfieldBase

    def __init__(self, *args, **kwargs):
        super(CustomForeignKey, self).__init__(*args, **kwargs)


    def db_type(self, connection):
        return "ndb"

    def to_python(self, value):
        # import pdb; pdb.set_trace()
        from google.appengine.api.datastore_types import Key

        if isinstance(value, Key) is True:
            return value.id()
        if value is None:
            return
        return value

    def get_db_prep_save(self, value, connection, prepared=False):
        save_value = ndb.Key(API_Chain, value.id).to_old_key()
        return save_value

    def formfield(self, **kwargs):
        return models.Field.formfield(self,ForeignKeyField, **kwargs)

我不知道为什么，但如果我使用__metaclass__ = models.SubfieldBase，则使用None值调用to_python，并且它表示外键不能为null。如果我继承自models.Field它可以工作，但不能作为外键。我想看看如何扩展models.ForeignKey的功能。感谢。

Answer 1

你需要SubfieldBase吗？它在幕后做了一些魔术，因此该字段有一个调用to_python的描述符。 ForeignKey有不同类型的描述符。我想，子字段描述符会覆盖ForeignKey描述符。换句话说，它们是不兼容的。

扩展ForeignKey的自定义模型字段示例

1 个答案: