2025-summer/team-3
12
Fork 0
Code Issues Pull requests Projects Releases Activity

fix register_changed_geometries on LocationGroup and LocationGroupCategory

Browse source
This commit is contained in:
Laura Klünder 2017-10-28 13:04:46 +02:00
parent 4f0dce05a1
commit 1db9db542f
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/c3nav/mapdata/models/locations.py
View file

@ -189,12 +189,13 @@ class LocationGroupCategory(TitledMixin, models.Model):
def register_changed_geometries(self):
from c3nav.mapdata.models.geometry.space import SpaceGeometryMixin
query = self.locationgroups.all()
query = self.groups.all()
for model in get_submodels(SpecificLocation):
related_name = SpecificLocation._meta.default_related_name
query.prefetch_related('locationgroup__'+related_name)
related_name = model._meta.default_related_name
subquery = model.objects.all()
if issubclass(model, SpaceGeometryMixin):
query = query.select_related('locationgorups__'+related_name+'__space')
subquery = subquery.select_related('space')
query.prefetch_related(Prefetch('groups__'+related_name, subquery))
for group in query:
group.register_changed_geometries(do_query=False)
@ -252,7 +253,7 @@ class LocationGroup(Location, models.Model):
def register_changed_geometries(self, do_query=True):
from c3nav.mapdata.models.geometry.space import SpaceGeometryMixin
for model in get_submodels(SpecificLocation):
query = getattr(self, SpecificLocation._meta.default_related_name).objects.all()
query = getattr(self, model._meta.default_related_name).all()
if do_query:
if issubclass(model, SpaceGeometryMixin):
query = query.select_related('space')