93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
# Copyright (C) 2014 Andrey Antukh <niwi@niwi.be>
|
|
# Copyright (C) 2014 Jesús Espino <jespinog@gmail.com>
|
|
# Copyright (C) 2014 David Barragán <bameda@dbarragan.com>
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from django.contrib.contenttypes.models import ContentType
|
|
|
|
from taiga.base import response
|
|
from taiga.base.api.utils import get_object_or_404
|
|
from taiga.base.api import ReadOnlyListViewSet
|
|
|
|
from . import serializers
|
|
from . import service
|
|
from . import permissions
|
|
|
|
|
|
class TimelineViewSet(ReadOnlyListViewSet):
|
|
serializer_class = serializers.TimelineSerializer
|
|
|
|
content_type = None
|
|
|
|
def get_content_type(self):
|
|
app_name, model = self.content_type.split(".", 1)
|
|
return get_object_or_404(ContentType, app_label=app_name, model=model)
|
|
|
|
def get_queryset(self):
|
|
ct = self.get_content_type()
|
|
model_cls = ct.model_class()
|
|
|
|
qs = model_cls.objects.all()
|
|
filtered_qs = self.filter_queryset(qs)
|
|
return filtered_qs
|
|
|
|
def response_for_queryset(self, queryset):
|
|
# Switch between paginated or standard style responses
|
|
page = self.paginate_queryset(queryset)
|
|
if page is not None:
|
|
serializer = self.get_pagination_serializer(page)
|
|
else:
|
|
serializer = self.get_serializer(queryset, many=True)
|
|
|
|
return response.Ok(serializer.data)
|
|
|
|
# Just for restframework! Because it raises
|
|
# 404 on main api root if this method not exists.
|
|
def list(self, request):
|
|
return response.NotFound()
|
|
|
|
def get_timeline(self, obj):
|
|
raise NotImplementedError
|
|
|
|
def retrieve(self, request, pk):
|
|
obj = self.get_object()
|
|
self.check_permissions(request, "retrieve", obj)
|
|
|
|
qs = self.get_timeline(obj)
|
|
return self.response_for_queryset(qs)
|
|
|
|
|
|
class ProfileTimeline(TimelineViewSet):
|
|
content_type = "users.user"
|
|
permission_classes = (permissions.UserTimelinePermission,)
|
|
|
|
def get_timeline(self, user):
|
|
return service.get_profile_timeline(user, accessing_user=self.request.user)
|
|
|
|
|
|
class UserTimeline(TimelineViewSet):
|
|
content_type = "users.user"
|
|
permission_classes = (permissions.UserTimelinePermission,)
|
|
|
|
def get_timeline(self, user):
|
|
return service.get_user_timeline(user, accessing_user=self.request.user)
|
|
|
|
|
|
class ProjectTimeline(TimelineViewSet):
|
|
content_type = "projects.project"
|
|
permission_classes = (permissions.ProjectTimelinePermission,)
|
|
|
|
def get_timeline(self, project):
|
|
return service.get_project_timeline(project, accessing_user=self.request.user)
|