Django使用自定义排序规则进行查询

Question

是否可以使用与数据库表不同的排序规则进行查询？

Answer 1

使用extra()有点麻烦。 现在可以使用Func()表达式实现类似的功能（自 Django 1.8 起）：

username_ci = Func(
    'username',
    function='utf8_general_ci',
    template='(%(expressions)s) COLLATE "%(function)s"')

这可以在annotate() ：

User.objects.annotate(uname_ci=username_ci).filter(uname_ci='joeblow').exists()

或者在order_by()在排序时覆盖默认的排序规则：

User.objects.order_by(username_ci)

现在，它看起来仍然很乱，但是如果您查看Func()的文档和代码，您会发现将其子类化并制作可重用的排序规则设置器非常容易。

我在 Postgres 数据库中使用了这个技巧。

Answer 2

以下是如何使用特定的排序规则而不是给定表/列的默认排序规则。 我假设您总是希望它不区分大小写 utf8_general_ci，但是您可以轻松地在代码中更改它或将其添加为变量。

请注意使用 params kwarg 而不是 db 文字函数。 参数存在的目的完全相同。

def iexact(**kw):
    fields = [['%s=%%s collate utf8_general_ci'%field,value] for (field,value) in kw.items()]
    return dict(where=[f[0] for f in fields], params=[f[1] for f in fields])

if User.objects.extra(**iexact(username='joeblow')).exists():
    status = "Found a user with this username!"

Answer 3

我用一点技巧解决了这个问题；

Django的extra方法和raw方法一样，都是直接使用查询语句；

MyModel.objects.extra(where=["name LIKE '%%" + name + "%%' COLLATE utf8_general_ci"])

但是像这个 sql 注入是可能的。 我们需要转义name变量。 我搜索了很多只为 db 转义字符串的函数。 在MySQL-python包中找到了一个，但它无法转义 unicode 字符串。 包也有literal方法connection但要使用它，我们需要一个实例（也许它是为了 db 特性）。

最后我使用了 Django 的db.connection.cursor 。

from django.db import connection
cursor = connection.cursor()
name = cursor.db.connection.literal(name)[1:-1]  # [1:-1] excluding quotes

通过这种方式，我们还需要一个实例，但我想这不需要数据库连接。 而且我想这种方法与数据库无关。 如果我错了，请纠正我。

Answer 4

上述解决方案有效。 如果获得相反的顺序，请使用以下代码段

sort_value = sort.strip()
if sort_value in ['name', '-name']:
    sort = Func('name', function='C', template='(%(expressions)s) COLLATE "%(function)s"')
if sort_value in ['-name']:
    f_res = queryset.order_by(sort).reverse()
else:
    f_res = queryset.order_by(sort)
return f_res