Commit 0d565dac authored by Alex Kerney's avatar Alex Kerney

Changed the DEFAULT_FORMATTERS to WKBElement instead of Shapely

parent bf33c0ba
...@@ -3,7 +3,8 @@ import json ...@@ -3,7 +3,8 @@ import json
from jinja2 import Markup from jinja2 import Markup
from wtforms.widgets import html_params from wtforms.widgets import html_params
from shapely.geometry import mapping from shapely.geometry import mapping
from shapely.geometry.base import BaseGeometry from geoalchemy2.shape import to_shape
from geoalchemy2.elements import WKBElement
def geom_formatter(view, value): def geom_formatter(view, value):
...@@ -12,12 +13,12 @@ def geom_formatter(view, value): ...@@ -12,12 +13,12 @@ def geom_formatter(view, value):
"disabled": "disabled", "disabled": "disabled",
"data-width": 100, "data-width": 100,
"data-height": 70, "data-height": 70,
"data-geometry-type": value.geom_type, "data-geometry-type": to_shape(value).geom_type,
"data-zoom": 15, "data-zoom": 15,
}) })
geojson = json.dumps(mapping(value)) geojson = json.dumps(mapping(to_shape(value)))
return Markup('<textarea %s>%s</textarea>' % (params, geojson)) return Markup('<textarea %s>%s</textarea>' % (params, geojson))
DEFAULT_FORMATTERS = BASE_FORMATTERS.copy() DEFAULT_FORMATTERS = BASE_FORMATTERS.copy()
DEFAULT_FORMATTERS[BaseGeometry] = geom_formatter DEFAULT_FORMATTERS[WKBElement] = geom_formatter
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment