Files
hossein-por-shop/backend/ticket/views.py
T

116 lines
4.3 KiB
Python

from rest_framework import generics, permissions
from rest_framework.response import Response
from rest_framework.views import APIView
from .models import Ticket, Message
from .serializers import TicketListSerializer, MessageSerializer, TicketSerializer
from utils.pagination import StructurePagination
from drf_spectacular.utils import extend_schema, OpenApiParameter, OpenApiTypes
class TicketCreateView(generics.CreateAPIView):
queryset = Ticket.objects.all()
serializer_class = TicketSerializer
permission_classes = [permissions.IsAuthenticated]
def perform_create(self, serializer):
serializer.save(customer=self.request.user)
class TicketListView(APIView):
serializer_class = TicketListSerializer
permission_classes = [permissions.IsAuthenticated]
pagination_class = StructurePagination
@extend_schema(
parameters=[
OpenApiParameter(
name="limit",
description="لیمیتش",
required=False,
type=OpenApiTypes.INT,
),
OpenApiParameter(
name="offset",
description="افستش",
required=False,
type=OpenApiTypes.INT,
),
OpenApiParameter(
name="filter",
description=(
"filter results by one of the following fields:\n"
"`in_progress`, `closed`, `resolved`."
),
required=False,
type=OpenApiTypes.STR,
),
OpenApiParameter(
name="sort",
description=(
"Sort results by one of the following fields:\n"
" `created_at`, `-created_at`."
"\nPrefix with `-` for descending order."
),
required=False,
type=OpenApiTypes.STR,
),
],
responses={
200: TicketListSerializer(many=True),
404: OpenApiTypes.OBJECT,
},
)
def get(self, request):
tickets = Ticket.objects.filter(customer=request.user)
filter_by = request.query_params.get('filter', None)
sort = request.query_params.get('sort', None)
if filter_by:
tickets.filter(status=str(filter_by))
if sort:
if sort not in ['created_at', '-created_at']:
return Response({'detail': 'wrong sort paramter'}, status=status.HTTP_400_BAD_REQUEST)
tickets.order_by(sort)
paginator = self.pagination_class()
paginated_tickets = paginator.paginate_queryset(tickets, request)
tickets_ser = self.serializer_class(instance=paginated_tickets, many=True, context={'request': request})
return paginator.get_paginated_response(tickets_ser.data)
class TicketDetailView(generics.RetrieveAPIView):
queryset = Ticket.objects.all()
serializer_class = TicketSerializer
permission_classes = [permissions.IsAuthenticated]
def get_queryset(self):
user = self.request.user
if user.is_staff:
return Ticket.objects.all()
return Ticket.objects.filter(customer=user)
class MessageCreateView(generics.CreateAPIView):
queryset = Message.objects.all()
serializer_class = MessageSerializer
permission_classes = [permissions.IsAuthenticated]
def perform_create(self, serializer):
ticket = serializer.validated_data.get('ticket')
if self.request.user != ticket.customer and not self.request.user.is_staff:
raise permissions.PermissionDenied("You are not authorized to send a message to this ticket.")
serializer.save(sender=self.request.user)
class UpdateTicketStatusView(APIView):
permission_classes = [permissions.IsAdminUser]
def post(self, request, pk):
try:
ticket = Ticket.objects.get(pk=pk)
except Ticket.DoesNotExist:
return Response({"error": "Ticket not found"}, status=404)
new_status = request.data.get('status')
if new_status not in ['open', 'in_progress', 'resolved', 'closed']:
return Response({"error": "Invalid status"}, status=400)
ticket.status = new_status
ticket.save()
return Response({"message": "Ticket status updated successfully"})