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

2024-10-03 19:28:14 发布

您现在位置:Python中文网/ 问答频道 /正文

如何在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"之前,这使得联合中的汽车具有一个值: {'name': 'Cybertruck', 'group': '42', 'number': 'Tesla'}

编辑:我正在使用Django的2.2(LTS)版本


Tags: namenumberage顺序modelsgroupcarselect
3条回答

不是Django虫。尽管查询列未按值排序,但queryset显示的顺序正确:

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

这是因为从数据库获取数据后,数据将被排序。QuerySet的源代码片段:

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)

您可以直接在values()下指定别名,而不是在annotate()下指定别名:

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,以避免与同名的模型字段冲突,并且所有内容都正确分组

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

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

查看docs以获得更详细的解释

相关问题 更多 >