Selecting features by mouse click using PyQGIS

Question

I am following a tutorial where I am trying to learn how to use PyQGIS to programmatically accept a user's feature selection from the map canvas of QGIS by mouse click. I have encountered a problem with the following code snippet where, as I understand it 'setSelectedFeatures' is no longer an attribute for a QgsVectorLayer.
I have some other posts that suggest the current method in PyQGIS is to use 'selectByIds' but cannot work out how to set up the code to incorporate that correctly, despite looking through the documentation (pygqis documentation).
Has anyone dealt with this type of functionality and could offer some guidance?
 if len(found_features) > 0:
        layer = found_features[0].mLayer
        feature = found_features[0].mFeature
        layer.setSelectedFeatures([feature.id()])

else:
     self.layer.removeSelection()

Ben W · Accepted Answer

You haven't specified or linked to the tutorial you are following, but I'm guessing you are doing something like my example below, sub-classing  QgsMapToolIdentifyFeature.
To use the selectByIds() method you just need to pass a list of integers containing the feature id numbers you want to select. You can also pass an optional argument to set the selection behaviour (SetSelection is the default). See the documentation for the complete list of options.
You can run my example below from the Python console and get an idea of example usage.
class selectTool(QgsMapToolIdentifyFeature):
    
    def __init__(self, iface):
        self.iface = iface
        self.canvas = self.iface.mapCanvas()
        self.layer = self.iface.activeLayer()
        QgsMapToolIdentifyFeature.__init__(self, self.canvas, self.layer)
        self.iface.currentLayerChanged.connect(self.active_changed)
        
    def active_changed(self, layer):
        self.layer.removeSelection()
        if isinstance(layer, QgsVectorLayer) and layer.isSpatial():
            self.layer = layer
            self.setLayer(self.layer)
            
    def canvasPressEvent(self, event):
        found_features = self.identify(event.x(), event.y(), [self.layer], QgsMapToolIdentify.TopDownAll)
        self.layer.selectByIds([f.mFeature.id() for f in found_features], QgsVectorLayer.AddToSelection)
        
    def deactivate(self):
        self.layer.removeSelection()

t = selectTool(iface)
iface.mapCanvas().setMapTool(t)

Selecting features by mouse click using PyQGIS

One Answer

Add your own answers!

Ask a Question