corrected the resolver

Author: akshatjain1004

backend/family_tree/student/schema.py

@@ -8,15 +8,26 @@ class Meta:
         model= Student
         fields= ("id","name","branch","year","picture","homeTown","extraCurriculars","socialMedia", "linkedIn","email","parentId","roll_no")
         filter_fields=["id","name","branch","year","email","parentId","roll_no"]
-
+def tree_for_batch(root,info,batch):
+    tree_for_batch=[]
+    for student in batch:
+        roll= student.roll_no
+        pathObjects=[]
+        while(Student.objects.get(roll_no=roll).parentId!="root"):
+            student=Student.objects.get(roll_no=roll)
+            pathObjects.append(student)
+            roll= student.parentId
+        pathObjects.append(Student.objects.get(roll_no=roll))
+        tree_for_batch.append(pathObjects)
+    return tree_for_batch
 
 class Query(graphene.ObjectType):
     students=graphene.List(StudentType)
     children=graphene.List(StudentType, parent_id=graphene.String())
     student_path= graphene.List(StudentType, roll=graphene.String())
     student_sibling= graphene.List(StudentType, roll=graphene.String())
     student_search= graphene.List(StudentType, search_query=graphene.String())
-    student_batch= graphene.List(StudentType, roll=graphene.String())
+    student_batch= graphene.List(graphene.List(StudentType), roll=graphene.String())
 
     def resolve_students(root,info):
         return Student.objects.all()
@@ -30,7 +41,7 @@ def resolve_student_path(root, info, roll):
             student=Student.objects.get(roll_no=roll)
             pathObjects.append(student)
             roll= student.parentId
-            pathObjects.append(Student.objects.get(roll_no=roll))
+        pathObjects.append(Student.objects.get(roll_no=roll))
         return pathObjects
 
     def resolve_student_sibling(root,info, roll):
@@ -42,10 +53,11 @@ def resolve_student_batch(root,info,roll):
         current_node= Student.objects.get(roll_no=roll)
         year_of_node= current_node.year
         current_batch= Student.objects.filter(year=year_of_node)
-        tree_for_batch=[]
-        for person in current_batch:
-            tree_for_batch.append(Query.resolve_student_path(root,info,person.roll_no))
-        return tree_for_batch
+        # tree_for_batch=[]
+        # for person in current_batch:
+        #     tree_for_batch.append(Query.resolve_student_path(root,info,person.roll_no))
+        # return tree_for_batch
+        return tree_for_batch(root,info,current_batch)
 
 
     def resolve_student_search(root,info, search_query):