129 lines
4.3 KiB
Python
129 lines
4.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from django.db.models import Q
|
|
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
from rest_framework.decorators import detail_route
|
|
|
|
from greenmine.base import filters
|
|
from greenmine.base.api import ModelCrudViewSet, ModelListViewSet
|
|
from greenmine.base.notifications.api import NotificationSenderMixin
|
|
|
|
from . import serializers
|
|
from . import models
|
|
from . import permissions
|
|
|
|
from .aggregates import stats
|
|
from .aggregates import filters as filters_aggr
|
|
|
|
|
|
class ProjectViewSet(ModelCrudViewSet):
|
|
model = models.Project
|
|
serializer_class = serializers.ProjectDetailSerializer
|
|
list_serializer_class = serializers.ProjectSerializer
|
|
permission_classes = (IsAuthenticated, permissions.ProjectPermission)
|
|
|
|
@detail_route(methods=['get'])
|
|
def stats(self, request, pk=None):
|
|
project = self.get_object()
|
|
return Response(stats.get_stats_for_project(project))
|
|
|
|
@detail_route(methods=['get'])
|
|
def issues_stats(self, request, pk=None):
|
|
project = self.get_object()
|
|
return Response(stats.get_stats_for_project_issues(project))
|
|
|
|
@detail_route(methods=['get'])
|
|
def issue_filters_data(self, request, pk=None):
|
|
project = self.get_object()
|
|
return Response(filters_aggr.get_issues_filters_data(project))
|
|
|
|
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):
|
|
obj.owner = self.request.user
|
|
super(ProjectViewSet, self).pre_save(obj)
|
|
|
|
|
|
class MembershipViewSet(ModelCrudViewSet):
|
|
model = models.Membership
|
|
serializer_class = serializers.MembershipSerializer
|
|
permission_classes = (IsAuthenticated, permissions.MembershipPermission)
|
|
|
|
|
|
# User Stories commin ViewSets
|
|
|
|
class PointsViewSet(ModelListViewSet):
|
|
model = models.Points
|
|
serializer_class = serializers.PointsSerializer
|
|
permission_classes = (IsAuthenticated, permissions.PointsPermission)
|
|
filter_backends = (filters.IsProjectMemberFilterBackend,)
|
|
filter_fields = ('project',)
|
|
|
|
|
|
class UserStoryStatusViewSet(ModelListViewSet):
|
|
model = models.UserStoryStatus
|
|
serializer_class = serializers.UserStoryStatusSerializer
|
|
permission_classes = (IsAuthenticated, permissions.UserStoryStatusPermission)
|
|
filter_backends = (filters.IsProjectMemberFilterBackend,)
|
|
filter_fields = ('project',)
|
|
|
|
|
|
# Tasks commin ViewSets
|
|
|
|
class TaskStatusViewSet(ModelListViewSet):
|
|
model = models.TaskStatus
|
|
serializer_class = serializers.TaskStatusSerializer
|
|
permission_classes = (IsAuthenticated, permissions.TaskStatusPermission)
|
|
filter_backends = (filters.IsProjectMemberFilterBackend,)
|
|
filter_fields = ("project",)
|
|
|
|
|
|
# Issues common ViewSets
|
|
|
|
class SeverityViewSet(ModelListViewSet):
|
|
model = models.Severity
|
|
serializer_class = serializers.SeveritySerializer
|
|
permission_classes = (IsAuthenticated, permissions.SeverityPermission)
|
|
filter_backends = (filters.IsProjectMemberFilterBackend,)
|
|
filter_fields = ("project",)
|
|
|
|
|
|
class PriorityViewSet(ModelListViewSet):
|
|
model = models.Priority
|
|
serializer_class = serializers.PrioritySerializer
|
|
permission_classes = (IsAuthenticated, permissions.PriorityPermission)
|
|
filter_backends = (filters.IsProjectMemberFilterBackend,)
|
|
filter_fields = ("project",)
|
|
|
|
|
|
class IssueTypeViewSet(ModelListViewSet):
|
|
model = models.IssueType
|
|
serializer_class = serializers.IssueTypeSerializer
|
|
permission_classes = (IsAuthenticated, permissions.IssueTypePermission)
|
|
filter_backends = (filters.IsProjectMemberFilterBackend,)
|
|
filter_fields = ("project",)
|
|
|
|
|
|
class IssueStatusViewSet(ModelListViewSet):
|
|
model = models.IssueStatus
|
|
serializer_class = serializers.IssueStatusSerializer
|
|
permission_classes = (IsAuthenticated, permissions.IssueStatusPermission)
|
|
filter_backends = (filters.IsProjectMemberFilterBackend,)
|
|
filter_fields = ("project",)
|
|
|
|
|
|
# Questions commin ViewSets
|
|
|
|
class QuestionStatusViewSet(ModelListViewSet):
|
|
model = models.QuestionStatus
|
|
serializer_class = serializers.QuestionStatusSerializer
|
|
permission_classes = (IsAuthenticated, permissions.QuestionStatusPermission)
|
|
filter_backends = (filters.IsProjectMemberFilterBackend,)
|
|
filter_fields = ("project",)
|