如何避免 SQL 围绕 Django 自定义数据库函数调用的括号？

Question

问题的简短介绍...

PostgreSQL 有非常整洁的数组字段（整数数组、字符串数组）和它们的函数，如 UNNEST 和 ANY。
Django 支持这些字段（我为此使用 djorm_pgarray），但原生不支持函数。
可以使用 .extra()，但 Django 1.8 引入了一个新概念 database functions。

让我提供一个最原始的例子，说明我基本上用所有这些做了什么。 Dealer 有一个它支持的品牌列表。 Vehicle 有品牌并与经销商相关联。但是碰巧Vehicle的make和Dealer的make list不匹配，也就是inevitable.

MAKE_CHOICES = [('honda', 'Honda'), ...]

class Dealer(models.Model):
    make_list = TextArrayField(choices=MAKE_CHOICES)

class Vehicle(models.Model):
    dealer = models.ForeignKey(Dealer, null=True, blank=True)
    make   = models.CharField(max_length=255, choices=MAKE_CHOICES, blank=True)

我有一个经销商和品牌数据库，我想计算车辆品牌与其经销商品牌列表相匹配的所有车辆。这就是我如何避免 .extra().

from django.db.models import functions

class SelectUnnest(functions.Func):
    function = 'SELECT UNNEST'

...

Vehicle.objects.filter(
    make__in=SelectUnnest('dealer__make_list')
).count()

结果SQL：

SELECT COUNT(*) AS "__count" FROM "myapp_vehicle" 
INNER JOIN "myapp_dealer" 
  ON ( "myapp_vehicle"."dealer_id" = "myapp_dealer"."id" ) 
WHERE "myapp_vehicle"."make" 
  IN (SELECT UNNEST("myapp_dealer"."make_list"))

它有效，而且比我们可以在 Django 中使用的传统 M2M 方法快得多。但是，对于这个任务，UNNEST 不是一个很好的解决方案：ANY 快得多。我们来试试吧。

class Any(functions.Func):
    function = 'ANY'

...

Vehicle.objects.filter(
    make=Any('dealer__make_list')
).count()

它生成以下 SQL:

 SELECT COUNT(*) AS "__count" FROM "myapp_vehicle" 
 INNER JOIN "myapp_dealer" 
   ON ( "myapp_vehicle"."dealer_id" = "myapp_dealer"."id" )
 WHERE "myapp_vehicle"."make" = 
  (ANY("myapp_dealer"."make_list"))

它失败了，因为 ANY 周围的大括号是假的。如果删除它们，它会在 psql 控制台中正常运行，而且速度很快。

所以我的问题。

有什么方法可以取下这些牙套吗？我在 Django 文档中找不到任何相关内容。
如果没有，- 也许还有其他方法可以改写此查询？

P。 S. 我认为针对不同后端的广泛的数据库函数库对于数据库繁重的 Django 应用程序非常有帮助。

当然，其中大部分不会是table。但是您通常不会经常将这样的项目从一个数据库后端迁移到另一个。在我们的示例中，使用数组字段和 PostGIS，我们坚持使用 PostgreSQL 并且不打算移动。

有人在开发这样的东西吗？

P。 P. S. 有人可能会说，在这种情况下，我们应该为 makes 和 intarray 使用单独的 table 而不是字符串数组，这是正确的并且将会完成，但问题的性质并没有改变.

更新。

TextArrayField 在 djorm_pgarray 处定义。在链接的源文件中，您可以看到它是如何工作的。
该值为文本字符串列表。在Python中，它被表示为一个列表。示例：['honda', 'mazda', 'anything else'].

这是数据库中关于它的说法。

=# select id, make from appname_tablename limit 3;
id | make
---+----------------------
58 | {vw}
76 | {lexus,scion,toyota}
39 | {chevrolet}

底层 PostgreSQL 字段类型是 text[].

Answer 1

我已经（或多或少）通过以下方法获得了你需要的东西：

from django.db.models.lookups import BuiltinLookup
from django.db.models.fields import Field

class Any(BuiltinLookup):
    lookup_name = 'any'

    def get_rhs_op(self, connection, rhs):
       return " = ANY(%s)" % (rhs,)

Field.register_lookup(Any)

并查询：

Vehicle.objects.filter(make__any=F('dealer__make_list')).count()

结果：

SELECT COUNT(*) AS "__count" FROM "zz_vehicle"
  INNER JOIN "zz_dealer" ON ("zz_vehicle"."dealer_id" = "zz_dealer"."id")
    WHERE "zz_vehicle"."make"  = ANY(("zz_dealer"."make_list"))

顺便说一句。而不是 djorm_pgarray 和 TextArrayField 你可以使用原生 django:

make_list = ArrayField(models.CharField(max_length=200), blank=True)

（简化你的依赖）

如何避免 SQL 围绕 Django 自定义数据库函数调用的括号？

How to avoid brackets in SQL around Django custom database function call?

python

django

postgresql

django-orm