如何使用数据为现有模型中的新模型字段设置默认值

时间:2019-02-24 18:20:26

标签: django django-models

我有一个带有两个字段的小型Django模型。该模型的数据库中已经有数据。

class MetaDataValue(Model):
    metadata_id = models.ForeignKey(MetaData, on_delete=models.CASCADE,)
    value = models.CharField('value', max_length=200,)

我需要添加另一个字段

short_value = models.CharField('short_value', max_length=200,)

我知道执行迁移时会抱怨,因为我没有数据库中现有行的默认值。

是否可以将short_value的默认值设置为数据库/模型中已经存在的value字段中的字符串?

我想这样做是因为我只需要为数据库中的约20行创建一个不同的short_value,并且此字段没有通用的默认值。我宁愿在short_value字段中不要使用“ fred”或“ default”之类的内容,因为某些字段具有数字,某些具有文本,某些具有数字和文本的组合。我还考虑过创建一个属性而不是另一个模型字段,但是没有一种简单的方法可以将value字段转换为short_value字段。

谢谢!

2 个答案:

答案 0 :(得分:3)

您只需将字段添加到模型中并调用:

python3 manage.py makemigrations

Django将提示您为此填写默认值“ 一次性”。例如,如果我在您的模型上运行它,则会看到:

You are trying to add a non-nullable field 'short_value' to metadatavalue without a default; we can't do that (the database needs something to populate existing rows).
Please select a fix:
 1) Provide a one-off default now (will be set on all existing rows with a null value for this column)
 2) Quit, and let me add a default in models.py
Select an option: 1
Please enter the default value now, as valid Python
The datetime and django.utils.timezone modules are available, so you can do e.g. timezone.now
Type 'exit' to exit this prompt
>>> 'default'
Migrations for 'app':
  app/migrations/0002_metadatavalue_short_value.py
    - Add field short_value to metadatavalue

(黑体字是我自己写的东西)。

因此,这些行将以文本default作为值。

“一次性”值不是为模型指定的默认值。它只是一个存储在迁移文件中的值,并已添加到可能已经存在的行中。

我们可以后期编辑迁移文件,并为该列提供默认值,然后运行一个将该值基本上从另一列复制到新列的函数,例如:

# Generated by Django 2.0.2 on 2019-02-24 19:45

from django.db import migrations, models

def copy_value_to_short_value(apps, schema_editor):
    MetaDataValue = apps.get_model('app', 'metadatavalue')
    db_alias = schema_editor.connection.alias
    from django.db.models import F
    MetaDataValue.objects.using(db_alias).all().update(
        short_value=F('value')
    )


class Migration(migrations.Migration):

    dependencies = [
        ('app', '0001_initial_commit'),
    ]

    operations = [
        migrations.AddField(
            model_name='metadatavalue',
            name='short_value',
            field=models.CharField(default='value', max_length=200, verbose_name='short_value'),
            preserve_default=False,
        ),
        migrations.RunPython(copy_value_to_short_value),
    ]

因此,我们定义了类似于Django ORM查询的函数copy_value_to_short_value,然后将migrations.RunPython(copy_value_to_short_value)添加到迁移中应完成的任务中。

您当然应该在运行迁移之前编辑迁移文件 ,因为否则迁移将出现在django_migrations表中,并且Django将迁移视为“完成”。

答案 1 :(得分:0)

类似于威廉的回答: 使用像“default”这样的字符串提供一次性默认值后 ,只需将 SQL 添加到您的迁移操作中即可:

migrations.RunSQL("UPDATE metadatavalue SET short_value = value;")