UPDATE

2026-04-10 16:12:51 +03:30
parent 20fd3842b6
commit 883573004c
143 changed files with 1380 additions and 2332 deletions
@@ -0,0 +1,98 @@
+"""
+Yield & Harvest Prediction API views.
+Response format: {"status": "success", "data": <payload>}. HTTP 200 only.
+Fetches all three prediction payloads (yield card, yield chart, harvest card)
+from the AI external adapter in a single call and persists a log entry
+if a valid farm_uuid is provided.
+"""
+
+from rest_framework import status
+from rest_framework.response import Response
+from rest_framework.views import APIView
+from drf_spectacular.types import OpenApiTypes
+from drf_spectacular.utils import OpenApiParameter, extend_schema
+
+from config.swagger import status_response
+from external_api_adapter import request as external_api_request
+from farm_hub.models import FarmHub
+from .models import YieldHarvestPredictionLog
+from .serializers import YieldHarvestSummarySerializer
+
+
+class YieldHarvestSummaryView(APIView):
+    """
+    GET endpoint for combined yield prediction and harvest prediction data.
+
+    Purpose:
+        Returns three dashboard card payloads in one response:
+          - yield_prediction_card  (kpi card shape)
+          - yield_prediction_chart (monthly chart + summary)
+          - harvest_prediction_card (harvest date + window)
+        Data is fetched from the AI external adapter. If farm_uuid is provided
+        and the farm exists, the result is persisted in YieldHarvestPredictionLog.
+
+    Input parameters:
+        - farm_uuid (query, optional): UUID of the farm.
+
+    Response structure:
+        - status: string, always "success".
+        - data: object with keys yield_prediction_card,
+          yield_prediction_chart, harvest_prediction_card.
+    """
+
+    @extend_schema(
+        tags=["Yield & Harvest Prediction"],
+        parameters=[
+            OpenApiParameter(
+                name="farm_uuid",
+                type=OpenApiTypes.UUID,
+                location=OpenApiParameter.QUERY,
+                required=False,
+                description="UUID of the farm for yield and harvest prediction.",
+                default="11111111-1111-1111-1111-111111111111"),
+        ],
+        responses={200: status_response("YieldHarvestSummaryResponse", data=YieldHarvestSummarySerializer())},
+    )
+    def get(self, request):
+        farm_uuid = request.query_params.get("farm_uuid")
+        query = {"farm_uuid": str(farm_uuid)} if farm_uuid else {}
+
+        adapter_response = external_api_request(
+            "ai",
+            "/yield-harvest/summary",
+            method="GET",
+            query=query,
+        )
+
+        response_data = adapter_response.data if isinstance(adapter_response.data, dict) else {}
+        summary = response_data.get("result", response_data.get("data", response_data))
+
+        self._persist_log(farm_uuid, summary)
+
+        return Response(
+            {"status": "success", "data": summary},
+            status=status.HTTP_200_OK,
+        )
+
+    @staticmethod
+    def _persist_log(farm_uuid, summary):
+        farm = None
+        if farm_uuid:
+            try:
+                farm = FarmHub.objects.get(farm_uuid=farm_uuid)
+            except (FarmHub.DoesNotExist, Exception):
+                pass
+
+        yield_card = summary.get("yield_prediction_card", {})
+        harvest_card = summary.get("harvest_prediction_card", {})
+
+        YieldHarvestPredictionLog.objects.create(
+            farm=farm,
+            yield_stats=yield_card.get("stats", ""),
+            yield_chip_text=yield_card.get("chipText", ""),
+            harvest_date=harvest_card.get("date") or None,
+            days_until_harvest=harvest_card.get("daysUntil"),
+            optimal_window_start=harvest_card.get("optimalWindowStart") or None,
+            optimal_window_end=harvest_card.get("optimalWindowEnd") or None,
+            chart_data=summary.get("yield_prediction_chart", {}),
+        )