128 lines
4.3 KiB
Python
128 lines
4.3 KiB
Python
import random
|
|
|
|
from rest_framework import viewsets
|
|
from oauth2_provider.contrib.rest_framework import (
|
|
TokenHasReadWriteScope,
|
|
)
|
|
from .models import (
|
|
Ticket,
|
|
TicketContent,
|
|
Answer,
|
|
Question,
|
|
SupportUnit,
|
|
Group,
|
|
SystemUserProfile
|
|
)
|
|
from .serializers import (
|
|
TicketSerializer,
|
|
TicketContentSerializer,
|
|
TicketAnswerSerializer,
|
|
TicketQuestionSerializer,
|
|
SupportUnitSerializer
|
|
)
|
|
from rest_framework.response import Response
|
|
from .helper import upload_listed_image
|
|
from rest_framework import status
|
|
from datetime import datetime
|
|
|
|
|
|
class CustomerTicketViewSet(viewsets.ModelViewSet):
|
|
queryset = Ticket.objects.all()
|
|
serializer_class = TicketSerializer
|
|
permission_classes = [TokenHasReadWriteScope]
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
# get user object
|
|
user = SystemUserProfile.objects.get(user=request.user)
|
|
|
|
# different style of ticket information
|
|
if 'pending' in request.GET:
|
|
ticket = self.queryset.filter(customer=user, state='pending')
|
|
serializer = self.serializer_class(ticket, many=True)
|
|
elif 'responded' in request.GET:
|
|
ticket = self.queryset.filter(customer=user, state='responded')
|
|
serializer = self.serializer_class(ticket, many=True)
|
|
elif 'closed' in request.GET:
|
|
ticket = self.queryset.filter(customer=user, state='closed')
|
|
serializer = self.serializer_class(ticket, many=True)
|
|
elif 'all' in request.GET:
|
|
ticket = self.queryset.filter(customer=user)
|
|
serializer = self.serializer_class(ticket, many=True)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
def retrieve(self, request, *args, **kwargs):
|
|
ticket = self.queryset.get(key=request.data['key'])
|
|
serializer = self.serializer_class(ticket)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
|
|
if 'ticket_key' in request.data.keys():
|
|
ticket_key = request.data['ticket_key']
|
|
request.data.pop('ticket_key')
|
|
else:
|
|
ticket_key = None
|
|
|
|
# get group unit information
|
|
group = Group.objects.get(
|
|
name__exact=request.data['support_unit']
|
|
)
|
|
request.data.pop('support_unit') # pop this block from body data
|
|
|
|
# get support unit object information
|
|
support_unit = SupportUnit.objects.get(
|
|
unit=group
|
|
)
|
|
|
|
# ticket number
|
|
ticket_id = random.randint(1000000, 9999999)
|
|
|
|
# process list of images
|
|
req = upload_listed_image(req=request, field='image')
|
|
|
|
# create ticket content
|
|
content_serializer = TicketContentSerializer(data=req.data)
|
|
if content_serializer.is_valid():
|
|
content_obj = content_serializer.create(validated_data=req.data)
|
|
|
|
# create question from customer
|
|
question = Question(
|
|
questioner=SystemUserProfile.objects.get(user=request.user),
|
|
content=content_obj
|
|
)
|
|
question.save()
|
|
|
|
if ticket_key is not None:
|
|
ticket = Ticket.objects.get(key=ticket_key)
|
|
ticket.question.add(question)
|
|
ticket.state = 'pending'
|
|
ticket.save()
|
|
else:
|
|
# create final ticket
|
|
ticket = Ticket(
|
|
customer=SystemUserProfile.objects.get(user=request.user),
|
|
support_unit=support_unit,
|
|
state='pending',
|
|
ticket_id=ticket_id
|
|
)
|
|
ticket.save()
|
|
ticket.question.add(question)
|
|
|
|
serializer = self.serializer_class(ticket)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
|
|
def update(self, request, *args, **kwargs):
|
|
ticket = self.queryset.get(key=request.data['ticket_key']) # contains ticket object
|
|
request.data.pop('ticket_key')
|
|
|
|
# send ticket object to serializer for update
|
|
serializer = self.serializer_class(data=request.data)
|
|
if serializer.is_valid():
|
|
ticket_updated = serializer.update(validated_data=request.data, instance=ticket)
|
|
serializer = self.serializer_class(ticket_updated)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
return Response(serializer.errors)
|
|
|
|
def destroy(self, request, *args, **kwargs):
|
|
pass
|