Export: fix shapefile export when no category is set

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: