53239fa9ac
change view perms so users can only see their own tickets changed default signup to user, rather than admin
120 lines
3.8 KiB
Python
120 lines
3.8 KiB
Python
# -*- encoding: utf-8 -*-
|
|
|
|
import logging
|
|
|
|
from django_filters import rest_framework as rest_filters
|
|
from django.views.decorators.cache import cache_page
|
|
from django.utils.decorators import method_decorator
|
|
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework import status, permissions, filters, generics
|
|
from rest_framework.pagination import PageNumberPagination
|
|
from rest_framework.authentication import SessionAuthentication, TokenAuthentication
|
|
|
|
from apps.home.models import Ticket, TicketPriority, TicketTag
|
|
from apps.authentication.models import Department
|
|
from .serializers import (
|
|
TicketSerializer, TicketTagSerializer, TicketPrioritySerializer,
|
|
UserSerializer
|
|
)
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class TicketPaginiation(PageNumberPagination):
|
|
page_size = 25
|
|
page_size_query_param = "page_size"
|
|
max_page_size = 100
|
|
|
|
|
|
class TicketListApiView(generics.ListAPIView):
|
|
"""Returns a list of all **Tickets** on the system.
|
|
|
|
Use a querystring to filter down, order, and search the results.
|
|
|
|
Querystring options are:
|
|
- uuid
|
|
- priority
|
|
- tags
|
|
- author
|
|
- author__department
|
|
"""
|
|
|
|
authentication_classes = [SessionAuthentication, TokenAuthentication]
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
pagination_class = TicketPaginiation
|
|
serializer_class = TicketSerializer
|
|
|
|
<<<<<<< Updated upstream
|
|
queryset = Ticket.objects.all()
|
|
=======
|
|
# queryset = Ticket.objects.all()
|
|
>>>>>>> Stashed changes
|
|
|
|
filter_backends = [filters.SearchFilter, rest_filters.DjangoFilterBackend, filters.OrderingFilter]
|
|
filterset_fields = ["uuid", "priority", "tags", "author", "author__department"]
|
|
search_fields = ["author__forename", "author__surname", "title", "description"]
|
|
ordering_fields = ["create_timestamp", "edit_timestamp"]
|
|
|
|
def get_queryset(self):
|
|
<<<<<<< Updated upstream
|
|
strict_tags = self.request.query_params.get("strict-tags")
|
|
if not strict_tags:
|
|
return self.queryset
|
|
|
|
tag_uuids = self.request.query_params.getlist("tags", [])
|
|
queryset = self.queryset
|
|
=======
|
|
if self.request.user.is_superuser:
|
|
queryset=Ticket.objects.all()
|
|
else:
|
|
queryset = Ticket.objects.filter(author=self.request.user)
|
|
|
|
strict_tags = self.request.query_params.get("strict-tags")
|
|
if not strict_tags:
|
|
return queryset
|
|
|
|
tag_uuids = self.request.query_params.getlist("tags", [])
|
|
>>>>>>> Stashed changes
|
|
|
|
log.debug("tag uuids %s", tag_uuids)
|
|
|
|
for uuid in tag_uuids:
|
|
if not uuid: continue
|
|
log.debug("uuid %s", uuid)
|
|
queryset = queryset.filter(tags__uuid__in=[uuid])
|
|
|
|
return queryset
|
|
|
|
|
|
class FilterCountListApiView(generics.ListAPIView):
|
|
authentication_classes = [SessionAuthentication, TokenAuthentication]
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get(self, request):
|
|
<<<<<<< Updated upstream
|
|
self._tickets = Ticket.objects.all()
|
|
=======
|
|
if self.request.user.is_superuser:
|
|
self._tickets = Ticket.objects.all()
|
|
else:
|
|
self._tickets = Ticket.objects.filter(author=self.request.user)
|
|
|
|
>>>>>>> Stashed changes
|
|
data = {"tickets": self._tickets.count()}
|
|
|
|
self._fill_data(TicketPriority, data, "priority")
|
|
self._fill_data(TicketTag, data, "tags")
|
|
self._fill_data(Department, data, "department", model_key="author__department")
|
|
|
|
return Response(data, status.HTTP_200_OK)
|
|
|
|
|
|
def _fill_data(self, model, data: dict, key: str, *, model_key:str=None):
|
|
data[key] = {} # prevent KeyError
|
|
objects = model.objects.all()
|
|
for obj in objects:
|
|
data[key][str(obj.uuid)] = self._tickets.filter(**{model_key or key: obj}).count()
|