makinacorpus / django-leaflet / 8232975240

from django.contrib.gis.forms.fields import GeometryField as BaseGeometryField

from .widgets import LeafletWidget

class GeometryField(BaseGeometryField):

    widget = LeafletWidget

    geom_type = 'GEOMETRY'

    def __init__(self, *args, **kwargs):

        super().__init__(*args, **kwargs)

        self.widget.geom_type = self.geom_type

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'