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

Question

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

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

这对于获取和读取结果非常有效，但当我尝试对结果集进行计数时出现错误，即'distance' is not a field。在进一步探索之后，似乎 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 会报错，因为“distance”是未定义的。有没有办法告诉 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