Редактирование содержимого KML с использованием lxml - PullRequest
Новая
       34

Редактирование содержимого KML с использованием lxml

0 голосов
/ 02 ноября 2019

У меня есть простой KML-файл, который я хочу изменить, используя etree.XMLParser. 

<kml xmlns="http://www.opengis.net/kml/2.2">
<Document id="root_doc">
    <Schema name="test" id="test">
        <SimpleField name="id" type="string"/>
        <SimpleField name="LAT" type="float"/>
        <SimpleField name="LONG" type="float"/>
    </Schema>
    <Folder>
        <name>test</name>
        <Placemark>
            <Style>
                <LineStyle><color>ff0000ff</color></LineStyle>
                <PolyStyle><fill>0</fill></PolyStyle>
            </Style>
            <ExtendedData>
            <SchemaData schemaUrl="#test">
                <SimpleData name="id">ABC</SimpleData>
                <SimpleData name="LAT">-35.06111</SimpleData>
                <SimpleData name="LONG">138.58389</SimpleData>
                </SchemaData>
            </ExtendedData>
            <Polygon>
                <outerBoundaryIs>
                    <LinearRing>
                        <coordinates>138.58389,-35.06111 138.585208,-35.059241 
                        </coordinates>
                    </LinearRing>
                </outerBoundaryIs>
            </Polygon>
        </Placemark>
    </Folder>
</Document>
</kml>

Во-первых, как я могу изменить идентификатор схемы в документе и schemaUrl в Placemark с 'test', чтобы сказать 'test_1? Я попробовал это, но не сработало: 

ns = {"kml": "http://www.opengis.net/kml/2.2"}
schema_element = placemark.xpath("kml:ExtendedData/kml:SchemaData[@schemaUrl]", namespaces=ns)[0]
schema_element.text = 'test_1'

Во-вторых, я хочу вставить новый стиль для документа. Я думал создать строку и элемент string_element. Вставьте строковый элемент под документом: 

style_element = etree.Element(etree.QName(ns.get("kml"), "style"), nsmap=ns)
style_element.text = style_str   

document_element = tree.xpath("//kml:Document", namespaces=ns) 
document_element.insert(1, style_str)

Это конечный вывод, который мне нужен: 

<kml xmlns="http://www.opengis.net/kml/2.2">
<Document id="root_doc">
    <Schema name="test" id="test_1">
        <SimpleField name="id" type="string"/>
        <SimpleField name="LAT" type="float"/>
        <SimpleField name="LONG" type="float"/>
    </Schema>
    <Style id="falseColor">
        <BalloonStyle>
            <text><![CDATA[<table border="0">
                <tr><td><b>ID</b></td><td>$[test/id]</td></tr>
                <tr><td><b>LAT</b></td><td>$[test/LAT]</td></tr>
                <tr><td><b>LON</b></td><td>$[test/LONG]</td></tr>
                </table>]]> 
            </text>
        </BalloonStyle>
        <PolyStyle>
            <colorMode>random</colorMode>
        </PolyStyle>
    </Style>
    <Folder>
        <name>test</name>
        <Placemark>
            <styleUrl>#falseColor</styleUrl>
            <ExtendedData>
                <SchemaData schemaUrl="#test_1">
                    <SimpleData name="id">ABC</SimpleData>
                    <SimpleData name="LAT">-35.06111</SimpleData>
                    <SimpleData name="LONG">138.58389</SimpleData>
                </SchemaData>
            </ExtendedData>
            <Polygon>
                <outerBoundaryIs>
                    <LinearRing>
                        <coordinates>138.58389,-35.06111 138.585208,-35.059241 
                        </coordinates>
                    </LinearRing>
                </outerBoundaryIs>
            </Polygon>
        </Placemark>
    </Folder>
</Document>
</kml>

Кто-нибудь может помочь?

Спасибо

1 Ответ

2 голосов
/ 02 ноября 2019

Часть 1: schemaUrl является атрибутом. Найдите элемент, которому принадлежит атрибут, и установите новое значение: 

schemadata_element = tree.xpath("//kml:SchemaData", namespaces=ns)[0]
schemadata_element.set("schemaUrl", "#test_1")

Атрибут id в элементе Schema обновляется таким же образом.

Часть 2:Добавление нового стиля немного сложнее, но не очень сложно. Создайте элемент Style в виде строки, проанализируйте его в объект Element с помощью fromstring() (убедитесь, что раздел CDATA сохранен с помощью параметра синтаксического анализатора strip_cdata=False), а затем вставьте элемент в нужное место.

Полный код, который выполняет обе части (и удаляет существующий элемент Style): 

from lxml import etree

ns = {"kml": "http://www.opengis.net/kml/2.2"}

parser = etree.XMLParser(remove_blank_text=True, strip_cdata=False)
tree = etree.parse("your_kml_file.xml", parser)

schema_element = tree.xpath("//kml:Schema", namespaces=ns)[0]
schema_element.set("id", "test_1")

schemadata_element = tree.xpath("//kml:SchemaData", namespaces=ns)[0]
schemadata_element.set("schemaUrl", "#test_1")

# Remove existing "Style" element and add "styleUrl" element 
old_style_element = tree.xpath("//kml:Style", namespaces=ns)[0] 
placemark = old_style_element.getparent()
placemark.remove(old_style_element)

styleurl_element = etree.Element(etree.QName(ns["kml"], "styleUrl"))
styleurl_element.text = "#falseColor"
placemark.insert(0, styleurl_element)

# Add new "Style" element 
style_str = """
<Style id="falseColor" xmlns="http://www.opengis.net/kml/2.2">
    <BalloonStyle>
        <text><![CDATA[<table border="0">
            <tr><td><b>ID</b></td><td>$[test/id]</td></tr>
            <tr><td><b>LAT</b></td><td>$[test/LAT]</td></tr>
            <tr><td><b>LON</b></td><td>$[test/LONG]</td></tr>
            </table>]]> 
        </text>
    </BalloonStyle>
    <PolyStyle>
       <colorMode>random</colorMode>
    </PolyStyle>
</Style>"""

style_element = etree.fromstring(style_str, parser)

document_element = tree.xpath("//kml:Document", namespaces=ns)[0] 
document_element.insert(1, style_element)
...