def branches(request, id=0):
if request.method == 'GET':
all_branches = Branch.objects.get(user=request.user)
serializer = BranchSerializer(all_branches)
return Response(serializer.data)
elif request.method == 'POST':
serializer = BranchSerializer(data=request.DATA)
if serializer.is_valid():
serializer.data.user = request.user
serializer.save()
return Response(serializer.data, status=status.HTTP_201_CREATED)
else:
return Response(serializer.errors,
status=status.HTTP_400_BAD_REQUEST)
elif request.method == 'PUT':
branch = Branch(id=id)
serializer = BranchSerializer(branch, data=request.DATA)
if serializer.is_valid():
serializer.data.user = request.user
serializer.save()
return Response(serializer.data, status=status.HTTP_201_CREATED)
else:
return Response(serializer.errors,
status=status.HTTP_400_BAD_REQUEST)
elif request.method == 'DELETE':
branch = Branch(id=id)
branch.delete()
return Response("DELETED", status=status.HTTP_200_OK)
