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

Question

问题的简短介绍...

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

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

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 控制台中运行，没有问题，而且速度很快。

所以我的问题。

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

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

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

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

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

更新。

• 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)

(简化你的依赖)