You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

widgets.py 3.7KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. import logging
  2. from django.conf import settings
  3. from django.contrib.gis import gdal
  4. from django.contrib.gis.geometry import json_regex
  5. from django.contrib.gis.geos import GEOSException, GEOSGeometry
  6. from django.forms.widgets import Widget
  7. from django.utils import translation
  8. logger = logging.getLogger('django.contrib.gis')
  9. class BaseGeometryWidget(Widget):
  10. """
  11. The base class for rich geometry widgets.
  12. Render a map using the WKT of the geometry.
  13. """
  14. geom_type = 'GEOMETRY'
  15. map_srid = 4326
  16. map_width = 600
  17. map_height = 400
  18. display_raw = False
  19. supports_3d = False
  20. template_name = '' # set on subclasses
  21. def __init__(self, attrs=None):
  22. self.attrs = {}
  23. for key in ('geom_type', 'map_srid', 'map_width', 'map_height', 'display_raw'):
  24. self.attrs[key] = getattr(self, key)
  25. if attrs:
  26. self.attrs.update(attrs)
  27. def serialize(self, value):
  28. return value.wkt if value else ''
  29. def deserialize(self, value):
  30. try:
  31. return GEOSGeometry(value)
  32. except (GEOSException, ValueError, TypeError) as err:
  33. logger.error("Error creating geometry from value '%s' (%s)", value, err)
  34. return None
  35. def get_context(self, name, value, attrs):
  36. context = super().get_context(name, value, attrs)
  37. # If a string reaches here (via a validation error on another
  38. # field) then just reconstruct the Geometry.
  39. if value and isinstance(value, str):
  40. value = self.deserialize(value)
  41. if value:
  42. # Check that srid of value and map match
  43. if value.srid and value.srid != self.map_srid:
  44. try:
  45. ogr = value.ogr
  46. ogr.transform(self.map_srid)
  47. value = ogr
  48. except gdal.GDALException as err:
  49. logger.error(
  50. "Error transforming geometry from srid '%s' to srid '%s' (%s)",
  51. value.srid, self.map_srid, err
  52. )
  53. context.update(self.build_attrs(self.attrs, {
  54. 'name': name,
  55. 'module': 'geodjango_%s' % name.replace('-', '_'), # JS-safe
  56. 'serialized': self.serialize(value),
  57. 'geom_type': gdal.OGRGeomType(self.attrs['geom_type']),
  58. 'STATIC_URL': settings.STATIC_URL,
  59. 'LANGUAGE_BIDI': translation.get_language_bidi(),
  60. **(attrs or {}),
  61. }))
  62. return context
  63. class OpenLayersWidget(BaseGeometryWidget):
  64. template_name = 'gis/openlayers.html'
  65. map_srid = 3857
  66. class Media:
  67. css = {
  68. 'all': (
  69. 'https://cdnjs.cloudflare.com/ajax/libs/ol3/4.6.5/ol.css',
  70. 'gis/css/ol3.css',
  71. )
  72. }
  73. js = (
  74. 'https://cdnjs.cloudflare.com/ajax/libs/ol3/4.6.5/ol.js',
  75. 'gis/js/OLMapWidget.js',
  76. )
  77. def serialize(self, value):
  78. return value.json if value else ''
  79. def deserialize(self, value):
  80. geom = super().deserialize(value)
  81. # GeoJSON assumes WGS84 (4326). Use the map's SRID instead.
  82. if geom and json_regex.match(value) and self.map_srid != 4326:
  83. geom.srid = self.map_srid
  84. return geom
  85. class OSMWidget(OpenLayersWidget):
  86. """
  87. An OpenLayers/OpenStreetMap-based widget.
  88. """
  89. template_name = 'gis/openlayers-osm.html'
  90. default_lon = 5
  91. default_lat = 47
  92. default_zoom = 12
  93. def __init__(self, attrs=None):
  94. super().__init__()
  95. for key in ('default_lon', 'default_lat', 'default_zoom'):
  96. self.attrs[key] = getattr(self, key)
  97. if attrs:
  98. self.attrs.update(attrs)