fix new error in locations_for_request

This commit is contained in:
Laura Klünder 2017-12-20 18:24:13 +01:00
parent 1d10563f5f
commit 182b50ef68

View file

@ -63,7 +63,7 @@ def locations_for_request(request) -> Mapping[int, LocationSlug]:
# add levels to spaces # add levels to spaces
remove_pks = set() remove_pks = set()
levels = {pk: obj for pk, obj in locations.items() if isinstance(obj, Level)} levels = {pk: obj for pk, obj in locations.items() if isinstance(obj, Level)}
for pk, obj in locations.keys(): for pk, obj in locations.items():
if isinstance(obj, LevelGeometryMixin): if isinstance(obj, LevelGeometryMixin):
level = levels.get(obj.level_id, None) level = levels.get(obj.level_id, None)
if level is None: if level is None:
@ -78,7 +78,7 @@ def locations_for_request(request) -> Mapping[int, LocationSlug]:
# add spaces to areas and POIs # add spaces to areas and POIs
remove_pks = set() remove_pks = set()
spaces = {pk: obj for pk, obj in locations.items() if isinstance(obj, Space)} spaces = {pk: obj for pk, obj in locations.items() if isinstance(obj, Space)}
for obj in locations.values(): for pk, obj in locations.items():
if isinstance(obj, SpaceGeometryMixin): if isinstance(obj, SpaceGeometryMixin):
space = spaces.get(obj.space_id, None) space = spaces.get(obj.space_id, None)
if space is None: if space is None: