add organization and user to related instances of dataset document

Author: Deepthi-Chand

search/documents/dataset_document.py

@@ -2,8 +2,16 @@
 
 from django_elasticsearch_dsl import Document, Index, KeywordField, fields
 
-from api.models import Dataset, DatasetMetadata, Metadata, Resource, Sector
+from api.models import (
+    Dataset,
+    DatasetMetadata,
+    Metadata,
+    Organization,
+    Resource,
+    Sector,
+)
 from api.utils.enums import DatasetStatus
+from authorization.models import User
 from DataSpace import settings
 from search.documents.analysers import html_strip, ngram_analyser
 
@@ -164,7 +172,10 @@ def get_queryset(self) -> Any:
         )
 
     def get_instances_from_related(
-        self, related_instance: Union[Resource, Metadata, DatasetMetadata, Sector]
+        self,
+        related_instance: Union[
+            Resource, Metadata, DatasetMetadata, Sector, Organization, User
+        ],
     ) -> Optional[Union[Dataset, List[Dataset]]]:
         """Get Dataset instances from related models."""
         if isinstance(related_instance, Resource):
@@ -176,6 +187,10 @@ def get_instances_from_related(
             return related_instance.dataset
         elif isinstance(related_instance, Sector):
             return list(related_instance.datasets.all())
+        elif isinstance(related_instance, Organization):
+            return list(related_instance.datasets.all())
+        elif isinstance(related_instance, User):
+            return list(related_instance.datasets.all())
         return None
 
     class Django:
@@ -189,4 +204,11 @@ class Django:
             "modified",
         ]
 
-        related_models = [Resource, Metadata, DatasetMetadata, Sector]
+        related_models = [
+            Resource,
+            Metadata,
+            DatasetMetadata,
+            Sector,
+            Organization,
+            User,
+        ]