Add method to retrieve sink

qgis · May 3, 2017 · 96ad511 · 96ad511
1 parent fde71cf
commit 96ad511
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/python/core/qgsfeaturesink.sip b/python/core/qgsfeaturesink.sip
@@ -80,6 +80,12 @@ class QgsProxyFeatureSink : QgsFeatureSink
     virtual bool addFeatures( QgsFeatureIterator &iterator );
 
 
+    QgsFeatureSink *destinationSink();
+%Docstring
+ Returns the destination QgsFeatureSink which the proxy will forward features to.
+ :rtype: QgsFeatureSink
+%End
+
 };
 
 

diff --git a/src/core/qgsfeaturesink.h b/src/core/qgsfeaturesink.h
@@ -84,6 +84,11 @@ class CORE_EXPORT QgsProxyFeatureSink : public QgsFeatureSink
     bool addFeatures( QgsFeatureList &features ) override;
     bool addFeatures( QgsFeatureIterator &iterator ) override;
 
+    /**
+     * Returns the destination QgsFeatureSink which the proxy will forward features to.
+     */
+    QgsFeatureSink *destinationSink() { return mSink; }
+
   private:
 
     QgsFeatureSink *mSink;

diff --git a/tests/src/python/test_qgsfeaturesink.py b/tests/src/python/test_qgsfeaturesink.py
@@ -74,6 +74,7 @@ def testProxyFeatureSink(self):
 
         store = QgsFeatureStore(fields, QgsCoordinateReferenceSystem())
         proxy = QgsProxyFeatureSink(store)
+        self.assertEqual(proxy.destinationSink(), store)
 
         self.assertEqual(len(store), 0)