带有额外选择的 Django 查询集计数

Posted 2023-02-23

【中文标题】带有额外选择的 Django 查询集计数

【英文标题】：Django queryset count with extra select

【发布时间】：2012-07-05 07:57:45

【问题描述】：

我有一个带有 PointField 位置坐标的模型。我有一个 mysql 函数，它计算称为 dist 的两点之间的距离。我使用额外（）“选择”来计算查询集中每个返回对象的距离。我还使用 extra() "where" 来过滤特定范围内的那些对象。像这样

query = queryset.extra(
    select=
        "distance":"dist(geomfromtext('%s'),geomfromtext('%s'))"%(loc1, loc2)
    ,
    where=["1 having `distance` <= %s"%(km)]
) #simplified example

这对于获取和读取结果很有效，但当我尝试计算结果集时，我得到了 'distance' 不是字段的错误。在进一步探索之后，似乎 count 忽略了 extra 中的“select”，而只使用了“where”。完整的 SQL 查询如下所示：

SELECT (dist(geomfromtext('POINT (-4.6858300000000003 36.5154300000000021)'),geomfromtext('POINT (-4.8858300000000003 36.5154300000000021)'))) AS `distance`, `testmodel`.`id`, `testmodel`.`name`, `testmodel`.`email`, (...) FROM `testmodel` WHERE 1 having `distance` <= 50.0

count 查询要短得多，并且没有 dist 选择部分：

SELECT COUNT( `testmodel`.`id`) FROM `testmodel` WHERE 1 having `distance` <= 50.0

从逻辑上讲，MySQL 会出错，因为“距离”未定义。有没有办法告诉 Django 它必须为计数包含额外的选择？

感谢您的任何想法！

【问题讨论】：

你需要坚持使用mysql吗？还是可以使用其他 RDBMS？

【参考方案1】：

如果您不打算使用任何其他数据库系统，您可以使用raw query。

    params = 'point1':wktpoint1, 'point2':wktpoint2
    query = """
        SELECT
            dist(%(point1)s, %(point2)s)
        FROM
            testmodel
    ;"""
    query_set = self.raw(query, params)

另外，如果你需要更多的 GIS 支持，你应该评估 PostgreSQL+PostGIS（如果你不喜欢重新发明***，你不应该自己做 dist 函数）

Django 通过 GeoDjango 提供 GIS 支持。那里有像distance 这样的函数。你应该检查支持here

为了使用GeoDjango，你需要在你的模型上添加一个字段，告诉他们使用GeoManager，然后你就可以开始做地理查询了，count应该没有问题。

使用 mysql 你可以使用 geodjango 做这样的事情

### models.py
from django.contrib.gis.db import models
class YourModel(models.Model):
    your_geo_field=models.PolygonField()
    #your_geo_field=models.PointField()
    #your_geo_field=models.GeometryField()
    objects = models.GeoManager()

### your code
from django.contrib.gis.geos import *
from django.contrib.gis.measure import D
a_geom=fromstr('POINT(-96.876369 29.905320)', srid=4326)
distance=5
YoourModel.objects.filter(your_geo_field__distance_lt=(a_geom, D(m=distance))).count()

你可以看到更好的例子here和参考here

【讨论】：

感谢您的回答。我不能使用 PostgreSQL，因为我们仅限于运行 MySQL 的 Amazon RDS。我可以使用原始 sql。但是我怎么做 COUNT() 把 dist() 结果作为条件？我只需要计算距离在一定范围内的结果。

MySQL 有基本的 GIS 支持，你试过 geodjango 吗？你可以这样做：YoourModel.objects.filter(your_geo_field__distance_lt=(a_geom, D(m=5))).count()

另外，尽量不要为距离选择 dist 函数使用别名。在原始查询中，您应该在 where 子句中使用带有参数的函数 (dist) 进行过滤，然后在 select 子句中使用 count(*)。