improve load_river command

Georiviere · Apr 24, 2024 · f60b38e · f60b38e
1 parent 1b90c7e
commit f60b38e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/georiviere/river/management/commands/load_rivers.py b/georiviere/river/management/commands/load_rivers.py
@@ -40,7 +40,7 @@ def handle(self, *args, **options):
 
         objs = (Stream(geom=feat.geom.geos,
                        source_location=Point(feat.geom.geos[0]),
-                       name=feat.get(name_column) or default_name) for feat in layer)
+                       name=feat.get(name_column) or default_name) for feat in layer if feat.geom.geos.geom_typeid == 1)
         count = 0
         while True:
             batch = list(islice(objs, batch_size))