sajad-dev/Backend
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

UPDATE

Browse Source
This commit is contained in:
Mohammad Sajad Pourajam
2026-04-28 19:01:00 +03:30
parent 9b7d412445
commit a75c4ca9c8
21 changed files with 1898 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fertilization_recommendation/migrations/0002_recommendation_status_lifecycle.py
+37
View File
@@ -0,0 +1,37 @@
from django.db import migrations, models
PENDING_STATUS = "pending_confirmation"
OLD_STATUSES = {"", "success", "error", None}
def migrate_existing_statuses(apps, schema_editor):
Recommendation = apps.get_model("fertilization_recommendation", "FertilizationRecommendationRequest")
Recommendation.objects.filter(status__in=[status for status in OLD_STATUSES if status is not None]).update(
status=PENDING_STATUS
)
Recommendation.objects.filter(status__isnull=True).update(status=PENDING_STATUS)
class Migration(migrations.Migration):
dependencies = [
("fertilization_recommendation", "0001_initial"),
]
operations = [
migrations.AlterField(
model_name="fertilizationrecommendationrequest",
name="status",
field=models.CharField(
choices=[
("in_progress", "در حال مصرف"),
("pending_confirmation", "منتظر تایید"),
("completed", "پایان یافته"),
],
db_index=True,
default="pending_confirmation",
max_length=64,
),
),
migrations.RunPython(migrate_existing_statuses, migrations.RunPython.noop),
]
fertilization_recommendation/models.py
+15 -1
View File
@@ -6,6 +6,15 @@ from farm_hub.models import FarmHub
class FertilizationRecommendationRequest(models.Model):
STATUS_IN_PROGRESS = "in_progress"
STATUS_PENDING_CONFIRMATION = "pending_confirmation"
STATUS_COMPLETED = "completed"
STATUS_CHOICES = (
(STATUS_IN_PROGRESS, "در حال مصرف"),
(STATUS_PENDING_CONFIRMATION, "منتظر تایید"),
(STATUS_COMPLETED, "پایان یافته"),
)
uuid = models.UUIDField(default=uuid.uuid4, unique=True, editable=False, db_index=True)
farm = models.ForeignKey(
FarmHub,
@@ -15,7 +24,12 @@ class FertilizationRecommendationRequest(models.Model):
crop_id = models.CharField(max_length=255, blank=True, default="")
growth_stage = models.CharField(max_length=255, blank=True, default="")
task_id = models.CharField(max_length=255, blank=True, default="", db_index=True)
status = models.CharField(max_length=64, blank=True, default="")
status = models.CharField(
max_length=64,
choices=STATUS_CHOICES,
default=STATUS_PENDING_CONFIRMATION,
db_index=True,
)
request_payload = models.JSONField(default=dict, blank=True)
response_payload = models.JSONField(default=dict, blank=True)
created_at = models.DateTimeField(auto_now_add=True)
fertilization_recommendation/serializers.py
+25 -1
View File
@@ -9,10 +9,17 @@ class FertilizationFarmDataSerializer(serializers.Serializer):
class FertilizationRecommendRequestSerializer(serializers.Serializer):
farm_uuid = serializers.UUIDField(required=True, help_text="UUID مزرعه برای دریافت توصیه کودی.")
plant_name = serializers.CharField(required=False, allow_blank=True, help_text="نام محصول یا گیاه.")
crop_id = serializers.CharField(required=False, allow_blank=True, help_text="شناسه یا نام محصول. این فیلد همان plant_name است.")
plant_name = serializers.CharField(required=False, allow_blank=True, help_text="نام محصول یا گیاه. این فیلد همان crop_id است.")
growth_stage = serializers.CharField(required=False, allow_blank=True, help_text="مرحله رشد گیاه.")
class FertilizationRecommendationListQuerySerializer(serializers.Serializer):
farm_uuid = serializers.UUIDField(required=True, help_text="UUID مزرعه برای دریافت لیست توصیه های کودی.")
page = serializers.IntegerField(required=False, min_value=1)
page_size = serializers.IntegerField(required=False, min_value=1, max_value=100)
class FertilizationSectionSerializer(serializers.Serializer):
type = serializers.ChoiceField(choices=["text", "list", "recommendation", "warning"])
title = serializers.CharField(required=False, allow_blank=True)
@@ -98,7 +105,24 @@ class AlternativeRecommendationSerializer(serializers.Serializer):
description = serializers.CharField(required=False, allow_blank=True)
class FertilizationRecommendationListItemSerializer(serializers.Serializer):
recommendation_uuid = serializers.UUIDField(source="uuid", read_only=True)
crop_id = serializers.CharField(read_only=True)
plant_name = serializers.CharField(source="crop_id", read_only=True)
growth_stage = serializers.CharField(read_only=True)
fertilizer_type = serializers.CharField(read_only=True, allow_blank=True)
status = serializers.CharField(read_only=True)
status_label = serializers.CharField(source="get_status_display", read_only=True)
requested_at = serializers.DateTimeField(source="created_at", read_only=True)
class FertilizationRecommendResponseDataSerializer(serializers.Serializer):
recommendation_uuid = serializers.UUIDField(read_only=True, required=False)
crop_id = serializers.CharField(read_only=True, required=False, allow_blank=True)
plant_name = serializers.CharField(read_only=True, required=False, allow_blank=True)
growth_stage = serializers.CharField(read_only=True, required=False, allow_blank=True)
status = serializers.CharField(read_only=True, required=False)
status_label = serializers.CharField(read_only=True, required=False)
primary_recommendation = PrimaryRecommendationSerializer(read_only=True)
nutrient_analysis = NutrientAnalysisSerializer(read_only=True)
application_guide = ApplicationGuideSerializer(read_only=True)
fertilization_recommendation/tests.py
+169 -2
View File
@@ -5,7 +5,8 @@ from unittest.mock import patch
from external_api_adapter.adapter import AdapterResponse
from farm_hub.models import FarmHub, FarmType
from .views import RecommendView
from .models import FertilizationRecommendationRequest
from .views import RecommendationDetailView, RecommendationListView, RecommendView
class FertilizationRecommendViewTests(TestCase):
@@ -78,7 +79,7 @@ class FertilizationRecommendViewTests(TestCase):
request = self.factory.post(
"/api/fertilization/recommend/",
{"farm_uuid": str(self.farm.farm_uuid), "plant_name": "گندم", "growth_stage": "vegetative"},
{"farm_uuid": str(self.farm.farm_uuid), "crop_id": "گندم", "growth_stage": "vegetative"},
format="json",
)
force_authenticate(request, user=self.user)
@@ -95,13 +96,179 @@ class FertilizationRecommendViewTests(TestCase):
self.assertEqual(response.data["data"]["primary_recommendation"]["application_interval"]["value"], 14.0)
self.assertEqual(response.data["data"]["alternative_recommendations"][0]["usage_method"], "fertigation")
self.assertEqual(response.data["data"]["sections"][0]["type"], "recommendation")
self.assertEqual(FertilizationRecommendationRequest.objects.count(), 1)
saved_request = FertilizationRecommendationRequest.objects.get()
self.assertEqual(saved_request.crop_id, "گندم")
self.assertEqual(saved_request.growth_stage, "vegetative")
self.assertEqual(
saved_request.status,
FertilizationRecommendationRequest.STATUS_PENDING_CONFIRMATION,
)
self.assertEqual(
saved_request.response_payload["data"]["primary_recommendation"]["fertilizer_code"],
"npk-202020",
)
mock_external_api_request.assert_called_once_with(
"ai",
"/api/fertilization/recommend/",
method="POST",
payload={
"farm_uuid": str(self.farm.farm_uuid),
"crop_id": "گندم",
"plant_name": "گندم",
"growth_stage": "vegetative",
},
)
@patch("fertilization_recommendation.views.external_api_request")
def test_recommend_accepts_plant_name_and_passes_it_directly_to_ai(self, mock_external_api_request):
mock_external_api_request.return_value = AdapterResponse(status_code=200, data={"data": {}})
request = self.factory.post(
"/api/fertilization/recommend/",
{"farm_uuid": str(self.farm.farm_uuid), "plant_name": "جو", "growth_stage": "flowering"},
format="json",
)
force_authenticate(request, user=self.user)
response = RecommendView.as_view()(request)
self.assertEqual(response.status_code, 200)
saved_request = FertilizationRecommendationRequest.objects.latest("created_at")
self.assertEqual(saved_request.crop_id, "جو")
mock_external_api_request.assert_called_once_with(
"ai",
"/api/fertilization/recommend/",
method="POST",
payload={
"farm_uuid": str(self.farm.farm_uuid),
"crop_id": "جو",
"plant_name": "جو",
"growth_stage": "flowering",
},
)
def test_recommendation_list_returns_paginated_summary_items(self):
first = FertilizationRecommendationRequest.objects.create(
farm=self.farm,
crop_id="گندم",
growth_stage="vegetative",
status=FertilizationRecommendationRequest.STATUS_PENDING_CONFIRMATION,
response_payload={
"data": {
"primary_recommendation": {
"fertilizer_type": "NPK",
}
}
},
)
second = FertilizationRecommendationRequest.objects.create(
farm=self.farm,
crop_id="ذرت",
growth_stage="flowering",
status=FertilizationRecommendationRequest.STATUS_PENDING_CONFIRMATION,
response_payload={
"data": {
"primary_recommendation": {
"fertilizer_type": "Micronutrient",
}
}
},
)
request = self.factory.get(
f"/api/fertilization/recommendations/?farm_uuid={self.farm.farm_uuid}&page=1&page_size=1"
)
force_authenticate(request, user=self.user)
response = RecommendationListView.as_view()(request)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.data["code"], 200)
self.assertEqual(len(response.data["data"]), 1)
self.assertEqual(response.data["pagination"]["page"], 1)
self.assertEqual(response.data["pagination"]["page_size"], 1)
self.assertEqual(response.data["pagination"]["total_pages"], 2)
self.assertEqual(response.data["pagination"]["total_items"], 2)
self.assertTrue(response.data["pagination"]["has_next"])
self.assertFalse(response.data["pagination"]["has_previous"])
self.assertEqual(response.data["data"][0]["recommendation_uuid"], str(second.uuid))
self.assertEqual(response.data["data"][0]["plant_name"], "ذرت")
self.assertEqual(response.data["data"][0]["growth_stage"], "flowering")
self.assertEqual(response.data["data"][0]["fertilizer_type"], "Micronutrient")
self.assertEqual(response.data["data"][0]["status"], "pending_confirmation")
self.assertEqual(response.data["data"][0]["status_label"], "منتظر تایید")
self.assertIn("requested_at", response.data["data"][0])
self.assertNotEqual(response.data["data"][0]["recommendation_uuid"], str(first.uuid))
def test_recommendation_detail_returns_same_shape_as_recommend_endpoint(self):
recommendation = FertilizationRecommendationRequest.objects.create(
farm=self.farm,
crop_id="گندم",
growth_stage="vegetative",
status=FertilizationRecommendationRequest.STATUS_PENDING_CONFIRMATION,
response_payload={
"data": {
"primary_recommendation": {
"fertilizer_code": "npk-202020",
"fertilizer_type": "NPK",
"summary": "خلاصه توصیه",
},
"nutrient_analysis": {
"macro": [{"key": "n", "name": "Nitrogen", "value": 20, "unit": "percent"}],
"micro": [],
},
"application_guide": {
"safety_warning": "در هوای خنک استفاده شود",
"steps": [{"step_number": 1, "title": "آماده سازی", "description": "در آب حل شود"}],
},
"alternative_recommendations": [
{"fertilizer_code": "alt-1", "fertilizer_name": "Alt", "fertilizer_type": "NPK"}
],
"sections": [{"type": "warning", "title": "هشدار", "content": "اختلاط نشود"}],
}
},
)
request = self.factory.get(f"/api/fertilization/recommendations/{recommendation.uuid}/")
force_authenticate(request, user=self.user)
response = RecommendationDetailView.as_view()(request, recommendation_uuid=recommendation.uuid)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.data["code"], 200)
self.assertEqual(response.data["data"]["primary_recommendation"]["fertilizer_code"], "npk-202020")
self.assertEqual(response.data["data"]["primary_recommendation"]["fertilizer_type"], "NPK")
self.assertEqual(response.data["data"]["nutrient_analysis"]["macro"][0]["value"], 20.0)
self.assertEqual(response.data["data"]["application_guide"]["steps"][0]["step_number"], 1)
self.assertEqual(response.data["data"]["sections"][0]["type"], "warning")
self.assertEqual(response.data["data"]["recommendation_uuid"], str(recommendation.uuid))
self.assertEqual(response.data["data"]["crop_id"], "گندم")
self.assertEqual(response.data["data"]["plant_name"], "گندم")
self.assertEqual(response.data["data"]["status"], "pending_confirmation")
self.assertEqual(response.data["data"]["status_label"], "منتظر تایید")
def test_recommendation_detail_falls_back_to_top_level_fertilizer_code(self):
recommendation = FertilizationRecommendationRequest.objects.create(
farm=self.farm,
crop_id="گندم",
growth_stage="vegetative",
status=FertilizationRecommendationRequest.STATUS_PENDING_CONFIRMATION,
response_payload={
"data": {
"fertilizer_code": "legacy-code-101",
"fertilizer_type": "NPK",
}
},
)
request = self.factory.get(f"/api/fertilization/recommendations/{recommendation.uuid}/")
force_authenticate(request, user=self.user)
response = RecommendationDetailView.as_view()(request, recommendation_uuid=recommendation.uuid)
self.assertEqual(response.status_code, 200)
self.assertEqual(
response.data["data"]["primary_recommendation"]["fertilizer_code"],
"legacy-code-101",
)
fertilization_recommendation/urls.py
+3 -1
View File
@@ -1,8 +1,10 @@
from django.urls import path
from .views import ConfigView, RecommendView
from .views import ConfigView, RecommendationDetailView, RecommendationListView, RecommendView
urlpatterns = [
path("config/", ConfigView.as_view(), name="fertilization-recommendation-config"),
path("recommendations/<uuid:recommendation_uuid>/", RecommendationDetailView.as_view(), name="fertilization-recommendation-detail"),
path("recommendations/", RecommendationListView.as_view(), name="fertilization-recommendation-list"),
path("recommend/", RecommendView.as_view(), name="fertilization-recommendation-recommend"),
]
fertilization_recommendation/views.py
+144 -13
View File
@@ -4,7 +4,10 @@ Fertilization Recommendation API views.
import logging
from drf_spectacular.types import OpenApiTypes
from drf_spectacular.utils import OpenApiParameter
from rest_framework import serializers, status
from rest_framework.pagination import PageNumberPagination
from rest_framework.response import Response
from rest_framework.views import APIView
from drf_spectacular.utils import extend_schema
@@ -15,6 +18,8 @@ from farm_hub.models import FarmHub
from .mock_data import CONFIG_RESPONSE_DATA
from .models import FertilizationRecommendationRequest
from .serializers import (
FertilizationRecommendationListItemSerializer,
FertilizationRecommendationListQuerySerializer,
FertilizationRecommendRequestSerializer,
FertilizationRecommendResponseDataSerializer,
)
@@ -23,6 +28,33 @@ from .serializers import (
logger = logging.getLogger(__name__)
class FertilizationRecommendationPagination(PageNumberPagination):
page_size = 10
page_size_query_param = "page_size"
max_page_size = 100
def get_paginated_response(self, data):
page_size = self.get_page_size(self.request) or self.page.paginator.per_page
return Response(
{
"code": 200,
"msg": "success",
"data": data,
"pagination": {
"page": self.page.number,
"page_size": page_size,
"total_pages": self.page.paginator.num_pages,
"total_items": self.page.paginator.count,
"has_next": self.page.has_next(),
"has_previous": self.page.has_previous(),
"next": self.get_next_link(),
"previous": self.get_previous_link(),
},
},
status=status.HTTP_200_OK,
)
class FarmAccessMixin:
@staticmethod
def _get_farm(request, farm_uuid):
@@ -161,21 +193,50 @@ class RecommendView(FarmAccessMixin, APIView):
return normalized
@staticmethod
def _first_non_empty(*values):
for value in values:
if value is None:
continue
text = str(value).strip()
if text:
return text
return ""
def _normalize_primary_recommendation(self, payload):
raw_data = payload.get("primary_recommendation")
if not isinstance(raw_data, dict):
return {}
raw_data = {}
normalized = {}
for key in (
"fertilizer_code",
"fertilizer_name",
"display_title",
"fertilizer_type",
"reasoning",
"summary",
):
value = self._to_string(raw_data.get(key)).strip()
scalar_fields = {
"fertilizer_code": (
raw_data.get("fertilizer_code"),
payload.get("fertilizer_code"),
),
"fertilizer_name": (
raw_data.get("fertilizer_name"),
payload.get("fertilizer_name"),
),
"display_title": (
raw_data.get("display_title"),
payload.get("display_title"),
),
"fertilizer_type": (
raw_data.get("fertilizer_type"),
payload.get("fertilizer_type"),
),
"reasoning": (
raw_data.get("reasoning"),
payload.get("reasoning"),
),
"summary": (
raw_data.get("summary"),
payload.get("summary"),
),
}
for key, values in scalar_fields.items():
value = self._first_non_empty(*values)
if value:
normalized[key] = value
@@ -305,8 +366,11 @@ class RecommendView(FarmAccessMixin, APIView):
serializer.is_valid(raise_exception=True)
payload = serializer.validated_data.copy()
farm = self._get_farm(request, payload.get("farm_uuid"))
crop_id = self._first_non_empty(payload.get("crop_id"), payload.get("plant_name"))
plant_name = self._first_non_empty(payload.get("plant_name"), payload.get("crop_id"))
payload["farm_uuid"] = str(farm.farm_uuid)
payload["plant_name"] = payload.get("plant_name", "")
payload["crop_id"] = crop_id
payload["plant_name"] = plant_name
payload["growth_stage"] = payload.get("growth_stage", "")
adapter_response = external_api_request(
@@ -329,10 +393,10 @@ class RecommendView(FarmAccessMixin, APIView):
FertilizationRecommendationRequest.objects.create(
farm=farm,
crop_id=payload.get("plant_name", ""),
crop_id=crop_id,
growth_stage=payload.get("growth_stage", ""),
task_id="",
status="success" if adapter_response.status_code < 400 else "error",
status=FertilizationRecommendationRequest.STATUS_PENDING_CONFIRMATION,
request_payload=payload,
response_payload=adapter_response.data if isinstance(adapter_response.data, dict) else {"raw": adapter_response.data},
)
@@ -354,3 +418,70 @@ class RecommendView(FarmAccessMixin, APIView):
},
status=status.HTTP_200_OK,
)
class RecommendationListView(FarmAccessMixin, APIView):
permission_classes = RecommendView.permission_classes
pagination_class = FertilizationRecommendationPagination
@extend_schema(
tags=["Fertilization Recommendation"],
parameters=[FertilizationRecommendationListQuerySerializer],
responses={200: code_response("FertilizationRecommendationListResponse")},
)
def get(self, request):
serializer = FertilizationRecommendationListQuerySerializer(data=request.query_params)
serializer.is_valid(raise_exception=True)
farm = self._get_farm(request, serializer.validated_data["farm_uuid"])
recommendations = farm.fertilization_recommendations.all().order_by("-created_at", "-id")
paginator = self.pagination_class()
page = paginator.paginate_queryset(recommendations, request, view=self)
items = []
view_helper = RecommendView()
for recommendation in page:
normalized_payload = view_helper._normalize_response_payload(recommendation.response_payload)
recommendation.fertilizer_type = (
normalized_payload.get("primary_recommendation", {}).get("fertilizer_type", "")
)
items.append(recommendation)
data = FertilizationRecommendationListItemSerializer(items, many=True).data
return paginator.get_paginated_response(data)
class RecommendationDetailView(FarmAccessMixin, APIView):
@extend_schema(
tags=["Fertilization Recommendation"],
parameters=[
OpenApiParameter(
name="recommendation_uuid",
type=OpenApiTypes.UUID,
location=OpenApiParameter.PATH,
required=True,
)
],
responses={
200: code_response("FertilizationRecommendationDetailResponse", data=FertilizationRecommendResponseDataSerializer()),
404: code_response("FertilizationRecommendationDetailNotFoundResponse"),
},
)
def get(self, request, recommendation_uuid):
recommendation = FertilizationRecommendationRequest.objects.filter(
uuid=recommendation_uuid,
farm__owner=request.user,
).select_related("farm").first()
if recommendation is None:
return Response({"code": 404, "msg": "Recommendation not found."}, status=status.HTTP_404_NOT_FOUND)
view_helper = RecommendView()
data = view_helper._normalize_response_payload(recommendation.response_payload)
data["recommendation_uuid"] = str(recommendation.uuid)
data["crop_id"] = recommendation.crop_id
data["plant_name"] = recommendation.crop_id
data["growth_stage"] = recommendation.growth_stage
data["status"] = recommendation.status
data["status_label"] = recommendation.get_status_display()
return Response({"code": 200, "msg": "success", "data": data}, status=status.HTTP_200_OK)