UPDATE

irrigation_recommendation/views.py

@@ -1,78 +1,38 @@
 """
 Irrigation Recommendation API views.
-No database. All responses are static mock data.
-Response format: {"status": "success", "data": <payload>}. HTTP 200 only.
-No processing, validation, or use of input parameters in responses.
 """
 
 from rest_framework import serializers, status
 from rest_framework.response import Response
 from rest_framework.views import APIView
 from drf_spectacular.types import OpenApiTypes
-from drf_spectacular.utils import extend_schema
+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 .mock_data import CONFIG_RESPONSE_DATA
+from .serializers import (
+    IrrigationRecommendRequestSerializer,
+    IrrigationRecommendResponseDataSerializer,
+    IrrigationTaskStatusDataSerializer,
+    IrrigationTaskSubmitDataSerializer,
+)
 
 
 class ConfigView(APIView):
-    """
-    GET endpoint for irrigation config (farm info and crop options).
-
-    Purpose:
-        Returns static farm info (soilType, waterQuality, climateZone) and
-        crop options list for the irrigation recommendation form. Used when
-        loading the irrigation recommendation page.
-
-    Input parameters:
-        None. Query parameters, if sent, are not read or used.
-
-    Response structure:
-        - status: string, always "success".
-        - data: object with keys farmInfo (object), cropOptions (array of
-          { id, labelKey, icon }).
-
-    No processing or validation is performed on inputs.
-    """
-
     @extend_schema(
         tags=["Irrigation Recommendation"],
         responses={200: status_response("IrrigationConfigResponse", data=serializers.JSONField())},
     )
     def get(self, request):
-        return Response(
-            {"status": "success", "data": CONFIG_RESPONSE_DATA},
-            status=status.HTTP_200_OK,
-        )
+        return Response({"status": "success", "data": CONFIG_RESPONSE_DATA}, status=status.HTTP_200_OK)
 
 
 class RecommendView(APIView):
-    """
-    POST endpoint for irrigation recommendation.
-
-    Purpose:
-        Returns a static irrigation plan (frequencyPerWeek, durationMinutes,
-        bestTimeOfDay, moistureLevel, warning). Body may contain crop_id
-        and farm info; not read or used in response.
-
-    Input parameters:
-        - body (optional): JSON. May contain "crop_id", "soilType", "waterQuality",
-          "climateZone". Data type: object. Location: body. Not read or validated;
-          not used in response.
-
-    Response structure:
-        - status: string, always "success".
-        - data: object with key "plan" (object with frequencyPerWeek,
-          durationMinutes, bestTimeOfDay, moistureLevel, warning).
-
-    No processing or validation is performed on inputs.
-    """
-
     @extend_schema(
         tags=["Irrigation Recommendation"],
-        request=OpenApiTypes.OBJECT,
-        responses={200: status_response("IrrigationRecommendResponse", data=serializers.JSONField())},
+        request=IrrigationRecommendRequestSerializer,
+        responses={200: status_response("IrrigationRecommendResponse", data=IrrigationRecommendResponseDataSerializer())},
     )
     def post(self, request):
         adapter_response = external_api_request(
@@ -82,3 +42,36 @@ class RecommendView(APIView):
             payload=request.data,
         )
         return Response(adapter_response.data, status=adapter_response.status_code)
+
+
+class RecommendTaskCreateView(APIView):
+    @extend_schema(
+        tags=["Irrigation Recommendation"],
+        request=IrrigationRecommendRequestSerializer,
+        responses={200: status_response("IrrigationRecommendTaskCreateResponse", data=IrrigationTaskSubmitDataSerializer())},
+    )
+    def post(self, request):
+        adapter_response = external_api_request(
+            "ai",
+            "/irrigation/recommend",
+            method="POST",
+            payload=request.data,
+        )
+        return Response(adapter_response.data, status=adapter_response.status_code)
+
+
+class RecommendTaskStatusView(APIView):
+    @extend_schema(
+        tags=["Irrigation Recommendation"],
+        parameters=[
+            OpenApiParameter(name="task_id", type=OpenApiTypes.STR, location=OpenApiParameter.PATH),
+        ],
+        responses={200: status_response("IrrigationRecommendTaskStatusResponse", data=IrrigationTaskStatusDataSerializer())},
+    )
+    def get(self, request, task_id):
+        adapter_response = external_api_request(
+            "ai",
+            f"/irrigation/recommend/status/{task_id}",
+            method="GET",
+        )
+        return Response(adapter_response.data, status=adapter_response.status_code)