Return also the projects owned by the logged in user in the projects API endpoint

remotes/origin/enhancement/email-actions
Alejandro Gómez 2013-06-02 10:57:37 +02:00
parent 621a636da2
commit 328f2bfcf0
1 changed files with 5 additions and 2 deletions

View File

@ -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