python - Django 覆盖 bulk_create

Question

如何覆盖 bulk_create 方法？我试试这个

class SomeModel(models.Model):
    field = models.CharField()

    def bulk_create(self, objs, batch_size=None):
        #do something
        return super(SomeModel, self).bulk_create(objs, batch_size)

但它不起作用。当我运行这段代码时

SomeModel.objects.bulk_create(objects_list)

它会创建新对象，但不会使用我的重写 bulk_create 方法。是否可以覆盖 bulk_create？如何？

Answer 1

bulk_create 是 Manager 类的一个方法，SomeModel.objects 是 Manager 的一个实例。您需要继承 Manager 并覆盖那里的方法，然后将管理器添加到 SomeModel:

class SomeModelManager(models.Manager):
    def bulk_create(self, objs, batch_size=None, ignore_conflicts=False):
        ...

class SomeModel(models.Model):
    objects = SomeModelManager()

请参阅 custom managers 上的文档想要查询更多的信息。