first push

ticket/customer_views.py

@@ -0,0 +1,127 @@
+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