From 328f2bfcf0501fbf513b9265bf56fcea75efa25c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alejandro=20G=C3=B3mez?= Date: Sun, 2 Jun 2013 10:57:37 +0200 Subject: [PATCH] Return also the projects owned by the logged in user in the projects API endpoint --- greenmine/scrum/api.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/greenmine/scrum/api.py b/greenmine/scrum/api.py index 376b7eef..9643d27e 100644 --- a/greenmine/scrum/api.py +++ b/greenmine/scrum/api.py @@ -1,7 +1,8 @@ # -*- coding: utf-8 -*- -import django_filters +from django.db.models import Q +import django_filters from rest_framework import generics from rest_framework.permissions import IsAuthenticated @@ -52,7 +53,9 @@ class ProjectList(generics.ListCreateAPIView): permission_classes = (IsAuthenticated,) def get_queryset(self): - return self.model.objects.filter(members=self.request.user) + return self.model.objects.filter( + Q(owner=self.request.user) | Q(members=self.request.user) + ) def pre_save(self, obj): obj.owner = self.request.user