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

时间:2012-07-04 21:01:59

标签: mysql django django-queryset

我有一个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'不是字段的错误。在进一步探索之后,似乎count忽略了额外的“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

计数查询要短得多,并且没有dist选择部分:

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

逻辑上,MySQL提供错误,因为“距离”未定义。有没有办法告诉Django它必须包括额外的计数选择?

感谢您的任何想法!

1 个答案:

答案 0 :(得分:0)

如果您不打算使用任何其他数据库系统,则可以使用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,您需要在yout模型上添加一个字段,告诉他们使用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