XML Example about glom HOT 6 CLOSED

df = (spark.read
    .format('com.databricks.spark.xml')
    .options(rowTag='Placemark')
    .load('file:/more_annoying_xml.kml')
)

df.printSchema()
# root
#  |-- Point: struct (nullable = true)
#  |    |-- altitudeMode: string (nullable = true)
#  |    |-- coordinates: string (nullable = true)
#  |-- TimeStamp: struct (nullable = true)
#  |    |-- when: string (nullable = true)
#  |-- description: string (nullable = true)
#  |-- name: string (nullable = true)


df.select('Point.coordinates', 'TimeStamp.when', 'name').printSchema()
# root
#  |-- coordinates: string (nullable = true)
#  |-- when: string (nullable = true)
#  |-- name: string (nullable = true)

import untangle
from glom import glom, T

annoying_xml = """<?xml version="1.0" encoding="UTF-8"?>
<kml xmlns="http://www.opengis.net/kml/2.2">
  <Placemark>
    <name>Simple placemark</name>
    <description>Attached to the ground. Intelligently places itself 
       at the height of the underlying terrain.</description>
    <Point>
      <coordinates>-122.0822035425683,37.42228990140251,0</coordinates>
    </Point>
  </Placemark>
</kml>
"""

ut = untangle.parse(annoying_xml)

glom(ut, {'coords': (T.kml.Placemark.Point.coordinates.cdata, T.split(','), [float]),
         'name': T.kml.Placemark.name.cdata})
# result:
# {'coords': [-122.0822035425683, 37.42228990140251, 0.0], 
#  'name': u'Simple placemark'}