在 Django 中批量创建自动字段

Question

我有两个模型，我想将所有数据从一个模型迁移到另一个模型。为了模拟问题，假设以下模型：

from django.db import models

class Book:
    name = models.CharField(max_length=100)
    is_archived = models.BooleanField(default=False)
    created_at = models.DateTimeField(auto_now_add=True)


class BookArchived:
    name = models.CharField(max_length=100)
    created_at = models.DateTimeField(auto_now_add=True)

迁移的代码是：

book_objs = []
for archived_book in BookArchived.objects.all():
    book_objs.append(Book(name=archived_book.name, is_archived=True, created_at=archived_book.created_at))

Book.objects.bulk_create(book_objs)

这段代码的问题在于，虽然我明确设置了created_at 字段（created_at=archived_book.created_at），但 Django 将所有这些字段都插入了当前时间。

我知道，如果明确设置了 pk 字段 (id=archived_book.id)，新对象将保留原始时间戳。但是数据库中已经有重叠的ID，所以这种方法会产生另一个问题。如何保留 BookArchived 实例的原始 created_at 值？

Answer 1

我发现这个 gem https://stackoverflow.com/a/59898220/519995 应该可以工作，但我自己没有测试过。

from contextlib import contextmanager

@contextmanager
def suppress_auto_now(model, field_names):
    """
    From https://stackoverflow.com/a/59898220/519995
    idea taken here https://stackoverflow.com/a/35943149/1731460
    """
    fields_state = {}
    for field_name in field_names:
        field = model._meta.get_field(field_name)
        fields_state[field] = {'auto_now': field.auto_now, 'auto_now_add': field.auto_now_add}

    for field in fields_state:
        field.auto_now = False
        field.auto_now_add = False
    try:
        yield
    finally:
        for field, state in fields_state.items():
            field.auto_now = state['auto_now']
            field.auto_now_add = state['auto_now_add']

像这样使用它：

with suppress_autotime(Book, ['created_at']):
        Book.objects.bulk_create(book_objs)

注意： 不要在您的视图/表单或 Django 应用程序的任何地方使用此上下文管理器。此上下文管理器更改字段的内部状态（通过临时将auto_now 和auto_now_add 设置为False）。这将导致 Django 在为并发请求（即相同的进程，不同的线程）执行上下文管理器的主体期间不使用timezone.now() 填充这些字段。虽然这可用于独立脚本（例如管理命令、数据迁移），但它们与 Django 应用程序不在同一进程中运行。

Answer 2

根据DateField 的documentation 中的注释：

按照目前的实现，将 auto_now 或 auto_now_add 设置为 True 将导致该字段设置为 editable=False 和 blank=True。

由于editable 将设置为False，因此向模型提供任何值都将不起作用。您需要在多次迁移中执行此步骤。

先有没有auto_now_add的模型：

class Book(models.Model):
    name = models.CharField(max_length=100)
    is_archived = models.BooleanField(default=False)
    created_at = models.DateTimeField()

使用python manage.py makemigrations为此生成迁移。接下来我们要做一个data migration [Django docs]。首先运行python manage.py makemigrations --empty <yourappname>，这将创建一个空迁移，您将对其进行编辑以将数据从BookArchived 复制到Book。这看起来像：

from django.db import migrations

def copy_legacy_books(apps, schema_editor):
    # We can't import the Person model directly as it may be a newer
    # version than this migration expects. We use the historical version.
    Book = apps.get_model('yourappname', 'Book')
    BookArchived = apps.get_model('yourappname', 'BookArchived')
    book_objs = []
    for archived_book in BookArchived.objects.all():
        book_objs.append(Book(name=archived_book.name, is_archived=True, created_at=archived_book.created_at))
    
    Book.objects.bulk_create(book_objs)

class Migration(migrations.Migration):

    dependencies = [
        ('yourappname', '0001_initial'),
    ]

    operations = [
        migrations.RunPython(copy_legacy_books),
    ]

现在只需将auto_now_add kwarg 添加到您的日期时间字段并使用python manage.py makemigrations 生成另一个迁移：

class Book:
    name = models.CharField(max_length=100)
    is_archived = models.BooleanField(default=False)
    created_at = models.DateTimeField(auto_now_add=True)