UPDATE
This commit is contained in:
@@ -8,38 +8,24 @@ class FertilizationFarmDataSerializer(serializers.Serializer):
|
||||
|
||||
|
||||
class FertilizationRecommendRequestSerializer(serializers.Serializer):
|
||||
farm_uuid = serializers.UUIDField(required=True)
|
||||
crop_id = serializers.CharField(required=False, allow_blank=True)
|
||||
growth_stage = serializers.CharField(required=False, allow_blank=True)
|
||||
farm_data = FertilizationFarmDataSerializer(required=False)
|
||||
soilType = serializers.CharField(required=False, allow_blank=True)
|
||||
organicMatter = serializers.CharField(required=False, allow_blank=True)
|
||||
waterEC = serializers.CharField(required=False, allow_blank=True)
|
||||
farm_uuid = serializers.UUIDField(required=True, help_text="UUID مزرعه برای دریافت توصیه کودی.")
|
||||
plant_name = serializers.CharField(required=False, allow_blank=True, help_text="نام محصول یا گیاه.")
|
||||
growth_stage = serializers.CharField(required=False, allow_blank=True, help_text="مرحله رشد گیاه.")
|
||||
|
||||
|
||||
class FertilizationPlanSerializer(serializers.Serializer):
|
||||
npkRatio = serializers.CharField(required=False, allow_blank=True)
|
||||
amountPerHectare = serializers.CharField(required=False, allow_blank=True)
|
||||
class FertilizationSectionSerializer(serializers.Serializer):
|
||||
type = serializers.ChoiceField(choices=["text", "list", "recommendation", "warning"])
|
||||
title = serializers.CharField(required=False, allow_blank=True)
|
||||
icon = serializers.CharField(required=False, allow_blank=True)
|
||||
content = serializers.CharField(required=False, allow_blank=True)
|
||||
items = serializers.ListField(child=serializers.CharField(), required=False)
|
||||
fertilizerType = serializers.CharField(required=False, allow_blank=True)
|
||||
amount = serializers.CharField(required=False, allow_blank=True)
|
||||
applicationMethod = serializers.CharField(required=False, allow_blank=True)
|
||||
applicationInterval = serializers.CharField(required=False, allow_blank=True)
|
||||
reasoning = serializers.CharField(required=False, allow_blank=True)
|
||||
timing = serializers.CharField(required=False, allow_blank=True)
|
||||
validityPeriod = serializers.CharField(required=False, allow_blank=True)
|
||||
expandableExplanation = serializers.CharField(required=False, allow_blank=True)
|
||||
|
||||
|
||||
class FertilizationRecommendResponseDataSerializer(serializers.Serializer):
|
||||
plan = FertilizationPlanSerializer(required=False)
|
||||
|
||||
|
||||
class FertilizationTaskSubmitDataSerializer(serializers.Serializer):
|
||||
task_id = serializers.CharField(required=False, allow_blank=True)
|
||||
status = serializers.CharField(required=False, allow_blank=True)
|
||||
|
||||
|
||||
class FertilizationTaskProgressSerializer(serializers.Serializer):
|
||||
message = serializers.CharField(required=False, allow_blank=True)
|
||||
|
||||
|
||||
class FertilizationTaskStatusDataSerializer(serializers.Serializer):
|
||||
task_id = serializers.CharField(required=False, allow_blank=True)
|
||||
status = serializers.CharField(required=False, allow_blank=True)
|
||||
progress = FertilizationTaskProgressSerializer(required=False)
|
||||
result = FertilizationRecommendResponseDataSerializer(required=False)
|
||||
sections = FertilizationSectionSerializer(many=True, read_only=True)
|
||||
|
||||
@@ -1,13 +1,8 @@
|
||||
from django.urls import path
|
||||
|
||||
from .views import ConfigView, RecommendTaskStatusView, RecommendView
|
||||
from .views import ConfigView, RecommendView
|
||||
|
||||
urlpatterns = [
|
||||
path("config/", ConfigView.as_view(), name="fertilization-recommendation-config"),
|
||||
path("recommend/", RecommendView.as_view(), name="fertilization-recommendation-recommend"),
|
||||
path(
|
||||
"recommend/status/<str:task_id>/",
|
||||
RecommendTaskStatusView.as_view(),
|
||||
name="fertilization-recommendation-task-status",
|
||||
),
|
||||
]
|
||||
|
||||
@@ -2,11 +2,12 @@
|
||||
Fertilization Recommendation API views.
|
||||
"""
|
||||
|
||||
import logging
|
||||
|
||||
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 drf_spectacular.utils import extend_schema
|
||||
|
||||
from config.swagger import status_response
|
||||
from external_api_adapter import request as external_api_request
|
||||
@@ -16,11 +17,12 @@ from .models import FertilizationRecommendationRequest
|
||||
from .serializers import (
|
||||
FertilizationRecommendRequestSerializer,
|
||||
FertilizationRecommendResponseDataSerializer,
|
||||
FertilizationTaskStatusDataSerializer,
|
||||
FertilizationTaskSubmitDataSerializer,
|
||||
)
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class FarmAccessMixin:
|
||||
@staticmethod
|
||||
def _get_farm(request, farm_uuid):
|
||||
@@ -35,9 +37,6 @@ class FarmAccessMixin:
|
||||
class ConfigView(FarmAccessMixin, APIView):
|
||||
@extend_schema(
|
||||
tags=["Fertilization Recommendation"],
|
||||
parameters=[
|
||||
OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=True, default="11111111-1111-1111-1111-111111111111"),
|
||||
],
|
||||
responses={200: status_response("FertilizationConfigResponse", data=serializers.JSONField())},
|
||||
)
|
||||
def get(self, request):
|
||||
@@ -48,6 +47,62 @@ class ConfigView(FarmAccessMixin, APIView):
|
||||
|
||||
|
||||
class RecommendView(FarmAccessMixin, APIView):
|
||||
@staticmethod
|
||||
def _normalize_sections(raw_sections):
|
||||
if not isinstance(raw_sections, list):
|
||||
return []
|
||||
|
||||
allowed_keys = {
|
||||
"type",
|
||||
"title",
|
||||
"icon",
|
||||
"content",
|
||||
"items",
|
||||
"fertilizerType",
|
||||
"amount",
|
||||
"applicationMethod",
|
||||
"timing",
|
||||
"validityPeriod",
|
||||
"expandableExplanation",
|
||||
}
|
||||
|
||||
normalized_sections = []
|
||||
for section in raw_sections:
|
||||
if not isinstance(section, dict) or not section.get("type"):
|
||||
continue
|
||||
|
||||
normalized_section = {}
|
||||
for key in allowed_keys:
|
||||
value = section.get(key)
|
||||
if value is None:
|
||||
continue
|
||||
if key == "items":
|
||||
if not isinstance(value, list):
|
||||
continue
|
||||
normalized_section[key] = [str(item) for item in value]
|
||||
continue
|
||||
normalized_section[key] = str(value) if key != "type" else value
|
||||
|
||||
normalized_sections.append(normalized_section)
|
||||
return normalized_sections
|
||||
|
||||
def _extract_public_sections(self, adapter_data):
|
||||
if not isinstance(adapter_data, dict):
|
||||
return []
|
||||
|
||||
data = adapter_data.get("data")
|
||||
if isinstance(data, dict) and isinstance(data.get("sections"), list):
|
||||
return self._normalize_sections(data.get("sections"))
|
||||
|
||||
result = data.get("result") if isinstance(data, dict) else None
|
||||
if isinstance(result, dict) and isinstance(result.get("sections"), list):
|
||||
return self._normalize_sections(result.get("sections"))
|
||||
|
||||
if isinstance(adapter_data.get("sections"), list):
|
||||
return self._normalize_sections(adapter_data.get("sections"))
|
||||
|
||||
return []
|
||||
|
||||
@extend_schema(
|
||||
tags=["Fertilization Recommendation"],
|
||||
request=FertilizationRecommendRequestSerializer,
|
||||
@@ -59,47 +114,53 @@ class RecommendView(FarmAccessMixin, APIView):
|
||||
payload = serializer.validated_data.copy()
|
||||
farm = self._get_farm(request, payload.get("farm_uuid"))
|
||||
payload["farm_uuid"] = str(farm.farm_uuid)
|
||||
payload["plant_name"] = payload.get("plant_name", "")
|
||||
payload["growth_stage"] = payload.get("growth_stage", "")
|
||||
|
||||
adapter_response = external_api_request(
|
||||
"ai",
|
||||
"/fertilization/recommend",
|
||||
"/api/fertilization/recommend/",
|
||||
method="POST",
|
||||
payload=payload,
|
||||
)
|
||||
|
||||
response_data = adapter_response.data if isinstance(adapter_response.data, dict) else {}
|
||||
public_sections = self._extract_public_sections(response_data)
|
||||
|
||||
logger.warning(
|
||||
"Fertilization recommendation response parsed: farm_uuid=%s status_code=%s response_keys=%s sections_count=%s",
|
||||
str(farm.farm_uuid),
|
||||
adapter_response.status_code,
|
||||
sorted(response_data.keys()) if isinstance(response_data, dict) else None,
|
||||
len(public_sections),
|
||||
)
|
||||
|
||||
FertilizationRecommendationRequest.objects.create(
|
||||
farm=farm,
|
||||
crop_id=payload.get("crop_id", ""),
|
||||
crop_id=payload.get("plant_name", ""),
|
||||
growth_stage=payload.get("growth_stage", ""),
|
||||
task_id=str(response_data.get("data", {}).get("task_id") or response_data.get("task_id") or ""),
|
||||
status=str(response_data.get("data", {}).get("status") or response_data.get("status") or ""),
|
||||
task_id="",
|
||||
status="success" if adapter_response.status_code < 400 else "error",
|
||||
request_payload=payload,
|
||||
response_payload=adapter_response.data if isinstance(adapter_response.data, dict) else {"raw": adapter_response.data},
|
||||
)
|
||||
return Response(adapter_response.data, status=adapter_response.status_code)
|
||||
if adapter_response.status_code >= 400:
|
||||
return Response(
|
||||
{
|
||||
"code": adapter_response.status_code,
|
||||
"msg": "error",
|
||||
"data": response_data if isinstance(response_data, dict) else {"message": str(adapter_response.data)},
|
||||
},
|
||||
status=adapter_response.status_code,
|
||||
)
|
||||
|
||||
|
||||
class RecommendTaskStatusView(FarmAccessMixin, APIView):
|
||||
@extend_schema(
|
||||
tags=["Fertilization Recommendation"],
|
||||
parameters=[
|
||||
OpenApiParameter(name="task_id", type=OpenApiTypes.STR, location=OpenApiParameter.PATH),
|
||||
OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=True, default="11111111-1111-1111-1111-111111111111"),
|
||||
],
|
||||
responses={200: status_response("FertilizationRecommendTaskStatusResponse", data=FertilizationTaskStatusDataSerializer())},
|
||||
)
|
||||
def get(self, request, task_id):
|
||||
farm = self._get_farm(request, request.query_params.get("farm_uuid"))
|
||||
adapter_response = external_api_request(
|
||||
"ai",
|
||||
f"/fertilization/status/{task_id}",
|
||||
method="GET",
|
||||
query={"farm_uuid": str(farm.farm_uuid)},
|
||||
return Response(
|
||||
{
|
||||
"code": 200,
|
||||
"msg": "success",
|
||||
"data": {
|
||||
"sections": public_sections,
|
||||
},
|
||||
},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
response_data = adapter_response.data if isinstance(adapter_response.data, dict) else {}
|
||||
FertilizationRecommendationRequest.objects.filter(farm=farm, task_id=task_id).update(
|
||||
status=str(response_data.get("data", {}).get("status") or response_data.get("status") or ""),
|
||||
response_payload=adapter_response.data if isinstance(adapter_response.data, dict) else {"raw": adapter_response.data},
|
||||
)
|
||||
return Response(adapter_response.data, status=adapter_response.status_code)
|
||||
|
||||
Reference in New Issue
Block a user