110 lines
3.3 KiB
Python
110 lines
3.3 KiB
Python
import logging
|
|
|
|
from django.conf import settings
|
|
from django.contrib.gis import gdal
|
|
from django.contrib.gis.geos import GEOSException, GEOSGeometry
|
|
from django.forms.widgets import Widget
|
|
from django.utils import translation
|
|
|
|
logger = logging.getLogger('django.contrib.gis')
|
|
|
|
|
|
class BaseGeometryWidget(Widget):
|
|
"""
|
|
The base class for rich geometry widgets.
|
|
Render a map using the WKT of the geometry.
|
|
"""
|
|
geom_type = 'GEOMETRY'
|
|
map_srid = 4326
|
|
map_width = 600
|
|
map_height = 400
|
|
display_raw = False
|
|
|
|
supports_3d = False
|
|
template_name = '' # set on subclasses
|
|
|
|
def __init__(self, attrs=None):
|
|
self.attrs = {}
|
|
for key in ('geom_type', 'map_srid', 'map_width', 'map_height', 'display_raw'):
|
|
self.attrs[key] = getattr(self, key)
|
|
if attrs:
|
|
self.attrs.update(attrs)
|
|
|
|
def serialize(self, value):
|
|
return value.wkt if value else ''
|
|
|
|
def deserialize(self, value):
|
|
try:
|
|
return GEOSGeometry(value)
|
|
except (GEOSException, ValueError) as err:
|
|
logger.error("Error creating geometry from value '%s' (%s)", value, err)
|
|
return None
|
|
|
|
def get_context(self, name, value, attrs):
|
|
context = super().get_context(name, value, attrs)
|
|
# If a string reaches here (via a validation error on another
|
|
# field) then just reconstruct the Geometry.
|
|
if value and isinstance(value, str):
|
|
value = self.deserialize(value)
|
|
|
|
if value:
|
|
# Check that srid of value and map match
|
|
if value.srid and value.srid != self.map_srid:
|
|
try:
|
|
ogr = value.ogr
|
|
ogr.transform(self.map_srid)
|
|
value = ogr
|
|
except gdal.GDALException as err:
|
|
logger.error(
|
|
"Error transforming geometry from srid '%s' to srid '%s' (%s)",
|
|
value.srid, self.map_srid, err
|
|
)
|
|
|
|
context.update(self.build_attrs(self.attrs, {
|
|
'name': name,
|
|
'module': 'geodjango_%s' % name.replace('-', '_'), # JS-safe
|
|
'serialized': self.serialize(value),
|
|
'geom_type': gdal.OGRGeomType(self.attrs['geom_type']),
|
|
'STATIC_URL': settings.STATIC_URL,
|
|
'LANGUAGE_BIDI': translation.get_language_bidi(),
|
|
**(attrs or {}),
|
|
}))
|
|
return context
|
|
|
|
|
|
class OpenLayersWidget(BaseGeometryWidget):
|
|
template_name = 'gis/openlayers.html'
|
|
map_srid = 3857
|
|
|
|
class Media:
|
|
css = {
|
|
'all': (
|
|
'https://cdnjs.cloudflare.com/ajax/libs/ol3/3.20.1/ol.css',
|
|
'gis/css/ol3.css',
|
|
)
|
|
}
|
|
js = (
|
|
'https://cdnjs.cloudflare.com/ajax/libs/ol3/3.20.1/ol.js',
|
|
'gis/js/OLMapWidget.js',
|
|
)
|
|
|
|
def serialize(self, value):
|
|
return value.json if value else ''
|
|
|
|
|
|
class OSMWidget(OpenLayersWidget):
|
|
"""
|
|
An OpenLayers/OpenStreetMap-based widget.
|
|
"""
|
|
template_name = 'gis/openlayers-osm.html'
|
|
default_lon = 5
|
|
default_lat = 47
|
|
default_zoom = 12
|
|
|
|
def __init__(self, attrs=None):
|
|
super().__init__()
|
|
for key in ('default_lon', 'default_lat', 'default_zoom'):
|
|
self.attrs[key] = getattr(self, key)
|
|
if attrs:
|
|
self.attrs.update(attrs)
|