Changes between Version 20 and Version 21 of BluePrint/GIS/ShapefileLayers

Timestamp:

04/28/13 10:18:19 (13 years ago)

Author:

jonathlt

Comment:

--

BluePrint/GIS/ShapefileLayers

@@ -28 +28 @@
 Alternatively, to use Python bindings, you could perform the import using GDAL which would allow more control over the import directly within Python, but should still handle the import into Postgres.
 
+
+{{{
+'''
+Dump the contents of a shapefile to an lxml etree object
+Assumes the shapefile is encoded in UTF-8 format
+Tested with TM_WORLD_BORDERS-0.3.shp
+'''
+import sys
+import ogr
+import os
+
+from lxml import etree
+
+if len(sys.argv) != 2:
+    print "Usage: importshape.py filename"
+    sys.exit(0)
+    
+shapefilename = sys.argv[1]
+
+layername = os.path.splitext(os.path.basename(shapefilename))[0]
+
+#Create the datasource
+ds = ogr.Open( shapefilename )
+
+#Open the shapefile
+if ds is None:
+    print "Open failed.\n"
+    sys.exit(0)
+    
+#Get the layer and iterate through the features
+lyr = ds.GetLayer(0)
+
+root = etree.Element("shapefile",name=layername)
+
+for feat in lyr:
+        
+    featurenode = etree.SubElement(root,"feature")
+    
+    feat_defn = lyr.GetLayerDefn()
+                
+    for i in range(feat_defn.GetFieldCount()):
+                
+        field_defn = feat_defn.GetFieldDefn(i)
+                        
+        fieldnode = etree.SubElement(featurenode,field_defn.GetName())
+                
+        if field_defn.GetType() == ogr.OFTInteger:
+            fieldnode.text = str(feat.GetFieldAsInteger(i))
+
+        elif field_defn.GetType() == ogr.OFTReal:
+            fieldnode.text = str(feat.GetFieldAsDouble(i))
+
+        elif field_defn.GetType() == ogr.OFTString:
+            FieldString = str(feat.GetFieldAsString(i))
+            fieldnode.text = FieldString.decode(encoding='UTF-8',errors='strict')
+            
+    wktnode = etree.SubElement(featurenode,"wkt")
+        
+    geom = feat.GetGeometryRef()        
+    wktnode.text = geom.ExportToWkt()    
+                
+
+#Test the etree object
+xmlString = etree.tostring(root, pretty_print=True)
+f = open("test.xml","w")
+f.write(xmlString)
+
+}}}
+
+
 ----
 [wiki:BluePrint/GIS GIS BluePrints]ic