为什么 django queryset.extra() 会抛出 OperationalError: (1242, 'Subquery returns more than 1 row')？

Question

我将 Django ORM 与 MySQL 一起使用，并在查询上浪费了大量时间。对于一些“高级”案例，我决定使用原始查询，因为对于那些，我无法使用注释。原始查询的问题是不要像注释或聚合一样向查询集添加“字段”。所以，我正在使用 额外。但现在我面临一个问题：

qs_products = Productos.objects.all()
qs_productos.extra({
    "stock":
    """
        SELECT SUM(items.cantidad)
        FROM 
            `encargosProveedor_listado_articulos` AS encargos, 
            `itemArticulosProveedor`AS items, `articulos` as articulos
        WHERE 
            encargos.itemarticulosproveedor_id=items.id and 
            articulos.id=items.articulos_id
        GROUP BY articulos.producto_id
    """
    })

这是直接来自我的数据库管理员的此查询的结果：

+---------------------+
| SUM(items.cantidad) |
+---------------------+
|          14         |
+---------------------+
|          4          |
+---------------------+

但是当使用 extra()

在 django 下运行这段代码时

MySQLdb._exceptions.OperationalError: (1242, '子查询返回超过 1 行')

返回多行有什么问题？查询返回两行，因为我有两个产品，这是合理的。我想将 stock 分配给每个产品。

替代方案？建议？提示？

Answer 1

您正在尝试在外部查询和extra 子查询之间进行连接，而数据库需要显式连接子句来执行此操作。我相信您可以在 extra 中添加 WHERE 子句以使其正常工作：

qs_productos.extra({
    "stock":
    """
        SELECT SUM(items.cantidad)
        FROM 
            `encargosProveedor_listado_articulos` AS encargos, 
            `itemArticulosProveedor`AS items, `articulos` as articulos
        WHERE 
            articulos.producto_id = productos.id and
            encargos.itemarticulosproveedor_id=items.id and 
            articulos.id=items.articulos_id       
        GROUP BY articulos.producto_id
    """
})

articulos.producto_id = productos.id 中的productos 需要替换为主查询中的实际表名。

由于 extra 已被弃用（从 Django 2.2 开始还没有完全弃用），因此这里是等效的 RawSQL 查询：

qs_productos.annotate(stock=RawSQL(
    """
        SELECT SUM(items.cantidad)
        FROM 
            `encargosProveedor_listado_articulos` AS encargos, 
            `itemArticulosProveedor`AS items, `articulos` as articulos
        WHERE 
            articulos.producto_id = productos.id and
            encargos.itemarticulosproveedor_id=items.id and 
            articulos.id=items.articulos_id       
        GROUP BY articulos.producto_id
    """,
    ()
))

RawSQL 中的空元组是必需的，因为该函数采用 params 元组参数，即使您没有要传递的参数，该参数也不是可选的。

作为奖励，RawSQL 对先前的values 调用不太敏感，并在需要时提供可选的output_field 参数。