使用 Django ORM 使用 UNION 指定 SELECT 中的列顺序

Posted 2023-02-24

【中文标题】使用 Django ORM 使用 UNION 指定 SELECT 中的列顺序

【英文标题】：Specify order of columns in SELECT with UNION using Django ORM

【发布时间】：2020-05-06 16:23:09

【问题描述】：

如何在 Django ORM 的 SELECT 查询中指定列的顺序？

我正在尝试合并两个表中的元素，但显然 union 中的元素与 SELECT 中的列顺序匹配，而不是列名（即使列名相同）。

考虑以下模型：

class Person(models.Model):
    first_name = models.CharField(max_length=256)
    last_name = models.CharField(max_length=256)
    age = models.IntegerField()


class Car(models.Model):
    number = models.IntegerField()
    brand = models.CharField(max_length=256)
    name = models.CharField(max_length=256)

以下代码：

Person.objects.create(first_name="John", last_name="Smith", age=25)
Car.objects.create(number=42, name="Cybertruck", brand="Tesla")

q1 = Person.objects.all().annotate(name=F('first_name'), group=F('last_name'), number=F('age')).values(
            'name', 'group', 'number')
q2 = Car.objects.all().annotate(group=F('brand')).values('name', 'group', 'number')

data = q1.union(q2)
print(data.query)
assert list(data) == [
    'name': 'John', 'group': 'Smith', 'number': 25,
    'name': 'Cybertruck', 'group': 'Tesla', 'number': 42,
])

如您所见，我在.values() 中输入了正确的顺序。

可以预料的是，union 中的列将按照传递给值的顺序（或按列名）进行匹配，但这就是发生的情况：

SELECT "testunion_person"."first_name" AS "name", "testunion_person"."last_name" AS "group", "testunion_person"."age" AS "number" FROM "testunion_person" UNION SELECT "testunion_car"."name", "testunion_car"."number", "testunion_car"."brand" AS "group" FROM "testunion_car"

在查询中 "testunion_car"."number" 在 "testunion_car"."brand" 之前，这使得 UNION 中的 Car 具有一个值： 'name': 'Cybertruck', 'group': '42', 'number': 'Tesla'

编辑：我使用的是 2.2 (LTS) 版本的 Django

【问题讨论】：

你使用的Django版本是什么

我正在使用 LTS Django 2.2

你用的是什么版本的python？

我不确定这是否相关，因为它更多的是关于 Django，但它是 Python 3.7

【参考方案1】：

除了在annotate() 下指定别名，您还可以直接在values() 下指定别名：

q1 = Person.objects.all().values(
    name=F('first_name'), group=F('last_name'), xnumber=F('age'))
q2 = Car.objects.all().values(
    'name', group=F('brand'), xnumber=F('number'))

我注意到即使在那时，它也没有正确地对字段进行排序。我将number字段重命名为xnumber以避免与同名的模型字段冲突并且所有内容都正确分组。

【讨论】：

另一种修复订单并避免我看到的冲突的方法是调用value() 两次。所以value().value(name=F('first_name'), group=F('last_name'), number=F('age')) 和number 不再冲突

谢谢！这应该是一个公认的答案！

【参考方案2】：

您可以使用.values_list 设置字段的顺序。

qs1 = Person.objects.values_list('name', 'group', 'number')
qs2 = Car.objects.values_list('brand', 'name', 'number')
qs1.union(qs2)

查看docs了解更详细的说明。

【讨论】：

实际上并没有——我的结果仍然是：('Cyber​​truck', 42, 'Tesla') ('John', 'Smith', 25)。这是Django中的错误吗？也许我应该填写一份错误报告...

【参考方案3】：

不是 Django 错误。尽管查询列未按值排序，但查询集显示正确的顺序：

In [13]: print(data)
<QuerySet ['name': 'Cybertruck', 'group': 42, 'number': 'Tesla', 'name': 'John', 'group': 'Smith', 'number': 25]>

这是因为从数据库中获取数据后将对其进行排序。 QuerySet的源代码sn-p：

class QuerySet:
    def __iter__(self):
        """
        The queryset iterator protocol uses three nested iterators in the
        default case:
            1. sql.compiler.execute_sql()
               - Returns 100 rows at time (constants.GET_ITERATOR_CHUNK_SIZE)
                 using cursor.fetchmany(). This part is responsible for
                 doing some column masking, and returning the rows in chunks.
            2. sql.compiler.results_iter()
               - Returns one row at time. At this point the rows are still just
                 tuples. In some cases the return values are converted to
                 Python values at this location.
            3. self.iterator()
               - Responsible for turning the rows into model objects.
        """
        self._fetch_all()
        return iter(self._result_cache)