Generated QgsRelations now support getting the related features

qgis · Jan 12, 2021 · 53ca67a · 53ca67a
1 parent 89c07ff
commit 53ca67a
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/core/qgsrelation.cpp b/src/core/qgsrelation.cpp
@@ -21,6 +21,8 @@
 #include "qgsproject.h"
 #include "qgsvectorlayer.h"
 #include "qgsrelation_p.h"
+#include "qgspolymorphicrelation.h"
+#include "qgsrelationmanager.h"
 
 QgsRelation::QgsRelation()
   : d( new QgsRelationPrivate() )
@@ -202,6 +204,21 @@ QString QgsRelation::getRelatedFeaturesFilter( const QgsFeature &feature ) const
 {
   QStringList conditions;
 
+  if ( ! d->mPolymorphicRelationId.isEmpty() )
+  {
+    QgsPolymorphicRelation polymorphicRelation = QgsProject::instance()->relationManager()->polymorphicRelation( d->mPolymorphicRelationId );
+
+    if ( polymorphicRelation.isValid() )
+    {
+      conditions << QgsExpression::createFieldEqualityExpression( polymorphicRelation.referencedLayerField(), polymorphicRelation.layerRepresentation( referencedLayer() ) );
+    }
+    else
+    {
+      QgsDebugMsg( "The polymorphic relation is invalid" );
+      conditions << QStringLiteral( " FALSE " );
+    }
+  }
+
   for ( const FieldPair &pair : qgis::as_const( d->mFieldPairs ) )
   {
     QVariant val( feature.attribute( pair.referencedField() ) );