33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from django.db.models import Q
|
|
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
from greenmine.base import filters
|
|
from greenmine.base.api import ModelCrudViewSet,
|
|
from greenmine.base.notifications.api import NotificationSenderMixin
|
|
|
|
from . import serializers
|
|
from . import models
|
|
from . import permissions
|
|
|
|
|
|
class ProjectViewSet(NotificationSenderMixin, ModelCrudViewSet):
|
|
model = models.Project
|
|
serializer_class = serializers.ProjectSerializer
|
|
permission_classes = (IsAuthenticated, permissions.ProjectPermission)
|
|
create_notification_template = "create_project_notification"
|
|
update_notification_template = "update_project_notification"
|
|
destroy_notification_template = "destroy_project_notification"
|
|
|
|
def get_queryset(self):
|
|
qs = super(ProjectViewSet, self).get_queryset()
|
|
qs = qs.filter(Q(owner=self.request.user) |
|
|
Q(members=self.request.user))
|
|
return qs.distinct()
|
|
|
|
def pre_save(self, obj):
|
|
super(ProjectViewSet, self).pre_save(obj)
|
|
obj.owner = self.request.user
|