fix: Only return valid taxonomy term types

techmatters · Sep 26, 2023 · 1f9430d · 1f9430d
1 parent ee432e0
commit 1f9430d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/terraso_backend/apps/graphql/schema/taxnomy_terms.py b/terraso_backend/apps/graphql/schema/taxnomy_terms.py
@@ -24,6 +24,8 @@
 
 logger = structlog.get_logger(__name__)
 
+VALID_TYPES = [key for (key, text) in TaxonomyTerm.TYPES]
+
 
 class TaxonomyTermNode(DjangoObjectType):
     id = graphene.ID(source="pk", required=True)
@@ -42,3 +44,7 @@ class Meta:
         )
         interfaces = (relay.Node,)
         connection_class = TerrasoConnection
+
+    @classmethod
+    def get_queryset(cls, queryset, info):
+        return queryset.filter(type__in=VALID_TYPES)