Django:自定义模型字段在表单中没有显示正确的值

时间:2012-06-28 14:46:18

标签: python django django-forms django-models

我需要在多个数据类型的关系数据库中存储任意数据,所以我想出了一个解决方案,除了存储数据本身还存储数据的数据类型(str,int等) 。这允许在retreival时将存储在db中的字符串转换为数据所适用的任何数据类型。为了存储数据类型,我创建了一个自定义模型字段:

class DataType(object):
    SUPPORTED_TYPES = {
        u'unicode': unicode,
        u'str': str,
        u'bool': bool,
        u'int': int,
        u'float': float
    }
    INVERSE_SUPPORTED_TYPES = dict(zip(SUPPORTED_TYPES.values(), SUPPORTED_TYPES.keys()))
    TYPE_CHOICES = dict(zip(SUPPORTED_TYPES.keys(), SUPPORTED_TYPES.keys()))

    def __init__(self, datatype=None):
        if not datatype:
            datatype = unicode

        t_datatype = type(datatype)

        if t_datatype in [str, unicode]:
            self.datatype = self.SUPPORTED_TYPES[datatype]

        elif t_datatype is type and datatype in self.INVERSE_SUPPORTED_TYPES.keys():
            self.datatype = datatype

        elif t_datatype is DataType:
            self.datatype = datatype.datatype

        else:
            raise TypeError('Unsupported %s' % str(t_datatype))

    def __unicode__(self):
        return self.INVERSE_SUPPORTED_TYPES[self.datatype]

    def __str__(self):
        return str(self.__unicode__())

    def __len__(self):
        return len(self.__unicode__())

    def __call__(self, *args, **kwargs):
        return self.datatype(*args, **kwargs)


class DataTypeField(models.CharField):
    __metaclass__ = models.SubfieldBase
    description = 'Field for storing python data-types in db with capability to get python the data-type back'

    def __init__(self, **kwargs):
        defaults = {}
        overwrites = {
            'max_length': 8
        }
        defaults.update(kwargs)
        defaults.update(overwrites)
        super(DataTypeField, self).__init__(**overwrites)

    def to_python(self, value):
        return DataType(value)

    def get_prep_value(self, value):
        return unicode(DataType(value))

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

所以这让我可以这样做:

class FooModel(models.Model):
    data = models.TextField()
    data_type = DataTypeField()

>>> foo = FooModel.objects.create(data='17.94', data_type=float)
>>> foo.data_type(foo.data)
17.94
>>> type(foo.data_type(foo.data))
float

所以我的问题是在Django Admin(我使用的是ModelAdmin)中,文本框中data_type的值没有正确显示。只要它是float(并且在db中它存储为float,我检查过),显示的值为0.0。对于int,它会显示0。对于bool,它会显示False。 Django实际上调用了它,而不是显示data_type的字符串表示,这意味着__call__会被一个参数调用,从而产生这些值。例如:

>>> DataType(float)()
0.0
>>> DataType(int)()
0
>>> DataType(bool)()
False

我想通过用以下代码替换__call__方法来修补它:

def __call__(self, *args, **kwargs):
    if not args and not kwargs:
        return self.__unicode__()
    return self.datatype(*args, **kwargs)

这会在表单中显示正确的值,但我觉得这不是很优雅。有没有办法让它变得更好?我无法弄清楚Django首先称之为字段值的位置。

感谢名单

1 个答案:

答案 0 :(得分:0)

为什么要调用DataType,请阅读:https://docs.djangoproject.com/en/1.4/topics/templates/#accessing-method-calls

干净的解决方案可能只是将调用重命名为更明确的内容。