38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
|
from django import forms
|
||
|
|
from django.contrib.gis.geos import GEOSGeometry, GEOSException
|
||
|
|
from django.utils.translation import ugettext_lazy as _
|
||
|
|
|
||
|
|
class GeometryField(forms.Field):
|
||
|
|
# By default a Textarea widget is used.
|
||
|
|
widget = forms.Textarea
|
||
|
|
|
||
|
|
default_error_messages = {
|
||
|
|
'no_geom' : _(u'No geometry value provided.'),
|
||
|
|
'invalid_geom' : _(u'Invalid Geometry value.'),
|
||
|
|
'invalid_geom_type' : _(u'Invalid Geometry type.'),
|
||
|
|
}
|
||
|
|
def __init__(self, **kwargs):
|
||
|
|
self.null = kwargs.pop('null')
|
||
|
|
self.geom_type = kwargs.pop('geom_type')
|
||
|
|
super(GeometryField, self).__init__(**kwargs)
|
||
|
|
|
||
|
|
def clean(self, value):
|
||
|
|
"""
|
||
|
|
Validates that the input value can be converted to a Geometry
|
||
|
|
object (which is returned). A ValidationError is raised if
|
||
|
|
the value cannot be instantiated as a Geometry.
|
||
|
|
"""
|
||
|
|
if not value:
|
||
|
|
if self.null:
|
||
|
|
# The geometry column allows NULL, return None.
|
||
|
|
return None
|
||
|
|
else:
|
||
|
|
raise forms.ValidationError(self.error_messages['no_geom'])
|
||
|
|
try:
|
||
|
|
geom = GEOSGeometry(value)
|
||
|
|
if geom.geom_type.upper() != self.geom_type:
|
||
|
|
raise forms.ValidationError(self.error_messages['invalid_geom_type'])
|
||
|
|
return geom
|
||
|
|
except GEOSException:
|
||
|
|
raise forms.ValidationError(self.error_messages['invalid_geom'])
|