start-pack
This commit is contained in:
bdrtr
commit
3e1fa59b3d
5723 changed files with 757971 additions and 0 deletions
|
|
@ -0,0 +1,13 @@
|
|||
from django.forms import * # NOQA
|
||||
|
||||
from .fields import ( # NOQA
|
||||
GeometryCollectionField,
|
||||
GeometryField,
|
||||
LineStringField,
|
||||
MultiLineStringField,
|
||||
MultiPointField,
|
||||
MultiPolygonField,
|
||||
PointField,
|
||||
PolygonField,
|
||||
)
|
||||
from .widgets import BaseGeometryWidget, OpenLayersWidget, OSMWidget # NOQA
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
|
|
@ -0,0 +1,144 @@
|
|||
from django import forms
|
||||
from django.contrib.gis.gdal import GDALException
|
||||
from django.contrib.gis.geos import GEOSException, GEOSGeometry
|
||||
from django.core.exceptions import ValidationError
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
|
||||
from .widgets import OpenLayersWidget
|
||||
|
||||
|
||||
class GeometryField(forms.Field):
|
||||
"""
|
||||
This is the basic form field for a Geometry. Any textual input that is
|
||||
accepted by GEOSGeometry is accepted by this form. By default,
|
||||
this includes WKT, HEXEWKB, WKB (in a buffer), and GeoJSON.
|
||||
"""
|
||||
|
||||
widget = OpenLayersWidget
|
||||
geom_type = "GEOMETRY"
|
||||
|
||||
default_error_messages = {
|
||||
"required": _("No geometry value provided."),
|
||||
"invalid_geom": _("Invalid geometry value."),
|
||||
"invalid_geom_type": _("Invalid geometry type."),
|
||||
"transform_error": _(
|
||||
"An error occurred when transforming the geometry "
|
||||
"to the SRID of the geometry form field."
|
||||
),
|
||||
}
|
||||
|
||||
def __init__(self, *, srid=None, geom_type=None, **kwargs):
|
||||
self.srid = srid
|
||||
if geom_type is not None:
|
||||
self.geom_type = geom_type
|
||||
super().__init__(**kwargs)
|
||||
self.widget.attrs["geom_type"] = self.geom_type
|
||||
|
||||
def to_python(self, value):
|
||||
"""Transform the value to a Geometry object."""
|
||||
if value in self.empty_values:
|
||||
return None
|
||||
|
||||
if not isinstance(value, GEOSGeometry):
|
||||
if hasattr(self.widget, "deserialize"):
|
||||
try:
|
||||
value = self.widget.deserialize(value)
|
||||
except GDALException:
|
||||
value = None
|
||||
else:
|
||||
try:
|
||||
value = GEOSGeometry(value)
|
||||
except (GEOSException, ValueError, TypeError):
|
||||
value = None
|
||||
if value is None:
|
||||
raise ValidationError(
|
||||
self.error_messages["invalid_geom"], code="invalid_geom"
|
||||
)
|
||||
|
||||
# Try to set the srid
|
||||
if not value.srid:
|
||||
try:
|
||||
value.srid = self.widget.map_srid
|
||||
except AttributeError:
|
||||
if self.srid:
|
||||
value.srid = self.srid
|
||||
return value
|
||||
|
||||
def clean(self, value):
|
||||
"""
|
||||
Validate that the input value can be converted to a Geometry object
|
||||
and return it. Raise a ValidationError if the value cannot be
|
||||
instantiated as a Geometry.
|
||||
"""
|
||||
geom = super().clean(value)
|
||||
if geom is None:
|
||||
return geom
|
||||
|
||||
# Ensuring that the geometry is of the correct type (indicated
|
||||
# using the OGC string label).
|
||||
if (
|
||||
str(geom.geom_type).upper() != self.geom_type
|
||||
and self.geom_type != "GEOMETRY"
|
||||
):
|
||||
raise ValidationError(
|
||||
self.error_messages["invalid_geom_type"], code="invalid_geom_type"
|
||||
)
|
||||
|
||||
# Transforming the geometry if the SRID was set.
|
||||
if self.srid and self.srid != -1 and self.srid != geom.srid:
|
||||
try:
|
||||
geom.transform(self.srid)
|
||||
except GEOSException:
|
||||
raise ValidationError(
|
||||
self.error_messages["transform_error"], code="transform_error"
|
||||
)
|
||||
|
||||
return geom
|
||||
|
||||
def has_changed(self, initial, data):
|
||||
"""Compare geographic value of data with its initial value."""
|
||||
|
||||
try:
|
||||
data = self.to_python(data)
|
||||
initial = self.to_python(initial)
|
||||
except ValidationError:
|
||||
return True
|
||||
|
||||
# Only do a geographic comparison if both values are available
|
||||
if initial and data:
|
||||
data.transform(initial.srid)
|
||||
# If the initial value was not added by the browser, the geometry
|
||||
# provided may be slightly different, the first time it is saved.
|
||||
# The comparison is done with a very low tolerance.
|
||||
return not initial.equals_exact(data, tolerance=0.000001)
|
||||
else:
|
||||
# Check for change of state of existence
|
||||
return bool(initial) != bool(data)
|
||||
|
||||
|
||||
class GeometryCollectionField(GeometryField):
|
||||
geom_type = "GEOMETRYCOLLECTION"
|
||||
|
||||
|
||||
class PointField(GeometryField):
|
||||
geom_type = "POINT"
|
||||
|
||||
|
||||
class MultiPointField(GeometryField):
|
||||
geom_type = "MULTIPOINT"
|
||||
|
||||
|
||||
class LineStringField(GeometryField):
|
||||
geom_type = "LINESTRING"
|
||||
|
||||
|
||||
class MultiLineStringField(GeometryField):
|
||||
geom_type = "MULTILINESTRING"
|
||||
|
||||
|
||||
class PolygonField(GeometryField):
|
||||
geom_type = "POLYGON"
|
||||
|
||||
|
||||
class MultiPolygonField(GeometryField):
|
||||
geom_type = "MULTIPOLYGON"
|
||||
|
|
@ -0,0 +1,125 @@
|
|||
import logging
|
||||
|
||||
from django.conf import settings
|
||||
from django.contrib.gis import gdal
|
||||
from django.contrib.gis.geometry import json_regex
|
||||
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
|
||||
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", "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, TypeError) 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,
|
||||
)
|
||||
|
||||
geom_type = gdal.OGRGeomType(self.attrs["geom_type"]).name
|
||||
context.update(
|
||||
self.build_attrs(
|
||||
self.attrs,
|
||||
{
|
||||
"name": name,
|
||||
"module": "geodjango_%s" % name.replace("-", "_"), # JS-safe
|
||||
"serialized": self.serialize(value),
|
||||
"geom_type": "Geometry" if geom_type == "Unknown" else 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://cdn.jsdelivr.net/npm/ol@v7.2.2/ol.css",
|
||||
"gis/css/ol3.css",
|
||||
)
|
||||
}
|
||||
js = (
|
||||
"https://cdn.jsdelivr.net/npm/ol@v7.2.2/dist/ol.js",
|
||||
"gis/js/OLMapWidget.js",
|
||||
)
|
||||
|
||||
def serialize(self, value):
|
||||
return value.json if value else ""
|
||||
|
||||
def deserialize(self, value):
|
||||
geom = super().deserialize(value)
|
||||
# GeoJSON assumes WGS84 (4326). Use the map's SRID instead.
|
||||
if geom and json_regex.match(value) and self.map_srid != 4326:
|
||||
geom.srid = self.map_srid
|
||||
return geom
|
||||
|
||||
|
||||
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)
|
||||
Loading…
Add table
Add a link
Reference in a new issue