PyQgis overwrite shapefile and adding layer to QGIS

Question

I am creating a shapefile in Python by:

driver = ogr.GetDriverByName('ESRI Shapefile')
datasource = driver.CreateDataSource('c:/temp/toke2.shp')
layer = datasource.CreateLayer('layerName',geom_type=ogr.wkbLineString)

If the file toke2.shp doesn't exist, it will be created. If it exists, I can't run this code and overwrite the old file. I need an overwrite function or a delete function. I run QGIS in admin mode.

My other problem is when I want to add the shapefile to QGIS by:

vlayer = QgsVectorLayer("c:/temp/toke2.shp", "toke2", "ogr")
QgsMapLayerRegistry.instance().addMapLayer(vlayer)

If I add the file in QGIS by "Layer > add vector layer", it works fine. When I use the code in the python console in QGIS it also works fine. But if I use this code in my plugin the file is added to QGIS, but it is empty. So for some reason it doesn't add the data points in the file, it is just adding an empty shapefile.

In my plugin I import this:

 from osgeo import ogr
 from osgeo import osr
 import math
 from PyQt4.QtCore import *
 from PyQt4.QtGui import *
 from qgis.core import *
 import qgis.utils
 import resources_rc

Does anyone have a solution to these problems?

Answer 1

You can check for existence of a file and remove it using:

import os

if os.path.isfile('c:/temp/toke2.shp'):
   os.remove('c:/temp/toke2.shp')

Answer 2

I made it work by using memory to create a layer instead of making a shp file.