44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from rest_framework import viewsets, mixins
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
|
|
from orchestra.api import router
|
|
|
|
from .models import Ticket, Queue
|
|
from .serializers import TicketSerializer, QueueSerializer
|
|
|
|
|
|
|
|
class TicketViewSet(viewsets.ModelViewSet):
|
|
model = Ticket
|
|
serializer_class = TicketSerializer
|
|
|
|
@action()
|
|
def mark_as_read(self, request, pk=None):
|
|
ticket = self.get_object()
|
|
ticket.mark_as_read_by(request.user)
|
|
return Response({'status': 'Ticket marked as read'})
|
|
|
|
@action()
|
|
def mark_as_unread(self, request, pk=None):
|
|
ticket = self.get_object()
|
|
ticket.mark_as_unread_by(request.user)
|
|
return Response({'status': 'Ticket marked as unread'})
|
|
|
|
def get_queryset(self):
|
|
qs = super(TicketViewSet, self).get_queryset()
|
|
qs = qs.select_related('creator', 'queue')
|
|
qs = qs.prefetch_related('messages__author')
|
|
return qs.filter(creator=self.request.user)
|
|
|
|
|
|
class QueueViewSet(mixins.ListModelMixin,
|
|
mixins.RetrieveModelMixin,
|
|
viewsets.GenericViewSet):
|
|
model = Queue
|
|
serializer_class = QueueSerializer
|
|
|
|
|
|
router.register(r'tickets', TicketViewSet)
|
|
router.register(r'ticket-queues', QueueViewSet)
|