在 Django 的 ORM 中使用带有 UPDATE 的子查询

Question

假设我有两个模型，我称之为 ModelA 和 ModelB。两种模型都有一些共同的字段（由 field_one、field_two 和 field_three 表示）。此外，ModelB 有一个指向 ModelA 的外键。

class ModelA(Model):
    field_one = models.IntegerField()
    field_two = models.TextField()
    field_three = models.BooleanField()

class ModelB(Model):
    field_one = models.IntegerField()
    field_two = models.TextField()
    field_three = models.BooleanField()
    model_a = models.ForeignKey(ModelA, on_delete=models.CASCADE)

我需要更新 ModelB 的所有实例，以将字段的值更新为关联的 ModelA 实例的值。我需要完全在数据库中执行此操作，而不需要实例化任何模型实例（不使用 .save() 或 bulk_update()）。

我知道如何使用子查询在 PostgreSQL 中完成此操作：

UPDATE model_b SET (field_one, field_two, field_three) =
    (SELECT field_one, field_two, field_three FROM model_a
     WHERE model_b.model_a_id = model_a.id);

如何在 Django 的 ORM 中表达上述查询？

这是我能得到的最接近的：

ModelB.objects.update(
    field_one=Subquery(ModelA.objects.filter(id=OuterRef('model_a_id')).values(field_one)[:1]),
    field_two=Subquery(ModelA.objects.filter(id=OuterRef('model_a_id')).values(field_two)[:1]),
    field_three=Subquery(ModelA.objects.filter(id=OuterRef('model_a_id')).values(field_three)[:1])
})

但是，这会导致每个字段的子查询：

UPDATE model_b SET
    field_one = (SELECT model_a.field_one FROM model_a WHERE model_a.id = model_b.model_a_id LIMIT 1),
    field_two = (SELECT model_a.field_two FROM model_a WHERE model_a.id = model_b.model_a_id LIMIT 1),
    field_three = (SELECT model_a.field_three FROM model_a WHERE model_a.id = model_b.model_a_id LIMIT 1);

Answer 1

通过在模型 B13=] 参数中传递模型 A 作为参数，模型和模型将继承模型 A 的每个字段

class ModelA(Model):
    field_one = models.IntegerField()
    field_two = models.TextField()
    field_three = models.BooleanField()

class ModelB(ModelA):
    pass

class ModelC(ModelA):
    pass

Answer 2

不幸的是，ORM 不支持注释分布在多个列中，我不知道它的功能请求。

如果你想坚持使用 ORM，你将不得不承受可能的性能损失（也许 PostgreSQL 足够聪明，只使用一个；EXPLAIN 会告诉你）否则 you'll have to switch to .raw SQL .

在 Django 的 ORM 中使用带有 UPDATE 的子查询

Using subqueries with UPDATE in Django's ORM

python

django

django-orm

django-queryset