XML 转换为 ESRI Shapefile 的 Python 脚本

1. 如何使用 Python 将 XML 编码为 ESRI Shapefile？本文提供了一个 Python 脚本，将 XML 转换为 ESRI Shapefile。

2、解决方案

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Requires pyshp: https://pypi.python.org/pypi/pyshp
#
# Conversion for http://daten.berlin.de/datensaetze/liste-der-gedenktafeln-berlin
# File: http://gedenktafeln-in-berlin.de/index.php?id=31&type=123
#

from xml.etree import ElementTree
from datetime import datetime
import shapefile
import os


def get_value(list, index, default):
    value = list[index]
    if value is None:
        value = default
    else:
        value = value.text
        if value is None:
            value = default
        else:
            value = value.encode("utf-8")  # 使用 UTF-8 编码
    return value


def add_shape(writer, attributes):
    uid = int(get_value(attributes, 0, 0))
    url = get_value(attributes, 1, "")
    tstamp = get_value(attributes, 2, None)
    if tstamp is not None:
        tstamp = datetime.strptime(tstamp, '%d.%m.%Y')
    ortsteil = get_value(attributes, 3, "")
    strasse = get_value(attributes, 4, "")
    longitude = get_value(attributes, 5, None)
    latitude = get_value(attributes, 6, None)
    Name = get_value(attributes, 7, "")
    inhalt = get_value(attributes, 8, "")
    erlauterung = get_value(attributes, 9, "")
    swo = get_value(attributes, 10, "")
    literatur = get_value(attributes, 11, "")
    personen = get_value(attributes, 12, "")
    entfernt = int(get_value(attributes, 13, 0))

    if longitude is not None or latitude is not None:
        longitude = float(longitude)
        latitude = float(latitude)

        # Fix interchanged coordinates
        temp = 0
        if longitude > latitude:
            temp = latitude
            latitude = longitude
            longitude = temp

        # Add coordinates
        writer.point(longitude, latitude)

    # Add attributes
    writer.record(uid, url, tstamp, ortsteil, strasse, Name, inhalt, erlauterung, swo, literatur, personen, entfernt)


xml_file = 'gedenktafeln.xml'
shape_file = 'gedenktafeln.shp'
projection = 'GEOGCS["GCS_WGS_1984",DATUM["D_WGS_1984",SPHEROID["WGS_1984",6378137.0,298.257223563]],PRIMEM["Greenwich",0.0],UNIT["Degree",0.0174532925199433]]'

tree = ElementTree.parse(xml_file)
writer = shapefile.Writer(shapefile.POINT)

writer.field('uid', fieldType='N', size=5, decimal=0)
writer.field('url', fieldType='C', size=255)  # 设置 URL 字段长度为 255
writer.field('tstamp', fieldType='C', size=19)  # 设置时间戳字段长度为 19
writer.field('ortsteil', fieldType='C', size=200)
writer.field('strasse', fieldType='C', size=200)
writer.field('Name', fieldType='C', size=255)
writer.field('inhalt', fieldType='C', size=255)
writer.field('erlauterung', fieldType='C', size=255)
writer.field('swo', fieldType='C', size=255)
writer.field('literatur', fieldType='C', size=255)
writer.field('personen', fieldType='C', size=255)
writer.field('entfernt', fieldType='N', size=1, decimal=0)

root = tree.getroot()
shapes = root.getchildren()

for shape in shapes:
    attributes = shape.getchildren()
    add_shape(writer, attributes)

try:
    writer.save(shape_file)
except Exception as e:
    print("ortsteil: " + ortsteil)
    print("strasse: " + strasse)
    print("Name: " + Name)
    print("inhalt: " + inhalt)
    print("erlauterung: " + erlauterung)
    print("swo: " + swo)
    print("literatur: " + literatur)
    print("personen: " + personen)
    print("entfernt: " + entfernt)
    raise

# create the PRJ file
with open(os.path.splitext(shape_file)[0] + os.extsep + 'prj', 'w') as prj:
    prj.write(projection)

此脚本实现了以下功能：

1. 使用 ElementTree 解析 XML 并获取节点和属性。
2. 使用 shapefile 模块创建 Shapefile Writer。
3. 为 Shapefile 添加字段。
4. 遍历 XML 中的形状并将其添加到 Shapefile 中。
5. 将 Shapefile 保存为磁盘文件。
6. 将投影写入一个 PRJ 文件。

注意事项：

1. 此脚本假定输入 XML 文件是有效的，并且包含必要的字段。
2. 此脚本假定输出 Shapefile 的投影与输入 XML 文件中的投影兼容。
3. 此脚本只处理点数据。如果需要处理线或面数据，则需要修改脚本。