mirror of
https://github.com/django-helpdesk/django-helpdesk.git
synced 2024-12-13 18:31:10 +01:00
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from rest_framework import viewsets
|
|
from rest_framework.permissions import IsAdminUser
|
|
from rest_framework.viewsets import GenericViewSet
|
|
from rest_framework.mixins import CreateModelMixin
|
|
from django.contrib.auth import get_user_model
|
|
|
|
from helpdesk.models import Ticket, FollowUp, FollowUpAttachment
|
|
from helpdesk.serializers import TicketSerializer, UserSerializer, FollowUpSerializer, FollowUpAttachmentSerializer
|
|
|
|
|
|
class TicketViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
A viewset that provides the standard actions to handle Ticket
|
|
"""
|
|
queryset = Ticket.objects.all()
|
|
serializer_class = TicketSerializer
|
|
permission_classes = [IsAdminUser]
|
|
|
|
def get_queryset(self):
|
|
tickets = Ticket.objects.all()
|
|
for ticket in tickets:
|
|
ticket.set_custom_field_values()
|
|
return tickets
|
|
|
|
def get_object(self):
|
|
ticket = super().get_object()
|
|
ticket.set_custom_field_values()
|
|
return ticket
|
|
|
|
|
|
class FollowUpViewSet(viewsets.ModelViewSet):
|
|
queryset = FollowUp.objects.all()
|
|
serializer_class = FollowUpSerializer
|
|
permission_classes = [IsAdminUser]
|
|
|
|
|
|
class FollowUpAttachmentViewSet(viewsets.ModelViewSet):
|
|
queryset = FollowUpAttachment.objects.all()
|
|
serializer_class = FollowUpAttachmentSerializer
|
|
permission_classes = [IsAdminUser]
|
|
|
|
|
|
class CreateUserView(CreateModelMixin, GenericViewSet):
|
|
queryset = get_user_model().objects.all()
|
|
serializer_class = UserSerializer
|
|
permission_classes = [IsAdminUser]
|