def create(self, request, **kwargs):
serializer = BranchSerializer(data=request.data)
if serializer.is_valid():
user = save_user(request.data, UserPermissions.IS_BRANCH)
company = Company(**request.data["company"])
city = City(**request.data["city"])
branch = Branch(
nickName=request.data["nickName"],
type=UserType.BRANCH,
address=request.data["address"],
phone=request.data["phone"],
company=company,
city=city,
user=user,
)
branch.save()
return Response(
{
"status": "SUCCESS",
"msg_status": "Sucursal creada satisfactoriamente. Puede ingresar con su nueva contrasena.",
}
)
else:
messages = []
make_error(serializer.errors.values(), messages)
return Response({"status": "FAILURE", "msg_status": messages})
def branch_info(request, id=0):
branch = Branch.objects.get(pk=id)
serializer = BranchSerializer(branch)
return Response(serializer.data)
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)
