import csv,sys
import osgeo.ogr, osgeo.osr #we will need some packages
from osgeo import ogr
from osgeo import *
import osgeo.osr as osr
def test():
boolValue = True
canvas = qgis.utils.iface.mapCanvas()
spatialReference = osr.SpatialReference() #will create a spatial reference locally to tell the system what the reference will be
spatialReference.ImportFromEPSG(4326) #here we define this reference to be the EPSG code
driver = osgeo.ogr.GetDriverByName('ESRI Shapefile') # will select the driver for our shp-file creation.
export_shp = r'E:\Folder'
shapeData = driver.CreateDataSource(export_shp) #so there we will store our data
outLayer = shapeData.CreateLayer(str('WaypointLayer'), spatialReference, osgeo.ogr.wkbPoint) #this will create a corresponding layer for our data with given spatial information.
outDataSource = driver.Open(export_shp, 1)
outLayer = outDataSource.GetLayer()
layer_defn = outLayer.GetLayerDefn() # gets parameters of the current shapefile
index = 0
# Reading the CSV file
cnt = 0
input_file = r'E:\Folder\CSV\points.csv'
with open(input_file, 'rb') as csvfile:
readerDict = csv.DictReader(csvfile)
# Create the fields in the layer
for header in readerDict.fieldnames:
field_name = ogr.FieldDefn(header, ogr.OFTString)
field_name.SetWidth(50)
outLayer.CreateField(field_name)
for row in readerDict:
print (row['latitude'], row['longitude'])
point = osgeo.ogr.Geometry(osgeo.ogr.wkbPoint)
point.AddPoint(float(row['longitude']), float(row['latitude'])) #we do have LATs and LONs as Strings, so we convert them
feature = osgeo.ogr.Feature(layer_defn)
feature.SetGeometry(point) #set the coordinates
feature.SetFID(index)
for field in readerDict.fieldnames:
i = feature.GetFieldIndex(field)
feature.SetField(i, row[field])
outLayer.CreateFeature(feature)
index += 1
shapeData.Destroy()
outDataSource.Destroy()
conLayer = QgsVectorLayer(export_shp,'pointLayer',"ogr")
QgsMapLayerRegistry.instance().addMapLayer(conLayer)
canvas.refresh()
The problem is that initially it is not updating the data in the attribute table but it creates the features when I executed once. When I executed again it is updating the field data. It sometimes create duplicates also.
What I am doing wrong here?
