Ticket #12416: write_kml.diff

django/contrib/gis/utils/kml.py

@@ +1 @@
+from django.utils.encoding import smart_unicode
+from django.utils.safestring import mark_safe
+
+_altitude_modes = {'clampToGround' : None,
+                   'relativeToGround' : None,
+                   'absolute' : None,
+                   }
+
+def write_kml(geom, altitude_mode=None, extrude=None, id=None, tessellate=None):
+    gtype = str(geom.geom_type)
+    
+    if gtype in ('MultiPolygon', 'MultiLineString', 'MultiPoint'):
+        geom_elem = 'MultiGeometry'
+        elem_replace = [gtype.replace('Multi', '')]
+    elif gtype == 'GeometryCollection':
+        geom_elem = 'MultiGeometry'
+        elem_replace = ['Point', 'LineString', 'LinearRing', 'Polygon']
+    else:
+        geom_elem = gtype
+        elem_replace = [gtype]
+
+    def kml_element(element, value):
+        return (u'<%(element)s>%(value)s</%(element)s>' % 
+                {'element' : element, 'value' : value}
+                )
+
+    tag_replace_kml = []
+
+    if not tessellate is None:
+        tessellate = int(bool(tessellate))
+        tag_replace_kml.append(kml_element('tessellate', tessellate))        
+    
+    if not extrude is None:
+        extrude = int(bool(extrude))
+        tag_replace_kml.append(kml_element('extrude', extrude))
+
+    if not altitude_mode is None:
+        if not altitude_mode in _altitude_modes:
+            raise ValueError('Altitude mode must be one of: "%s"' % 
+                             '","'.join(_altitude_modes.keys())) 
+        tag_replace_kml.append(kml_element('altitudeMode', altitude_mode))
+
+    kml = geom.kml
+    
+    if tag_replace_kml:
+        replace_kml = u''.join(tag_replace_kml)
+        for elem in elem_replace:
+            elem = u'<%s>' % elem
+            kml = kml.replace(elem, elem + replace_kml)
+
+    if not id is None:
+        elem = u'<%s>' % geom_elem
+        kml = kml.replace(elem, u'<%s id="%s">' % (geom_elem, smart_unicode(id)))
+
+    return kml