team-3/src/c3nav/mapdata/fields.py

20 lines
487 B
Python
Raw Normal View History

import json
from django.db import models
from shapely.geometry import mapping, shape
2016-09-20 16:13:47 +02:00
from .utils import sort_geojson
class GeometryField(models.TextField):
def from_db_value(self, value, expression, connection, context):
if value is None:
return value
return shape(json.loads(value))
def to_python(self, value):
return shape(json.loads(value))
def get_prep_value(self, value):
2016-09-20 16:13:47 +02:00
return json.dumps(sort_geojson(mapping(value)))