From b7d3c863a71185f80eb899db907c2a656667be78 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 3 Aug 2016 15:46:28 +0200 Subject: Export: fix shapefile export when no category is set --- chimere/utils.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'chimere/utils.py') diff --git a/chimere/utils.py b/chimere/utils.py index f5ddcd5..f9173f2 100644 --- a/chimere/utils.py +++ b/chimere/utils.py @@ -462,11 +462,17 @@ class ShapefileManager(ImportManager): for item in queryset: # duplicate items when in several categories - for category in item.categories.all(): + q = item.categories + if not q.count(): + categories = [None] + else: + categories = q.all() + for category in categories: feat = ogr.Feature(feature_def) feat.SetField('name', str(unicode_normalize(item.name)[:80])) - feat.SetField('category', - str(unicode_normalize(category.name)[:80])) + if category: + feat.SetField('category', + str(unicode_normalize(category.name)[:80])) geom = getattr(item, geo_field.name) if not geom: -- cgit v1.2.3