diff options
| -rw-r--r-- | chimere/utils.py | 12 |
1 files changed, 9 insertions, 3 deletions
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: |