OpenConceptLab/ocl_issues#991 | API get parents of a concept

Author: snyaggarwal

core/concepts/models.py

@@ -805,6 +805,12 @@ def child_concept_queryset(self):
             return Concept.objects.filter(uri__in=urls)
         return Concept.objects.none()
 
+    def parent_concept_queryset(self):
+        urls = self.parent_concept_urls
+        if urls:
+            return Concept.objects.filter(uri__in=urls)
+        return Concept.objects.none()
+
     @staticmethod
     def __format_hierarchy_uris(uris):
         return list({drop_version(uri) for uri in uris})

core/concepts/serializers.py

@@ -481,3 +481,27 @@ class ConceptHierarchySerializer(ModelSerializer):
     class Meta:
         model = Concept
         fields = ('uuid', 'id', 'url', 'children', 'name')
+
+
+class ConceptChildrenSerializer(ModelSerializer):
+    uuid = CharField(source='id')
+    id = EncodedDecodedCharField(source='mnemonic')
+    url = CharField(source='uri')
+    children = ListField(source='child_concept_urls')
+    name = CharField(source='display_name')
+
+    class Meta:
+        model = Concept
+        fields = ('uuid', 'id', 'url', 'children', 'name')
+
+
+class ConceptParentsSerializer(ModelSerializer):
+    uuid = CharField(source='id')
+    id = EncodedDecodedCharField(source='mnemonic')
+    url = CharField(source='uri')
+    parents = ListField(source='parent_concept_urls')
+    name = CharField(source='display_name')
+
+    class Meta:
+        model = Concept
+        fields = ('uuid', 'id', 'url', 'parents', 'name')

core/concepts/tests/tests.py

@@ -951,6 +951,35 @@ def test_child_concept_queryset(self):
             list(child_child_concept.child_concept_queryset().values_list('uri', flat=True)), [])
         self.assertEqual(child_child_concept.parent_concept_urls, [child_concept.uri])
 
+    def test_parent_concept_queryset(self):
+        parent_concept = ConceptFactory()
+        self.assertEqual(parent_concept.parent_concept_queryset().count(), 0)
+        self.assertEqual(parent_concept.parent_concept_urls, [])
+
+        child_concept = Concept.persist_new({
+            **factory.build(dict, FACTORY_CLASS=ConceptFactory), 'mnemonic': 'c1', 'parent': parent_concept.parent,
+            'names': [LocalizedTextFactory.build(locale='en', name='English', locale_preferred=True)],
+            'parent_concept_urls': [parent_concept.uri]
+        })
+        self.assertEqual(
+            list(parent_concept.parent_concept_queryset().values_list('uri', flat=True)), [])
+        self.assertEqual(
+            list(child_concept.parent_concept_queryset().values_list('uri', flat=True)), [parent_concept.uri])
+        self.assertEqual(child_concept.parent_concept_urls, [parent_concept.uri])
+
+        child_child_concept = Concept.persist_new({
+            **factory.build(dict, FACTORY_CLASS=ConceptFactory), 'mnemonic': 'c2', 'parent': parent_concept.parent,
+            'names': [LocalizedTextFactory.build(locale='en', name='English', locale_preferred=True)],
+            'parent_concept_urls': [child_concept.uri]
+        })
+        self.assertEqual(
+            list(parent_concept.parent_concept_queryset().values_list('uri', flat=True)), [])
+        self.assertEqual(
+            list(child_concept.parent_concept_queryset().values_list('uri', flat=True)), [parent_concept.uri])
+        self.assertEqual(
+            list(child_child_concept.parent_concept_queryset().values_list('uri', flat=True)), [child_concept.uri])
+        self.assertEqual(child_child_concept.parent_concept_urls, [child_concept.uri])
+
 
 class OpenMRSConceptValidatorTest(OCLTestCase):
     def setUp(self):

core/concepts/urls.py

@@ -30,6 +30,11 @@
         views.ConceptChildrenView.as_view(),
         name='concept-children'
     ),
+    path(
+        "<str:concept>/parents/",
+        views.ConceptParentsView.as_view(),
+        name='concept-parents'
+    ),
     path('<str:concept>/atom/', ConceptFeed()),
     path(
         "<str:concept>/descriptions/",

core/concepts/views.py

@@ -31,7 +31,8 @@
 from core.concepts.serializers import (
     ConceptDetailSerializer, ConceptListSerializer, ConceptDescriptionSerializer, ConceptNameSerializer,
     ConceptVersionDetailSerializer,
-    ConceptVersionListSerializer, ConceptHierarchySerializer, ConceptSummarySerializer, ConceptMinimalSerializer)
+    ConceptVersionListSerializer, ConceptSummarySerializer, ConceptMinimalSerializer,
+    ConceptChildrenSerializer, ConceptParentsSerializer)
 from core.mappings.serializers import MappingListSerializer
 
 
@@ -279,14 +280,23 @@ def destroy(self, request, *args, **kwargs):
 
 
 class ConceptChildrenView(ConceptBaseView, ListAPIView):
-    serializer_class = ConceptHierarchySerializer
+    serializer_class = ConceptChildrenSerializer
 
     def get_queryset(self):
         instance = get_object_or_404(super().get_queryset(), id=F('versioned_object_id'))
         self.check_object_permissions(self.request, instance)
         return instance.child_concept_queryset()
 
 
+class ConceptParentsView(ConceptBaseView, ListAPIView):
+    serializer_class = ConceptParentsSerializer
+
+    def get_queryset(self):
+        instance = get_object_or_404(super().get_queryset(), id=F('versioned_object_id'))
+        self.check_object_permissions(self.request, instance)
+        return instance.parent_concept_queryset()
+
+
 class ConceptReactivateView(ConceptBaseView, UpdateAPIView):
     serializer_class = ConceptDetailSerializer