def children(self, request, pk=None):
try:
organization = Organization.objects.get(pk=pk)
children = organization.children()
response = OrganizationSerializer(children, many=True)
return Response(response.data, status=status.HTTP_200_OK)
except Organization.DoesNotExist:
return Response("Organization does not exist",
status=status.HTTP_404_NOT_FOUND)
def parents(self, request, pk=None):
"""
Возвращает родителей запрашиваемой организации
TODO: Написать два действия для ViewSet (parents и children), используя методы модели
"""
try:
organization = Organization.objects.get(pk=pk)
parents = organization.parents()
response = OrganizationSerializer(parents, many=True)
return Response(response.data, status=status.HTTP_200_OK)
except Organization.DoesNotExist:
return Response("Organization does not exist",
status=status.HTTP_404_NOT_FOUND)