Geo Django子类化查询集

时间:2011-08-12 05:37:00

标签: django django-models django-queryset geo subclassing

我正在使用GeoDjango搜索一堆不同类型的位置。例如,House和Appartment模型都是Location的子类。

使用下面的Subclassing Queryset,我可以执行类似Location.objects.all()的操作并让它返回给我[<House: myhouse>, <House: yourhouse>, <Appartment: myappartment>],这是我的愿望。

但是,我还想确定每个位置的距离。通常,如果没有Subclassing Queryset,附件2中的代码会返回从给定点到每个位置的距离.... [ (<Location: Location object>, Distance(m=866.092847284))]

但是,如果我尝试使用子类化查询集找到距离,我会收到如下错误:

AttributeError:'House'对象没有属性'distance'

你知道如何保留返回子类对象的查询集但是在子类对象上有distance属性的能力吗?任何建议都非常感谢。

图表1:

class SubclassingQuerySet(models.query.GeoQuerySet):
    def __getitem__(self, k):
        result = super(SubclassingQuerySet, self).__getitem__(k)
        if isinstance(result, models.Model) :
            return result.as_leaf_class()
        else :
            return result
    def __iter__(self):
        for item in super(SubclassingQuerySet, self).__iter__():
            yield item.as_leaf_class()

class LocationManager(models.GeoManager):
    def get_query_set(self):
        return SubclassingQuerySet(self.model)

class Location(models.Model):
    content_type = models.ForeignKey(ContentType,editable=False,null=True)
    objects = LocationManager()

class House(Location):
    address = models.CharField(max_length=255, blank=True, null=True)
    objects = LocationManager()

class Appartment(Location):
    address = models.CharField(max_length=255, blank=True, null=True)
    unit = models.CharField(max_length=255, blank=True, null=True)
    objects = LocationManager()

图表2:

from django.contrib.gis.measure import D 
from django.contrib.gis.geos import fromstr
ref_pnt =  fromstr('POINT(-87.627778 41.881944)')

location_objs = Location.objects.filter(
        point__distance_lte=(ref_pnt, D(m=1000) 
              )).distance(ref_pnt).order_by('distance')
[ (l, l.distance) for l in location_objs.distance(ref_pnt) ]   # <--- errors out here

2 个答案:

答案 0 :(得分:0)

我正忙着尝试解决这个问题。怎么样:

class QuerySetManager(models.GeoManager):
    '''
    Generates a new QuerySet method and extends the original query object manager in the Model
    '''
    def get_query_set(self):
        return super(QuerySetManager, self).get_query_set()

其余的可以从DjangoSnippet开始。

答案 1 :(得分:0)

您必须在所有子类中重新分配经理。

来自Django文档:

  

在非抽象基类上定义的管理器不会被子类继承。如果要从非抽象基础重用管理器,请在子类上显式重新声明它。这些类型的管理器可能对它们所定义的类非常具体,因此继承它们通常会导致意外结果(特别是对于默认管理器而言)。因此,它们不会传递给子类。

https://docs.djangoproject.com/en/dev/topics/db/managers/#custom-managers-and-model-inheritance