import - rancher page main dashboard

apps/herd/web/api/v1/api.py

@@ -9,6 +9,7 @@ from apps.core.api import BaseViewSet
 from apps.core.mixins.search_mixin import DynamicSearchMixin
 from apps.core.mixins.soft_delete_mixin import SoftDeleteMixin
 from apps.herd.models import Herd, Rancher
+from apps.herd.services.rancher_dashboard_service import RancherDashboardService
 from apps.herd.web.api.v1.serializers import HerdSerializer, RancherSerializer
 from apps.livestock.web.api.v1.serializers import LiveStockSerializer
 from apps.product.web.api.v1.serializers import product_serializers
@@ -23,6 +24,13 @@ class HerdViewSet(BaseViewSet, SoftDeleteMixin, viewsets.ModelViewSet):
     filter_backends = [filters.SearchFilter]
     search_fields = [
         "rancher__ranching_farm",
+        "rancher__first_name",
+        "rancher__last_name",
+        "rancher__mobile",
+        "rancher__national_code",
+        "rancher__address",
+        "rancher__province__name",
+        "rancher__city__name",
         "name",
         "code",
         "province__name",
@@ -105,7 +113,7 @@ class HerdViewSet(BaseViewSet, SoftDeleteMixin, viewsets.ModelViewSet):
             return self.get_paginated_response(serializer.data)
 
 
-class RancherViewSet(BaseViewSet, SoftDeleteMixin, viewsets.ModelViewSet, DynamicSearchMixin):
+class RancherViewSet(BaseViewSet, RancherDashboardService, SoftDeleteMixin, viewsets.ModelViewSet, DynamicSearchMixin):
     queryset = Rancher.objects.all()
     serializer_class = RancherSerializer
     search_fields = [
@@ -165,3 +173,25 @@ class RancherViewSet(BaseViewSet, SoftDeleteMixin, viewsets.ModelViewSet, Dynami
         if page is not None:  # noqa
             serializer = product_serializers.IncentivePlanRancherSerializer(page, many=True)
             return self.get_paginated_response(serializer.data)
+
+    @action(
+        methods=['get'],
+        detail=False,
+        url_path='rancher_main_dashboard',
+        url_name='rancher_main_dashboard',
+        name='rancher_main_dashboard'
+    )
+    def rancher_main_dashboard(self, request, pk=None):
+        """ rancher main dashboard """
+
+        query_param = self.request.query_params  # noqa
+
+        query_string = query_param.get('search', None)
+
+        rancher_dashboard_data = self.get_main_dashboard(
+            self,
+            rancher_search_fields=self.search_fields,
+            herd_search_fields=HerdViewSet.search_fields,
+            query_string=query_string,
+        )
+        return Response(rancher_dashboard_data)