gold eggs backend

core/apps/eggs/views/party.py

@@ -0,0 +1,45 @@
+"""
+Party view
+"""
+
+from django.db.models import Q
+from rest_framework import viewsets
+from rest_framework.decorators import action
+from rest_framework.permissions import IsAuthenticated
+from rest_framework.response import Response
+
+from core.apps.eggs import models
+from core.apps.eggs import serializers
+from core.http.permissions import IsRole
+
+
+class PartViewSet(viewsets.ModelViewSet):
+    queryset = models.Party.objects.all()
+    http_method_names = ["get", "post", "put", "patch", "delete"]
+    permission_classes = [IsAuthenticated, IsRole(["admin", "sklad", "courier"])]
+
+    def get_serializer_class(self):
+        if self.request.method == "GET":
+            return serializers.PartySerializer
+        else:
+            return serializers.PartyCreateSerializer
+
+    @action(detail=False, methods=["get"])
+    def search(self, request):
+        search_query = request.query_params.get("query", "")
+        results = self.queryset.filter(
+            Q(user_id__username__icontains=search_query)
+            | Q(count__icontains=search_query)
+            | Q(price__icontains=search_query)
+            | Q(sold__icontains=search_query)
+            | Q(sold_price__icontains=search_query)
+            | Q(invoices__name__icontains=search_query)
+            | Q(groups__product_id__name__icontains=search_query)
+            | Q(groups__entry_price__icontains=search_query)
+            | Q(groups__unit_price__icontains=search_query)
+            | Q(groups__wholesale_price__icontains=search_query)
+            | Q(groups__quantity__icontains=search_query)
+            | Q(groups__broken_eggs__icontains=search_query)
+        ).distinct()
+        serializer = self.get_serializer(results, many=True)
+        return Response()