def branch_list(request, companyId): #print(request.data) company = Company.objects.get(pk=companyId) if request.method == 'GET': branches = company.branches.all().order_by('-created_at') branches = BranchSerializer(branches,many=True) print(branches.data) print("hello") return Response(branches.data, status=status.HTTP_200_OK) if request.method == 'POST': serializer = BranchSerializer(data=request.data,context={ "company": company }) if serializer.is_valid(): serializer.save() return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def branch_detail(request, companyId, branchId): # print(request.data) try: company=Company.objects.get(pk=companyId) branch=company.branches.get(pk=branchId) except Company.DoesNotExist or Branch.DoesNotExist: return Response(status=status.HTTP_404_NOT_FOUND) if request.method == 'GET': serializer = BranchSerializer(branch) return Response(serializer.data) elif request.method == 'PUT': serializer = BranchSerializer(branch, data=request.data) if serializer.is_valid(): serializer.save() return Response(serializer.data) else: return Response(status=status.HTTP_400_BAD_REQUEST) elif request.method == 'DELETE': print(branch) branch.delete() return Response(status=status.HTTP_204_NO_CONTENT)