# simple script to convert csv to shapefile.
from osgeo import ogr
import csv

driver = ogr.GetDriverByName('ESRI Shapefile')

def loadCSV(path=None):
    global driver
    print("loading csv file %s" % path)
    fileReader = csv.reader(open(path), delimiter=',',)
    shape = driver.CreateDataSource("%s.shp" % path)
    header = fileReader.next()
    layer = shape.CreateLayer(path,
                              geom_type=ogr.wkbPoint)
    layer.CreateField(ogr.FieldDefn("name",ogr.OFTString))
    layer.CreateField(ogr.FieldDefn("pop",ogr.OFTInteger))
    for line in fileReader:
        feature = ogr.Feature(layer.GetLayerDefn())
        feature.SetField("name",line[0])
        feature.SetField("pop",line[3])
        point = ogr.Geometry(ogr.wkbPoint)
        point.SetPoint_2D(0,float(line[2]),float(line[1]))
        feature.SetGeometry(point)
        layer.CreateFeature(feature)
        feature.Destroy()

loadCSV(path="demographics.csv")