gistfile1.txt

class SaveAgencyList(APIView):
    def get(self, request, format=None):
        saved_agencies = SaveAgency.objects.filter(created_by=request.user).prefetch_related("saved_agency")
        o = []
        for agency in saved_agencies:
            agency_id = agency.saved_agency.id
            try:
                agency_query = Agency.objects.get(id=agency_id)
                o = o + [agency_query]
            except:
                pass
        serializer = AgencySerializer(o, many=True)
        print serializer.data
        return Response(serializer.data)
    @login_required(login_url='/1971/accounts/login/')
    def post(self, request, format=None):
        user = request.user
        data = request.data
        print data["id"]
        agency_id = data["id"]
        agency = Agency.objects.get(id=agency_id)
        if SaveAgency.objects.filter(saved_agency=agency, created_by=user).exists()==True:
            messages.add_message(request._request, messages.SUCCESS, 'You have already saved this agency, choose another one')
        else:
            save_agency = SaveAgency(saved_agency=agency, created_by=user, modified_by=user)
            save_agency.save()
        return HttpResponse(status=201)