UPDATE
This commit is contained in:
@@ -0,0 +1 @@
|
||||
|
||||
@@ -0,0 +1,7 @@
|
||||
from django.apps import AppConfig
|
||||
|
||||
|
||||
class CropHealthConfig(AppConfig):
|
||||
default_auto_field = "django.db.models.BigAutoField"
|
||||
name = "crop_health"
|
||||
verbose_name = "Crop Health"
|
||||
@@ -0,0 +1,27 @@
|
||||
FARM_HEALTH_SCORE = {
|
||||
"id": "farm_health_score",
|
||||
"title": "امتیاز سلامت مزرعه",
|
||||
"subtitle": "تحلیل هوشمند",
|
||||
"stats": "87%",
|
||||
"avatarColor": "success",
|
||||
"avatarIcon": "tabler-heartbeat",
|
||||
"chipText": "خوب",
|
||||
"chipColor": "success",
|
||||
}
|
||||
|
||||
|
||||
NDVI_HEALTH_CARD = {
|
||||
"ndviIndex": 0.78,
|
||||
"mean_ndvi": 0.78,
|
||||
"ndvi_map": {
|
||||
"type": "FeatureCollection",
|
||||
"features": [],
|
||||
},
|
||||
"vegetation_health_class": "Healthy",
|
||||
"observation_date": "2026-04-10",
|
||||
"satellite_source": "sentinel-2",
|
||||
"healthData": [
|
||||
{"title": "تنش نیتروژن", "value": "پایین", "color": "success", "icon": "tabler-leaf"},
|
||||
{"title": "سلامت محصول", "value": "خوب", "color": "success", "icon": "tabler-plant"},
|
||||
],
|
||||
}
|
||||
@@ -0,0 +1,2 @@
|
||||
from django.db import models
|
||||
|
||||
@@ -0,0 +1,38 @@
|
||||
from rest_framework import serializers
|
||||
|
||||
|
||||
class CropHealthRequestSerializer(serializers.Serializer):
|
||||
farm_uuid = serializers.UUIDField(help_text="UUID مزرعه برای دریافت تحلیل سلامت گیاه.")
|
||||
|
||||
|
||||
class HealthDataItemSerializer(serializers.Serializer):
|
||||
title = serializers.CharField(required=False, allow_blank=True, help_text="عنوان آیتم سلامت.")
|
||||
value = serializers.JSONField(required=False, help_text="مقدار آیتم سلامت؛ میتواند عدد، متن یا ساختار JSON باشد.")
|
||||
color = serializers.CharField(required=False, allow_blank=True, help_text="رنگ نمایشی آیتم سلامت.")
|
||||
icon = serializers.CharField(required=False, allow_blank=True, help_text="آیکون نمایشی آیتم سلامت.")
|
||||
|
||||
|
||||
class NdviHealthCardSerializer(serializers.Serializer):
|
||||
ndviIndex = serializers.FloatField(required=False, help_text="شاخص NDVI نرمالشده برای مزرعه.")
|
||||
mean_ndvi = serializers.FloatField(required=False, help_text="میانگین NDVI محاسبهشده.")
|
||||
ndvi_map = serializers.JSONField(required=False, help_text="لایه یا متادیتای نقشه NDVI.")
|
||||
vegetation_health_class = serializers.CharField(required=False, allow_blank=True, help_text="کلاس سلامت پوشش گیاهی.")
|
||||
observation_date = serializers.DateField(required=False, help_text="تاریخ مشاهده ماهوارهای.")
|
||||
satellite_source = serializers.CharField(required=False, allow_blank=True, help_text="منبع تصویر ماهوارهای.")
|
||||
healthData = HealthDataItemSerializer(many=True, required=False)
|
||||
|
||||
|
||||
class FarmHealthScoreSerializer(serializers.Serializer):
|
||||
id = serializers.CharField(required=False, allow_blank=True)
|
||||
title = serializers.CharField(required=False, allow_blank=True)
|
||||
subtitle = serializers.CharField(required=False, allow_blank=True)
|
||||
stats = serializers.CharField(required=False, allow_blank=True)
|
||||
avatarColor = serializers.CharField(required=False, allow_blank=True)
|
||||
avatarIcon = serializers.CharField(required=False, allow_blank=True)
|
||||
chipText = serializers.CharField(required=False, allow_blank=True)
|
||||
chipColor = serializers.CharField(required=False, allow_blank=True)
|
||||
|
||||
|
||||
class CropHealthSummarySerializer(serializers.Serializer):
|
||||
ndviHealthCard = NdviHealthCardSerializer(required=False)
|
||||
farmHealthScore = FarmHealthScoreSerializer(required=False)
|
||||
@@ -0,0 +1,10 @@
|
||||
from copy import deepcopy
|
||||
|
||||
from .mock_data import FARM_HEALTH_SCORE, NDVI_HEALTH_CARD
|
||||
|
||||
|
||||
def get_crop_health_summary_data(farm=None):
|
||||
return {
|
||||
"ndviHealthCard": deepcopy(NDVI_HEALTH_CARD),
|
||||
"farmHealthScore": deepcopy(FARM_HEALTH_SCORE),
|
||||
}
|
||||
@@ -0,0 +1,110 @@
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.test import TestCase
|
||||
from django.urls import Resolver404, resolve
|
||||
from rest_framework.test import APIRequestFactory, force_authenticate
|
||||
|
||||
from external_api_adapter.adapter import AdapterResponse
|
||||
from farm_hub.models import FarmHub, FarmType
|
||||
from unittest.mock import patch
|
||||
|
||||
from .views import CropHealthSummaryView, NdviHealthView
|
||||
|
||||
|
||||
class NdviHealthViewTests(TestCase):
|
||||
def setUp(self):
|
||||
self.factory = APIRequestFactory()
|
||||
self.user = get_user_model().objects.create_user(
|
||||
username="ndvi-user",
|
||||
password="secret123",
|
||||
email="ndvi@example.com",
|
||||
phone_number="09120000020",
|
||||
)
|
||||
self.other_user = get_user_model().objects.create_user(
|
||||
username="ndvi-other-user",
|
||||
password="secret123",
|
||||
email="ndvi-other@example.com",
|
||||
phone_number="09120000021",
|
||||
)
|
||||
self.farm_type = FarmType.objects.create(name="NDVI Farm Type")
|
||||
self.farm = FarmHub.objects.create(
|
||||
owner=self.user,
|
||||
farm_type=self.farm_type,
|
||||
name="NDVI Farm",
|
||||
)
|
||||
|
||||
@patch("crop_health.views.external_api_request")
|
||||
def test_post_ndvi_health_returns_expected_payload(self, mock_external_api_request):
|
||||
mock_external_api_request.return_value = AdapterResponse(
|
||||
status_code=200,
|
||||
data={"data": {"result": {"ndviIndex": 0.78, "mean_ndvi": 0.78, "vegetation_health_class": "Healthy", "satellite_source": "sentinel-2"}}},
|
||||
)
|
||||
|
||||
request = self.factory.post(
|
||||
"/api/crop-health/ndvi-health/",
|
||||
{"farm_uuid": str(self.farm.farm_uuid)},
|
||||
format="json",
|
||||
)
|
||||
force_authenticate(request, user=self.user)
|
||||
|
||||
response = NdviHealthView.as_view()(request)
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
self.assertEqual(response.data["code"], 200)
|
||||
self.assertEqual(response.data["msg"], "success")
|
||||
self.assertEqual(response.data["data"]["ndviIndex"], 0.78)
|
||||
self.assertEqual(response.data["data"]["mean_ndvi"], 0.78)
|
||||
self.assertEqual(response.data["data"]["vegetation_health_class"], "Healthy")
|
||||
self.assertEqual(response.data["data"]["satellite_source"], "sentinel-2")
|
||||
mock_external_api_request.assert_called_once_with(
|
||||
"ai",
|
||||
"/api/soil-data/ndvi-health/",
|
||||
method="POST",
|
||||
payload={"farm_uuid": str(self.farm.farm_uuid)},
|
||||
)
|
||||
|
||||
def test_post_ndvi_health_requires_farm_uuid(self):
|
||||
request = self.factory.post("/api/crop-health/ndvi-health/", {}, format="json")
|
||||
force_authenticate(request, user=self.user)
|
||||
|
||||
response = NdviHealthView.as_view()(request)
|
||||
|
||||
self.assertEqual(response.status_code, 400)
|
||||
self.assertIn("farm_uuid", response.data)
|
||||
|
||||
def test_post_ndvi_health_returns_404_for_missing_farm(self):
|
||||
request = self.factory.post(
|
||||
"/api/crop-health/ndvi-health/",
|
||||
{"farm_uuid": "11111111-1111-1111-1111-111111111111"},
|
||||
format="json",
|
||||
)
|
||||
force_authenticate(request, user=self.user)
|
||||
|
||||
response = NdviHealthView.as_view()(request)
|
||||
|
||||
self.assertEqual(response.status_code, 404)
|
||||
self.assertEqual(response.data["msg"], "Farm not found.")
|
||||
|
||||
def test_post_ndvi_health_does_not_expose_other_users_farm(self):
|
||||
request = self.factory.post(
|
||||
"/api/crop-health/ndvi-health/",
|
||||
{"farm_uuid": str(self.farm.farm_uuid)},
|
||||
format="json",
|
||||
)
|
||||
force_authenticate(request, user=self.other_user)
|
||||
|
||||
response = NdviHealthView.as_view()(request)
|
||||
|
||||
self.assertEqual(response.status_code, 404)
|
||||
self.assertEqual(response.data["msg"], "Farm not found.")
|
||||
|
||||
def test_crop_health_routes_exist(self):
|
||||
self.assertIs(resolve("/api/crop-health/ndvi-health/").func.view_class, NdviHealthView)
|
||||
self.assertIs(resolve("/api/crop-health/summary/").func.view_class, CropHealthSummaryView)
|
||||
|
||||
def test_removed_soil_health_alias_routes_no_longer_resolve(self):
|
||||
with self.assertRaises(Resolver404):
|
||||
resolve("/api/soil/health/ndvi-health/")
|
||||
with self.assertRaises(Resolver404):
|
||||
resolve("/api/soil/health/summary/")
|
||||
with self.assertRaises(Resolver404):
|
||||
resolve("/api/soil-data/ndvi-health/")
|
||||
@@ -0,0 +1,8 @@
|
||||
from django.urls import path
|
||||
|
||||
from .views import CropHealthSummaryView, NdviHealthView
|
||||
|
||||
urlpatterns = [
|
||||
path("ndvi-health/", NdviHealthView.as_view(), name="crop-health-ndvi-health"),
|
||||
path("summary/", CropHealthSummaryView.as_view(), name="crop-health-summary"),
|
||||
]
|
||||
@@ -0,0 +1,82 @@
|
||||
from rest_framework import status
|
||||
from rest_framework.permissions import IsAuthenticated
|
||||
from rest_framework.response import Response
|
||||
from rest_framework.views import APIView
|
||||
from drf_spectacular.utils import extend_schema
|
||||
|
||||
from config.swagger import farm_uuid_query_param, status_response
|
||||
from external_api_adapter import request as external_api_request
|
||||
from farm_hub.models import FarmHub
|
||||
|
||||
from .serializers import CropHealthRequestSerializer, CropHealthSummarySerializer, NdviHealthCardSerializer
|
||||
from .services import get_crop_health_summary_data
|
||||
|
||||
|
||||
class CropHealthSummaryView(APIView):
|
||||
@extend_schema(
|
||||
tags=["Crop Health"],
|
||||
parameters=[
|
||||
farm_uuid_query_param(required=False, description="UUID of the farm for crop health data."),
|
||||
],
|
||||
responses={200: status_response("CropHealthSummaryResponse", data=CropHealthSummarySerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
return Response(
|
||||
{"status": "success", "data": get_crop_health_summary_data()},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
|
||||
class NdviHealthView(APIView):
|
||||
permission_classes = [IsAuthenticated]
|
||||
|
||||
@staticmethod
|
||||
def _extract_result(adapter_data):
|
||||
if not isinstance(adapter_data, dict):
|
||||
return {}
|
||||
|
||||
data = adapter_data.get("data")
|
||||
if isinstance(data, dict) and isinstance(data.get("result"), dict):
|
||||
return data["result"]
|
||||
if isinstance(data, dict):
|
||||
return data
|
||||
|
||||
result = adapter_data.get("result")
|
||||
if isinstance(result, dict):
|
||||
return result
|
||||
|
||||
return adapter_data
|
||||
|
||||
@extend_schema(
|
||||
tags=["Crop Health"],
|
||||
request=CropHealthRequestSerializer,
|
||||
responses={200: status_response("NdviHealthResponse", data=NdviHealthCardSerializer())},
|
||||
)
|
||||
def post(self, request):
|
||||
serializer = CropHealthRequestSerializer(data=request.data)
|
||||
serializer.is_valid(raise_exception=True)
|
||||
|
||||
try:
|
||||
farm = FarmHub.objects.get(farm_uuid=serializer.validated_data["farm_uuid"], owner=request.user)
|
||||
except FarmHub.DoesNotExist:
|
||||
return Response({"code": 404, "msg": "Farm not found."}, status=status.HTTP_404_NOT_FOUND)
|
||||
|
||||
adapter_response = external_api_request(
|
||||
"ai",
|
||||
"/api/soil-data/ndvi-health/",
|
||||
method="POST",
|
||||
payload={"farm_uuid": str(farm.farm_uuid)},
|
||||
)
|
||||
if adapter_response.status_code >= 400:
|
||||
response_data = (
|
||||
adapter_response.data
|
||||
if isinstance(adapter_response.data, dict)
|
||||
else {"message": str(adapter_response.data)}
|
||||
)
|
||||
return Response(
|
||||
{"code": adapter_response.status_code, "msg": "error", "data": response_data},
|
||||
status=adapter_response.status_code,
|
||||
)
|
||||
|
||||
data = self._extract_result(adapter_response.data)
|
||||
return Response({"code": 200, "msg": "success", "data": data}, status=status.HTTP_200_OK)
|
||||
Reference in New Issue
Block a user