Django 以特定顺序获取模型实例时出现问题

发布于04月13日

我有用户的Passport 数据列表.我需要让用户的查询设置与列表中订购的Passport 数据相同的顺序.以下是包含Passport 数据的列表:

lst = ['AA1111111', 'AD2222222', 'AA3333333', 'AA4444444', 'АВ5555555']

我试图做这样的事情:

empty_queryset = Users.objects.none()
for passport_number in lst:
    user = Users.objects.filter(passport_number__exact=passport_number)
    empty_queryset |= user

我期待的是:

<QuerySet [<Users: AA1111111>, <Users: AD2222222>, <Users: AA3333333>, <Users: AA4444444>, <Users: АВ5555555>]>

但它的顺序很混乱:

<QuerySet [<Users: АВ5555555>, <Users: AA1111111>, <Users: AD2222222>, <Users: AA3333333>, <Users: AA4444444>]>

然后我try 了这个:

Users.objects.filter(passport_number__in=[i for i in lst])

但还是没有起作用

推荐答案

两者本质上做的事情是相同的，并发出一个类似于以下的查询:

SELECT *
FROM users
WHERE passport_number IN ('AA1111111',
                          'AD2222222',
                          'AA3333333',
                          'AA4444444',
                          'АВ5555555')

由于没有ORDER BY条款，因此数据库可以按照它想要的任何顺序呈现数据.通常，如果涉及索引，它可以像索引一样排序，但这更多的是一个实现细节.

您可以与.union(…) ^[Django-doc]个人一起工作:

SELECT *
FROM users
WHERE passport_number = 'AA1111111'
UNION
SELECT *
FROM users
WHERE passport_number = 'AD2222222'
-- …

那么看起来像:

Users.objects.none().union(
    *[User.objects.filter(passport_number=p) for p in lst], all=True
)

另一种 Select 是在Django/Python级别对项目进行排序:

lst = ['AA1111111', 'AD2222222', 'AA3333333', 'AA4444444', 'АВ5555555']
order = {k: i for i, k in enumerate(lst)}

items = sorted(
    Users.objects.filter(passport_number__in=lst),
    key=lambda x: order.get(x.passport_number),
)

但因此它是Users中的list，而不是QuerySet，因此无法通过QuerySet API进行额外的过滤、翻页等.