This commit is contained in:
2026-04-27 00:40:59 +03:30
parent 2cd96ceec6
commit 64e67c282c
56 changed files with 3912 additions and 745 deletions
+15 -29
View File
@@ -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 -6
View File
@@ -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",
),
]
+96 -35
View File
@@ -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)