Django 自定义字段：仅对数据库中的值运行 to_python() ？

Question

如何确保仅当从数据库加载字段中的数据时才调用自定义字段的 *to_python()* 方法？

我正在尝试使用自定义字段来处理 Base64单个模型属性的编码/解码。一切似乎都正常工作，直到我实例化模型的新实例并使用其明文值设置此属性...此时，Django 尝试解码该字段但失败，因为它是明文。

自定义字段实现的吸引力在于，我认为我可以处理 100% 的编码/解码逻辑，因此我的代码的其他部分不需要了解它。我做错了什么？

（注意：这只是一个说明我的问题的示例，我不需要关于应该或不应该如何使用 Base64 编码的建议）

def encode(value):
    return base64.b64encode(value)

def decode(value):
    return base64.b64decode(value)


class EncodedField(models.CharField):
    __metaclass__ = models.SubfieldBase

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

    def get_prep_value(self, value):
        return encode(value)

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

class Person(models.Model):
    internal_id = EncodedField(max_length=32)

...当我在交互式外壳。为什么在这里调用 to_python() ？

>>> from myapp.models import *
>>> Person(internal_id="foo")
Traceback (most recent call last):
  File "<console>", line 1, in <module>
  File "/usr/local/lib/python2.6/dist-packages/django/db/models/base.py", line 330, in __init__
    setattr(self, field.attname, val)
  File "/usr/local/lib/python2.6/dist-packages/django/db/models/fields/subclassing.py", line 98, in __set__
    obj.__dict__[self.field.name] = self.field.to_python(value)
  File "../myapp/models.py", line 87, in to_python
    return decode(value)
  File "../myapp/models.py", line 74, in decode
    return base64.b64decode(value)
  File "/usr/lib/python2.6/base64.py", line 76, in b64decode
    raise TypeError(msg)
TypeError: Incorrect padding

我原以为我能够做这样的事情...

>>> from myapp.models import *
>>> obj = Person(internal_id="foo")
>>> obj.internal_id
'foo'
>>> obj.save()
>>> newObj = Person.objects.get(internal_id="foo")
>>> newObj.internal_id
'foo'
>>> newObj.internal_id = "bar"
>>> newObj.internal_id
'bar'
>>> newObj.save()

...我做错了什么？

Answer 1

（来自 http://davidcramer.posterous.com/代码/181/custom-fields-in-django.html
和 https:// /docs.djangoproject.com/en/dev/howto/custom-model-fields/#converting-database-values-to-python-objects)

看来你需要能够测试如果它是一个实例，问题是它们是相同的类型（字符串与 b64 编码的字符串）。所以除非你能确定差异，否则我建议确保你总是：

Person(internal_id="foo".encode('base64', 'strict'))

或

Person(internal_id=base64.b64encod("foo"))

或一些这样的编码。

编辑：-我正在查看https://github.com/django-extensions/django-extensions/blob/f278a9d91501933c7d51dffc2ec30341a1872a18/django_extensions/db/fields/encrypted.py并认为你可以做类似的事情。

Answer 2

我有完全相同的问题，但是使用 JSON 数据。我想将数据以JSON格式存储在数据库中。但是，如果您尝试存储已序列化的 JSON 对象，它将以反序列化方式返回。所以问题是，进来的并不总是出来的。特别是如果您尝试将数字存储为字符串，它将作为 int 或 float 返回，因为它在存储之前由 to_python 反序列化。

解决方案很简单，但不太优雅。只需确保将数据的“类型”与数据一起存储，在我的例子中，它是 JSON 数据，所以我用“json:”作为前缀，因此您始终知道数据是否来自数据库。

def get_prep_value(self, value):
    if value is not None:
        value = "json:"+json.dumps(value)
    return value
def to_python(self, value):
    if type(value) in (str, unicode) and value.startswith("json:"):
        value = value[5:]
        try:
            return json.loads(value)
        except:
            # If an error was raised, just return the plain value
            return value
    else:
        return value

话虽这么说，但令人烦恼的是，您不能期望一致的行为，或者您无法判断 to_python 是对用户设置的值还是对数据库中的值进行操作。

Answer 3

当您第一次为字段分配值时，是否只会收到 TypeError ？您可以在它周围写一个 try/ except ：

def to_python(self, value):
  try:
   return decode(value)
  except TypeError:
   return value

这不是最干净的解决方案，但您可以尝试一下，看看它是否可以让您按照您期望的方式处理该字段。