UPDATE
This commit is contained in:
@@ -0,0 +1,7 @@
|
||||
from django.apps import AppConfig
|
||||
|
||||
|
||||
class YieldHarvestConfig(AppConfig):
|
||||
default_auto_field = "django.db.models.BigAutoField"
|
||||
name = "yield_harvest"
|
||||
verbose_name = "Yield & Harvest Prediction"
|
||||
@@ -0,0 +1,43 @@
|
||||
import django.db.models.deletion
|
||||
import uuid
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
initial = True
|
||||
|
||||
dependencies = [
|
||||
("farm_hub", "0007_farmhub_subscription_plan"),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.CreateModel(
|
||||
name="YieldHarvestPredictionLog",
|
||||
fields=[
|
||||
("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
|
||||
("uuid", models.UUIDField(db_index=True, default=uuid.uuid4, editable=False, unique=True)),
|
||||
("yield_stats", models.CharField(blank=True, default="", max_length=64)),
|
||||
("yield_chip_text", models.CharField(blank=True, default="", max_length=32)),
|
||||
("harvest_date", models.DateField(blank=True, null=True)),
|
||||
("days_until_harvest", models.IntegerField(blank=True, null=True)),
|
||||
("optimal_window_start", models.DateField(blank=True, null=True)),
|
||||
("optimal_window_end", models.DateField(blank=True, null=True)),
|
||||
("chart_data", models.JSONField(blank=True, default=dict)),
|
||||
("fetched_at", models.DateTimeField(auto_now_add=True)),
|
||||
(
|
||||
"farm",
|
||||
models.ForeignKey(
|
||||
blank=True,
|
||||
null=True,
|
||||
on_delete=django.db.models.deletion.CASCADE,
|
||||
related_name="yield_harvest_predictions",
|
||||
to="farm_hub.farmhub",
|
||||
),
|
||||
),
|
||||
],
|
||||
options={
|
||||
"db_table": "yield_harvest_prediction_logs",
|
||||
"ordering": ["-fetched_at"],
|
||||
},
|
||||
),
|
||||
]
|
||||
@@ -0,0 +1,51 @@
|
||||
"""
|
||||
Static mock data for Yield & Harvest Prediction API.
|
||||
Mirrors the yieldPredictionChart and harvestPredictionCard dashboard card shapes.
|
||||
"""
|
||||
|
||||
YIELD_PREDICTION_CARD = {
|
||||
"id": "yield_prediction",
|
||||
"title": "پیشبینی عملکرد",
|
||||
"subtitle": "این فصل",
|
||||
"stats": "42 تن",
|
||||
"avatarColor": "secondary",
|
||||
"avatarIcon": "tabler-chart-bar",
|
||||
"chipText": "+8%",
|
||||
"chipColor": "success",
|
||||
}
|
||||
|
||||
YIELD_PREDICTION_CHART = {
|
||||
"categories": [
|
||||
"ژانویه", "فوریه", "مارس", "آوریل", "می", "ژوئن",
|
||||
"ژوئیه", "آگوست", "سپتامبر", "اکتبر", "نوامبر", "دسامبر",
|
||||
],
|
||||
"series": [
|
||||
{"name": "امسال", "data": [35, 38, 40, 42, 45, 48, 50, 48, 46, 44, 42, 42]},
|
||||
{"name": "سال گذشته", "data": [32, 34, 36, 38, 40, 42, 44, 42, 40, 38, 36, 38]},
|
||||
],
|
||||
"summary": [
|
||||
{
|
||||
"title": "عملکرد پیشبینیشده",
|
||||
"subtitle": "این فصل",
|
||||
"amount": "42 تن",
|
||||
"avatarColor": "primary",
|
||||
"avatarIcon": "tabler-chart-bar",
|
||||
},
|
||||
{
|
||||
"title": "تاریخ برداشت",
|
||||
"subtitle": "حدود ۱۵ اکتبر",
|
||||
"amount": "+8%",
|
||||
"avatarColor": "success",
|
||||
"avatarIcon": "tabler-calendar",
|
||||
},
|
||||
],
|
||||
}
|
||||
|
||||
HARVEST_PREDICTION_CARD = {
|
||||
"date": "2025-10-15",
|
||||
"dateFormatted": "۱۵ اکتبر ۲۰۲۵",
|
||||
"daysUntil": 58,
|
||||
"description": "بر اساس تجمع GDD فعلی و پیشبینی آب و هوا. بازه بهینه برداشت: ۱۲ تا ۱۸ اکتبر.",
|
||||
"optimalWindowStart": "2025-10-12",
|
||||
"optimalWindowEnd": "2025-10-18",
|
||||
}
|
||||
@@ -0,0 +1,32 @@
|
||||
import uuid as uuid_lib
|
||||
|
||||
from django.db import models
|
||||
|
||||
from farm_hub.models import FarmHub
|
||||
|
||||
|
||||
class YieldHarvestPredictionLog(models.Model):
|
||||
uuid = models.UUIDField(default=uuid_lib.uuid4, unique=True, editable=False, db_index=True)
|
||||
farm = models.ForeignKey(
|
||||
FarmHub,
|
||||
on_delete=models.CASCADE,
|
||||
related_name="yield_harvest_predictions",
|
||||
null=True,
|
||||
blank=True,
|
||||
)
|
||||
yield_stats = models.CharField(max_length=64, blank=True, default="")
|
||||
yield_chip_text = models.CharField(max_length=32, blank=True, default="")
|
||||
harvest_date = models.DateField(null=True, blank=True)
|
||||
days_until_harvest = models.IntegerField(null=True, blank=True)
|
||||
optimal_window_start = models.DateField(null=True, blank=True)
|
||||
optimal_window_end = models.DateField(null=True, blank=True)
|
||||
chart_data = models.JSONField(default=dict, blank=True)
|
||||
fetched_at = models.DateTimeField(auto_now_add=True)
|
||||
|
||||
class Meta:
|
||||
db_table = "yield_harvest_prediction_logs"
|
||||
ordering = ["-fetched_at"]
|
||||
|
||||
def __str__(self):
|
||||
farm_label = str(self.farm_id) if self.farm_id else "no-farm"
|
||||
return f"{farm_label} — {self.yield_stats} harvest:{self.harvest_date}"
|
||||
@@ -0,0 +1,46 @@
|
||||
from rest_framework import serializers
|
||||
|
||||
|
||||
class YieldPredictionCardSerializer(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 ChartSeriesSerializer(serializers.Serializer):
|
||||
name = serializers.CharField(required=False, allow_blank=True)
|
||||
data = serializers.ListField(child=serializers.FloatField(), required=False)
|
||||
|
||||
|
||||
class ChartSummaryItemSerializer(serializers.Serializer):
|
||||
title = serializers.CharField(required=False, allow_blank=True)
|
||||
subtitle = serializers.CharField(required=False, allow_blank=True)
|
||||
amount = serializers.CharField(required=False, allow_blank=True)
|
||||
avatarColor = serializers.CharField(required=False, allow_blank=True)
|
||||
avatarIcon = serializers.CharField(required=False, allow_blank=True)
|
||||
|
||||
|
||||
class YieldPredictionChartSerializer(serializers.Serializer):
|
||||
categories = serializers.ListField(child=serializers.CharField(), required=False)
|
||||
series = ChartSeriesSerializer(many=True, required=False)
|
||||
summary = ChartSummaryItemSerializer(many=True, required=False)
|
||||
|
||||
|
||||
class HarvestPredictionCardSerializer(serializers.Serializer):
|
||||
date = serializers.CharField(required=False, allow_blank=True)
|
||||
dateFormatted = serializers.CharField(required=False, allow_blank=True)
|
||||
daysUntil = serializers.IntegerField(required=False)
|
||||
description = serializers.CharField(required=False, allow_blank=True)
|
||||
optimalWindowStart = serializers.CharField(required=False, allow_blank=True)
|
||||
optimalWindowEnd = serializers.CharField(required=False, allow_blank=True)
|
||||
|
||||
|
||||
class YieldHarvestSummarySerializer(serializers.Serializer):
|
||||
yield_prediction_card = YieldPredictionCardSerializer(required=False)
|
||||
yield_prediction_chart = YieldPredictionChartSerializer(required=False)
|
||||
harvest_prediction_card = HarvestPredictionCardSerializer(required=False)
|
||||
@@ -0,0 +1,7 @@
|
||||
from django.urls import path
|
||||
|
||||
from .views import YieldHarvestSummaryView
|
||||
|
||||
urlpatterns = [
|
||||
path("summary/", YieldHarvestSummaryView.as_view(), name="yield-harvest-summary"),
|
||||
]
|
||||
@@ -0,0 +1,98 @@
|
||||
"""
|
||||
Yield & Harvest Prediction API views.
|
||||
Response format: {"status": "success", "data": <payload>}. HTTP 200 only.
|
||||
Fetches all three prediction payloads (yield card, yield chart, harvest card)
|
||||
from the AI external adapter in a single call and persists a log entry
|
||||
if a valid farm_uuid is provided.
|
||||
"""
|
||||
|
||||
from rest_framework import 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 farm_hub.models import FarmHub
|
||||
from .models import YieldHarvestPredictionLog
|
||||
from .serializers import YieldHarvestSummarySerializer
|
||||
|
||||
|
||||
class YieldHarvestSummaryView(APIView):
|
||||
"""
|
||||
GET endpoint for combined yield prediction and harvest prediction data.
|
||||
|
||||
Purpose:
|
||||
Returns three dashboard card payloads in one response:
|
||||
- yield_prediction_card (kpi card shape)
|
||||
- yield_prediction_chart (monthly chart + summary)
|
||||
- harvest_prediction_card (harvest date + window)
|
||||
Data is fetched from the AI external adapter. If farm_uuid is provided
|
||||
and the farm exists, the result is persisted in YieldHarvestPredictionLog.
|
||||
|
||||
Input parameters:
|
||||
- farm_uuid (query, optional): UUID of the farm.
|
||||
|
||||
Response structure:
|
||||
- status: string, always "success".
|
||||
- data: object with keys yield_prediction_card,
|
||||
yield_prediction_chart, harvest_prediction_card.
|
||||
"""
|
||||
|
||||
@extend_schema(
|
||||
tags=["Yield & Harvest Prediction"],
|
||||
parameters=[
|
||||
OpenApiParameter(
|
||||
name="farm_uuid",
|
||||
type=OpenApiTypes.UUID,
|
||||
location=OpenApiParameter.QUERY,
|
||||
required=False,
|
||||
description="UUID of the farm for yield and harvest prediction.",
|
||||
default="11111111-1111-1111-1111-111111111111"),
|
||||
],
|
||||
responses={200: status_response("YieldHarvestSummaryResponse", data=YieldHarvestSummarySerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
farm_uuid = request.query_params.get("farm_uuid")
|
||||
query = {"farm_uuid": str(farm_uuid)} if farm_uuid else {}
|
||||
|
||||
adapter_response = external_api_request(
|
||||
"ai",
|
||||
"/yield-harvest/summary",
|
||||
method="GET",
|
||||
query=query,
|
||||
)
|
||||
|
||||
response_data = adapter_response.data if isinstance(adapter_response.data, dict) else {}
|
||||
summary = response_data.get("result", response_data.get("data", response_data))
|
||||
|
||||
self._persist_log(farm_uuid, summary)
|
||||
|
||||
return Response(
|
||||
{"status": "success", "data": summary},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def _persist_log(farm_uuid, summary):
|
||||
farm = None
|
||||
if farm_uuid:
|
||||
try:
|
||||
farm = FarmHub.objects.get(farm_uuid=farm_uuid)
|
||||
except (FarmHub.DoesNotExist, Exception):
|
||||
pass
|
||||
|
||||
yield_card = summary.get("yield_prediction_card", {})
|
||||
harvest_card = summary.get("harvest_prediction_card", {})
|
||||
|
||||
YieldHarvestPredictionLog.objects.create(
|
||||
farm=farm,
|
||||
yield_stats=yield_card.get("stats", ""),
|
||||
yield_chip_text=yield_card.get("chipText", ""),
|
||||
harvest_date=harvest_card.get("date") or None,
|
||||
days_until_harvest=harvest_card.get("daysUntil"),
|
||||
optimal_window_start=harvest_card.get("optimalWindowStart") or None,
|
||||
optimal_window_end=harvest_card.get("optimalWindowEnd") or None,
|
||||
chart_data=summary.get("yield_prediction_chart", {}),
|
||||
)
|
||||
Reference in New Issue
Block a user