0

我正在尝试实现一个自定义 Django 字段,该字段将在 DB 中保留我的模型类之一的名称(例如“Cat”、“Dog”、“Restaurant”、“SuperPrivateCommercialDataPiece”等)并在何时返回类对象要求:

class Cat(models.Model):
    ...

class SomeDataPiece(models.Model):
    relatedTo = MyGloriousFieldType(null=True)

...

newPiece = SomeDataPiece()
newPiece.relatedTo = Cat
print newPiece.relatedTo # should print <class 'myproj.myapp.models.Cat'>

我真的做到了。我已经 subclassed models.Field, set__metaclass__等:

class MyGloriousFieldType(models.Field):

    description = "No description, just gloriosity."

    __metaclass__ = models.SubfieldBase

    def __init__(self, *args, **kwargs):
        kwargs['max_length'] = 40
        super(BlockTypeField, self).__init__(*args, **kwargs)

    def db_type(self, connection):
        return 'char(40)'

    def to_python(self, value):
        if not value:
            return None
        elif inspect.isclass(value) and issubclass(value, models.Model):
            return value
        elif hasattr(myproj.myapp.models, value):
            return getattr(myproj.myapp.models, value)
        else:
            raise ValidationError("Invalid model name string")

    def get_db_prep_value(self, value, **kwargs):
        if inspect.isclass(value) and issubclass(value, models.Model):
            return value.__name__
        else:
            # it is actually never raised
            raise ValidationError("Invalid value, Model subclass expected")

    def value_to_string(self, instance):
        value = self._get_val_from_obj(obj)
        return self.get_prep_value(value)

在上面的代码中,它按预期工作。我还在 admin 中注册了一些包含此类字段的模型,现在我什至可以创建此类对象;为 MyGloriousFieldType 字段创建文本输入。

当我尝试使用 MyGloriousFieldType 字段编辑已经存在的对象时,危机就开始了。首先,它用“Cat object”而不是“Cat”填充文本字段。其次,当我将其改回“Cat”并单击“保存更改”时,它会出现错误:

TypeError at /admin/myapp/somedatapiece/3/
  unbound method prepare_database_save() must be
  called with Cat instance as first argument (got 
  MyGloriousFieldType instance instead)

那么,我做错了什么?

4

1 回答 1

0

嗯,我已经想通了。问题出在 Django 核心内部。字段不能返回包含某些models.Field属性的 Python 对象,例如prepare_database_save,否则会发生错误。我解决这个问题的方法是返回 tuple(MyClassObject, )而不是MyClassObjectin to_python(). 奇怪,但我能做什么......也许我会尝试在 Django 的 Github 页面上的某个地方发布一个问题。

于 2013-08-12T08:02:38.653 回答