Filter owned templates and display system templates for everyone

recircle/template/views.py

@@ -3,6 +3,7 @@ from rest_framework.viewsets import ModelViewSet
 
 
 from template.serializers import ImageTemplateModelSerializer, ImageTemplateListItemSerializer
 from template.serializers import ImageTemplateModelSerializer, ImageTemplateListItemSerializer
 from template.models import ImageTemplate
 from template.models import ImageTemplate
+from django.db.models import Q
 
 
 
 
 class ImageTemplateViewSet(ModelViewSet):
 class ImageTemplateViewSet(ModelViewSet):
@@ -41,7 +42,7 @@ class ImageTemplateViewSet(ModelViewSet):
         return super(ImageTemplateViewSet, self).update(request, partial=True)
         return super(ImageTemplateViewSet, self).update(request, partial=True)
 
 
     def list(self, request, *args, **kwargs):
     def list(self, request, *args, **kwargs):
-        # TODO: Filter for owned and shared tempaltes
-        query = ImageTemplate.objects.all()
+        # TODO: Filter for shared tempaltes
+        query = ImageTemplate.objects.filter(Q(created_by=request.user.id) | Q(type="SYSTEM"))
         serializer = ImageTemplateListItemSerializer(query, many=True)
         serializer = ImageTemplateListItemSerializer(query, many=True)
         return Response(serializer.data)
         return Response(serializer.data)