Question

我有一个从django片段复制的base64字段。

https://djangosnippets.org/snippets/1669/

class Base64Field(models.TextField):
    """
    https://djangosnippets.org/snippets/1669/
    Example use:
    class Foo(models.Model):
        data = Base64Field()

    foo = Foo()
    foo.data = 'Hello world!'
    print foo.data # will 'Hello world!'
    print foo.data_base64 # will print 'SGVsbG8gd29ybGQh\n'

    """

    def contribute_to_class(self, cls, name):

        if not self.db_column:
            self.db_column = name

        self.field_name =name+ '_base64'
        super(Base64Field, self).contribute_to_class(cls, self.field_name)
        setattr(cls, name, property(self.get_data, self.set_data))

    def get_data(self, obj):
        return base64.decodestring(getattr(obj, self.field_name))

    def set_data(self, obj, data):
        setattr(obj, self.field_name, base64.encodestring(data))

    def deconstruct(self):
        ame, path, args, kwargs = super(Base64Field, self).deconstruct()
        from pprint import pprint
        pprint(vars(self))
        return ame, path, args, kwargs

我在迁移此字段时遇到问题例如

class EmailStatus(models.Model):
    attachment = Base64Field(null=True, blank=True, db_column='attachment', name="attachment", verbose_name="attachment")

我在迁移时遇到的错误是

raise FieldDoesNotExist('%s has no field named %r' % (self.object_name, field_name))

django.core.exceptions.FieldDoesNotExist: EmailStatus has no field named u'attachment'

现在我可以看到为什么会这样。但是无法找到解决方法。我想我可能需要在解构领域改变一些东西。我为此尝试过多种方法，但所有这些都破了。

e.g。删除_base64。保存和检索数据时，它不起作用。

我尝试更改迁移文件中的名称，但它不起作用。

class Migration(migrations.Migration):

    initial = True

    dependencies = [
        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
    ]

    operations = [
        migrations.CreateModel(name='EmailStatus',
            fields=[('attachment', gradsite.gradnotes.models.Base64Field(blank=True, null=True)),])]

我认为由于contrib_to_class中名称的更改，迁移自动检测程序变得混乱。我不确定什么可以解决。

Answer 1

class Base64Field(models.TextField):
    def contribute_to_class(self, cls, name, private_only=False):
        if not self.db_column:
            self.db_column = name

        self.field_name = name + '_base64'
        super().contribute_to_class(cls,
                                    name)
        setattr(cls, self.field_name, property(self.get_data, self.set_data))

    def get_data(self, obj):
        return base64.b64encode(getattr(obj, self.name).encode('utf-8'))

    def set_data(self, obj, data):
        setattr(obj, self.field_name, base64.b64decode(data).decode('utf-8'))

这似乎有效。 self.field_name中的name和contribute_to_class之间存在混淆，导致使用了错误的值（因此makemigrations第二次/当使用时没有拾取字段migrate）。

我已经进行了python3特定更改，即super调用和base64函数的使用。 set_data方法可能是错误的（我没有考虑太多，因为您可能正在使用python2并且编码会有所不同），但迁移工作正常。

已添加奖励：private_only方法中缺少contribute_to_class参数。

这是我得到的：

from test_app import models

e = models.EmailStatus()
e.attachment = "Hello world!"

e.attachment  # Prints 'Hello world!'
e.attachment_base64  # Prints b'SGVsbG8gd29ybGQh'

Answer 2

有用的链接。

https://code.djangoproject.com/ticket/24563

有用的提示：

失败的迁移：   https://github.com/codefisher/djangopress/blob/master/djangopress/forum/migrations/0011_auto_20150426_1821.py

模特：   https://github.com/codefisher/djangopress/blob/master/djangopress/forum/models.py

这对其他人可能有帮助。

如何迁移django自定义base64字段。字段不存在

2 个答案: