78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
"""
|
|
Irrigation Recommendation API views.
|
|
"""
|
|
|
|
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 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):
|
|
@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)
|
|
|
|
|
|
class RecommendView(APIView):
|
|
@extend_schema(
|
|
tags=["Irrigation Recommendation"],
|
|
request=IrrigationRecommendRequestSerializer,
|
|
responses={200: status_response("IrrigationRecommendResponse", data=IrrigationRecommendResponseDataSerializer())},
|
|
)
|
|
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 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)
|