From eb8eb311d698107844e19cdf3030275308c1e1dd Mon Sep 17 00:00:00 2001 From: Tony Mack Date: Tue, 4 Feb 2014 20:50:39 -0500 Subject: [PATCH] add rbac to rest api views --- planetstack/apigen/api.template.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/planetstack/apigen/api.template.py b/planetstack/apigen/api.template.py index 81a1eca..d7570e1 100644 --- a/planetstack/apigen/api.template.py +++ b/planetstack/apigen/api.template.py @@ -53,11 +53,17 @@ serializerLookUp = { {% for object in generator.all %} class {{ object.camel }}List(generics.ListCreateAPIView): - queryset = {{ object.camel }}.objects.all() + #queryset = {{ object.camel }}.objects.all() serializer_class = {{ object.camel }}Serializer + + def get_queryset(self): + return {{ object.camel }}.select_by_user(self.request.user class {{ object.camel }}Detail(generics.RetrieveUpdateDestroyAPIView): - queryset = {{ object.camel }}.objects.all() + #queryset = {{ object.camel }}.objects.all() serializer_class = {{ object.camel }}Serializer + + def get_queryset(self): + return {{ object.camel }}.select_by_user(self.request.user) {% endfor %} -- 2.43.0