Skip to content

Commit f39d161

Browse files
committed
2 parents 8e6eb2a + dd7ab3e commit f39d161

File tree

2 files changed

+58
-14
lines changed

2 files changed

+58
-14
lines changed

backend/family_tree/student/schema.py

Lines changed: 27 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,20 +8,32 @@ class Meta:
88
model= Student
99
fields= ("id","name","branch","year","picture","homeTown","extraCurriculars","socialMedia", "linkedIn","email","parentId","roll_no")
1010
filter_fields=["id","name","branch","year","email","parentId","roll_no"]
11-
11+
def tree_for_batch(root,info,batch):
12+
tree_for_batch=[]
13+
for student in batch:
14+
roll= student.roll_no
15+
pathObjects=[]
16+
while(Student.objects.get(roll_no=roll).parentId!="root"):
17+
student=Student.objects.get(roll_no=roll)
18+
pathObjects.append(student)
19+
roll= student.parentId
20+
pathObjects.append(Student.objects.get(roll_no=roll))
21+
tree_for_batch.append(pathObjects)
22+
return tree_for_batch
1223

1324
class Query(graphene.ObjectType):
1425
students=graphene.List(StudentType)
15-
children=graphene.List(StudentType, parent_id=graphene.String())
26+
children=graphene.List(StudentType, parentId=graphene.String())
1627
student_path= graphene.List(StudentType, roll=graphene.String())
1728
student_sibling= graphene.List(StudentType, roll=graphene.String())
1829
student_search= graphene.List(StudentType, search_query=graphene.String())
30+
student_batch= graphene.List(graphene.List(StudentType), roll=graphene.String())
1931

2032
def resolve_students(root,info):
2133
return Student.objects.all()
2234

23-
def resolve_children(parent_id):
24-
return Student.objects.filter(parentId=parent_id)
35+
def resolve_children(root, info, parentId):
36+
return Student.objects.filter(parentId=parentId)
2537

2638
def resolve_student_path(root, info, roll):
2739
pathObjects=[]
@@ -37,6 +49,17 @@ def resolve_student_sibling(root,info, roll):
3749
return Student.objects.filter(parentId=student.parentId).exclude(roll_no=roll)
3850

3951

52+
def resolve_student_batch(root,info,roll):
53+
current_node= Student.objects.get(roll_no=roll)
54+
year_of_node= current_node.year
55+
current_batch= Student.objects.filter(year=year_of_node)
56+
# tree_for_batch=[]
57+
# for person in current_batch:
58+
# tree_for_batch.append(Query.resolve_student_path(root,info,person.roll_no))
59+
# return tree_for_batch
60+
return tree_for_batch(root,info,current_batch)
61+
62+
4063
def resolve_student_search(root,info, search_query):
4164
return Student.objects.filter(Q(name__icontains=search_query) | Q(roll_no__icontains= search_query))[0:8]
4265

backend/family_tree/student/tests.py

Lines changed: 31 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,37 @@
11
from django.test import TestCase
2-
import json
32
from graphene_django.utils.testing import GraphQLTestCase
3+
import json
44
from student.models import Student
55
from mixer.backend.django import mixer
66

7-
class testCasesForStudents(GraphQLTestCase):
7+
8+
class StudentTestCases(GraphQLTestCase):
89
GRAPHQL_URL = 'http://localhost/graphql'
10+
911
def setUp(self):
10-
super().setUp()
11-
self.student1 = mixer.blend(Student, name='student1', roll_no='1', parentId='root')
12-
self.student2 = mixer.blend(Student, name='student2', roll_no='2', parentId='1')
13-
self.student3 = mixer.blend(Student, name='student3', roll_no='3', parentId='1')
14-
self.student4 = mixer.blend(Student, name='student4', roll_no='4', parentId='1')
12+
super().setUp()
13+
self.student1 = mixer.blend(Student, name='student1', roll_no='1', parentId='root')
14+
self.student2 = mixer.blend(Student, name='student2', roll_no='2', parentId='1')
15+
self.student3 = mixer.blend(Student, name='student3', roll_no='3', parentId='1')
16+
self.student4 = mixer.blend(Student, name='student4', roll_no='4', parentId='1')
17+
18+
19+
def test_children_query(self):
20+
response = self.query('''
21+
query {
22+
children(parentId: "1"){
23+
rollNo
24+
name
25+
}
26+
}
27+
''')
28+
29+
content = json.loads(response.content)
30+
self.assertResponseNoErrors(response)
31+
self.assertDictEqual(content['data']['children'][0], {'rollNo': '2', 'name': 'student2'})
32+
self.assertDictEqual(content['data']['children'][1], {'rollNo': '3', 'name': 'student3'})
33+
self.assertDictEqual(content['data']['children'][2], {'rollNo': '4', 'name': 'student4'})
34+
1535
def test_student_query(self):
1636
response = self.query('''
1737
query {
@@ -21,10 +41,11 @@ def test_student_query(self):
2141
}
2242
}
2343
''')
24-
content= json.loads(response.content)
44+
content = json.loads(response.content)
2545
self.assertResponseNoErrors(response)
2646
self.assertDictEqual(content['data']['students'][0], {'id': str(self.student1.id), 'name': self.student1.name})
27-
self.assertDictEqual(content['data']['students'][1], {'id': str(self.student2.id), 'name': self.student2.name})
47+
self.assertDictEqual(content['data']['students'][1], {'id': str(self.student2.id), 'name': self.student2.name})\
48+
2849
def test_student_sibling(self):
2950
response = self.query('''
3051
query {
@@ -37,4 +58,4 @@ def test_student_sibling(self):
3758
content= json.loads(response.content)
3859
self.assertResponseNoErrors(response)
3960
self.assertDictEqual(content['data']['studentSibling'][0], {'id': str(self.student3.id), 'name': self.student3.name})
40-
self.assertDictEqual(content['data']['studentSibling'][1], {'id': str(self.student4.id), 'name': self.student4.name})
61+
self.assertDictEqual(content['data']['studentSibling'][1], {'id': str(self.student4.id), 'name': self.student4.name})

0 commit comments

Comments
 (0)