143 lines
5.5 KiB
Python
143 lines
5.5 KiB
Python
from django.contrib.admin import ModelAdmin
|
|
from django.contrib.gis.admin.widgets import OpenLayersWidget
|
|
from django.contrib.gis.db import models
|
|
from django.contrib.gis.gdal import HAS_GDAL, OGRGeomType
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
|
|
spherical_mercator_srid = 3857
|
|
|
|
|
|
class GeoModelAdmin(ModelAdmin):
|
|
"""
|
|
The administration options class for Geographic models. Map settings
|
|
may be overloaded from their defaults to create custom maps.
|
|
"""
|
|
# The default map settings that may be overloaded -- still subject
|
|
# to API changes.
|
|
default_lon = 0
|
|
default_lat = 0
|
|
default_zoom = 4
|
|
display_wkt = False
|
|
display_srid = False
|
|
extra_js = []
|
|
num_zoom = 18
|
|
max_zoom = False
|
|
min_zoom = False
|
|
units = False
|
|
max_resolution = False
|
|
max_extent = False
|
|
modifiable = True
|
|
mouse_position = True
|
|
scale_text = True
|
|
layerswitcher = True
|
|
scrollable = True
|
|
map_width = 600
|
|
map_height = 400
|
|
map_srid = 4326
|
|
map_template = 'gis/admin/openlayers.html'
|
|
openlayers_url = 'http://openlayers.org/api/2.13.1/OpenLayers.js'
|
|
point_zoom = num_zoom - 6
|
|
wms_url = 'http://vmap0.tiles.osgeo.org/wms/vmap0'
|
|
wms_layer = 'basic'
|
|
wms_name = 'OpenLayers WMS'
|
|
wms_options = {'format': 'image/jpeg'}
|
|
debug = False
|
|
widget = OpenLayersWidget
|
|
|
|
@property
|
|
def media(self):
|
|
"Injects OpenLayers JavaScript into the admin."
|
|
media = super(GeoModelAdmin, self).media
|
|
media.add_js([self.openlayers_url])
|
|
media.add_js(self.extra_js)
|
|
return media
|
|
|
|
def formfield_for_dbfield(self, db_field, **kwargs):
|
|
"""
|
|
Overloaded from ModelAdmin so that an OpenLayersWidget is used
|
|
for viewing/editing 2D GeometryFields (OpenLayers 2 does not support
|
|
3D editing).
|
|
"""
|
|
if isinstance(db_field, models.GeometryField) and db_field.dim < 3:
|
|
kwargs.pop('request', None)
|
|
# Setting the widget with the newly defined widget.
|
|
kwargs['widget'] = self.get_map_widget(db_field)
|
|
return db_field.formfield(**kwargs)
|
|
else:
|
|
return super(GeoModelAdmin, self).formfield_for_dbfield(db_field, **kwargs)
|
|
|
|
def get_map_widget(self, db_field):
|
|
"""
|
|
Returns a subclass of the OpenLayersWidget (or whatever was specified
|
|
in the `widget` attribute) using the settings from the attributes set
|
|
in this class.
|
|
"""
|
|
is_collection = db_field.geom_type in ('MULTIPOINT', 'MULTILINESTRING', 'MULTIPOLYGON', 'GEOMETRYCOLLECTION')
|
|
if is_collection:
|
|
if db_field.geom_type == 'GEOMETRYCOLLECTION':
|
|
collection_type = 'Any'
|
|
else:
|
|
collection_type = OGRGeomType(db_field.geom_type.replace('MULTI', ''))
|
|
else:
|
|
collection_type = 'None'
|
|
|
|
class OLMap(self.widget):
|
|
template = self.map_template
|
|
geom_type = db_field.geom_type
|
|
|
|
wms_options = ''
|
|
if self.wms_options:
|
|
wms_options = ["%s: '%s'" % pair for pair in self.wms_options.items()]
|
|
wms_options = ', %s' % ', '.join(wms_options)
|
|
|
|
params = {'default_lon': self.default_lon,
|
|
'default_lat': self.default_lat,
|
|
'default_zoom': self.default_zoom,
|
|
'display_wkt': self.debug or self.display_wkt,
|
|
'geom_type': OGRGeomType(db_field.geom_type),
|
|
'field_name': db_field.name,
|
|
'is_collection': is_collection,
|
|
'scrollable': self.scrollable,
|
|
'layerswitcher': self.layerswitcher,
|
|
'collection_type': collection_type,
|
|
'is_generic': db_field.geom_type == 'GEOMETRY',
|
|
'is_linestring': db_field.geom_type in ('LINESTRING', 'MULTILINESTRING'),
|
|
'is_polygon': db_field.geom_type in ('POLYGON', 'MULTIPOLYGON'),
|
|
'is_point': db_field.geom_type in ('POINT', 'MULTIPOINT'),
|
|
'num_zoom': self.num_zoom,
|
|
'max_zoom': self.max_zoom,
|
|
'min_zoom': self.min_zoom,
|
|
'units': self.units, # likely should get from object
|
|
'max_resolution': self.max_resolution,
|
|
'max_extent': self.max_extent,
|
|
'modifiable': self.modifiable,
|
|
'mouse_position': self.mouse_position,
|
|
'scale_text': self.scale_text,
|
|
'map_width': self.map_width,
|
|
'map_height': self.map_height,
|
|
'point_zoom': self.point_zoom,
|
|
'srid': self.map_srid,
|
|
'display_srid': self.display_srid,
|
|
'wms_url': self.wms_url,
|
|
'wms_layer': self.wms_layer,
|
|
'wms_name': self.wms_name,
|
|
'wms_options': wms_options,
|
|
'debug': self.debug,
|
|
}
|
|
return OLMap
|
|
|
|
|
|
class OSMGeoAdmin(GeoModelAdmin):
|
|
map_template = 'gis/admin/osm.html'
|
|
num_zoom = 20
|
|
map_srid = spherical_mercator_srid
|
|
max_extent = '-20037508,-20037508,20037508,20037508'
|
|
max_resolution = '156543.0339'
|
|
point_zoom = num_zoom - 6
|
|
units = 'm'
|
|
|
|
def __init__(self, *args):
|
|
if not HAS_GDAL:
|
|
raise ImproperlyConfigured("OSMGeoAdmin is not usable without GDAL libs installed")
|
|
super(OSMGeoAdmin, self).__init__(*args)
|