summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJustin Bronn <jbronn@gmail.com>2011-03-15 02:27:48 +0000
committerJustin Bronn <jbronn@gmail.com>2011-03-15 02:27:48 +0000
commitddb1cee0e5a0fcb0c87ff39e6becd46ba981a29b (patch)
tree3235982ecf1ff6065d5981500f69de6cf73b9916
parente8dcaa4bbac8349054734581794424895b85f059 (diff)
downloaddjango-ddb1cee0e5a0fcb0c87ff39e6becd46ba981a29b.tar.gz
[1.2.X] Fixed #15378 -- Now properly handle OGR layers that have features with invalid geometries. Thanks, kunitoki for bug report and initial patch.
Backport of r15813 from trunk. git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@15814 bcc190cf-cafb-0310-a4f2-bffc1f526a37
-rw-r--r--django/contrib/gis/tests/layermap/models.py3
-rw-r--r--django/contrib/gis/tests/layermap/tests.py14
-rw-r--r--django/contrib/gis/utils/layermapping.py5
3 files changed, 19 insertions, 3 deletions
diff --git a/django/contrib/gis/tests/layermap/models.py b/django/contrib/gis/tests/layermap/models.py
index 3a34d16f3f..51213eb0b8 100644
--- a/django/contrib/gis/tests/layermap/models.py
+++ b/django/contrib/gis/tests/layermap/models.py
@@ -43,6 +43,9 @@ class ICity1(CityBase):
class ICity2(ICity1):
dt_time = models.DateTimeField(auto_now=True)
+class Invalid(models.Model):
+ point = models.PointField()
+
# Mapping dictionaries for the models above.
co_mapping = {'name' : 'Name',
'state' : {'name' : 'State'}, # ForeignKey's use another mapping dictionary for the _related_ Model (State in this case).
diff --git a/django/contrib/gis/tests/layermap/tests.py b/django/contrib/gis/tests/layermap/tests.py
index 6394b041d8..1ffd09117b 100644
--- a/django/contrib/gis/tests/layermap/tests.py
+++ b/django/contrib/gis/tests/layermap/tests.py
@@ -4,16 +4,19 @@ from decimal import Decimal
from django.utils.copycompat import copy
-from django.contrib.gis.gdal import DataSource
+from django.contrib.gis.gdal import DataSource, OGRException
from django.contrib.gis.tests.utils import mysql
from django.contrib.gis.utils.layermapping import LayerMapping, LayerMapError, InvalidDecimal, MissingForeignKey
-from models import City, County, CountyFeat, Interstate, ICity1, ICity2, State, city_mapping, co_mapping, cofeat_mapping, inter_mapping
+from models import \
+ City, County, CountyFeat, Interstate, ICity1, ICity2, Invalid, State, \
+ city_mapping, co_mapping, cofeat_mapping, inter_mapping
shp_path = os.path.realpath(os.path.join(os.path.dirname(__file__), os.pardir, 'data'))
city_shp = os.path.join(shp_path, 'cities', 'cities.shp')
co_shp = os.path.join(shp_path, 'counties', 'counties.shp')
inter_shp = os.path.join(shp_path, 'interstates', 'interstates.shp')
+invalid_shp = os.path.join(shp_path, 'invalid', 'emptypoints.shp')
# Dictionaries to hold what's expected in the county shapefile.
NAMES = ['Bexar', 'Galveston', 'Harris', 'Honolulu', 'Pueblo']
@@ -266,3 +269,10 @@ class LayerMapTest(unittest.TestCase):
self.assertEqual(6, ICity1.objects.count())
self.assertEqual(3, ICity2.objects.count())
+ def test07_invalid_layer(self):
+ "Tests LayerMapping on invalid geometries. See #15378."
+ invalid_mapping = {'point': 'POINT'}
+ lm = LayerMapping(Invalid, invalid_shp, invalid_mapping,
+ source_srs=4326)
+ lm.save(silent=True)
+
diff --git a/django/contrib/gis/utils/layermapping.py b/django/contrib/gis/utils/layermapping.py
index cec1989796..945c745be4 100644
--- a/django/contrib/gis/utils/layermapping.py
+++ b/django/contrib/gis/utils/layermapping.py
@@ -292,7 +292,10 @@ class LayerMapping(object):
if isinstance(model_field, GeometryField):
# Verify OGR geometry.
- val = self.verify_geom(feat.geom, model_field)
+ try:
+ val = self.verify_geom(feat.geom, model_field)
+ except OGRException:
+ raise LayerMapError('Could not retrieve geometry from feature.')
elif isinstance(model_field, models.base.ModelBase):
# The related _model_, not a field was passed in -- indicating
# another mapping for the related Model.