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

Question

我有一个带有 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 它必须为计数包含额外的选择？

感谢您的任何想法！

Answer 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