UPDATE

irrigation_recommendation/services.py

@@ -0,0 +1,77 @@
+from copy import deepcopy
+
+from .mock_data import IRRIGATION_DASHBOARD_RECOMMENDATION, RECOMMEND_RESPONSE_DATA, WATER_NEED_PREDICTION
+from .models import IrrigationRecommendationRequest
+
+
+def _extract_result(response_payload):
+    if not isinstance(response_payload, dict):
+        return {}
+
+    data = response_payload.get("data")
+    if isinstance(data, dict) and isinstance(data.get("result"), dict):
+        return data["result"]
+
+    result = response_payload.get("result")
+    if isinstance(result, dict):
+        return result
+
+    return {}
+
+
+def _get_latest_result(farm):
+    if farm is None:
+        return {}
+
+    for request in IrrigationRecommendationRequest.objects.filter(farm=farm):
+        result = _extract_result(request.response_payload)
+        if result:
+            return result
+
+    return {}
+
+
+def get_water_need_prediction_data(farm=None):
+    default_data = deepcopy(WATER_NEED_PREDICTION)
+    result = _get_latest_result(farm)
+    water_balance = result.get("water_balance", {})
+    daily = water_balance.get("daily", [])
+
+    if not daily:
+        return default_data
+
+    categories = [item.get("forecast_date") or f"روز {index + 1}" for index, item in enumerate(daily)]
+    series_data = [float(item.get("gross_irrigation_mm") or 0) for item in daily]
+
+    return {
+        "totalNext7Days": round(sum(series_data), 2),
+        "unit": "mm",
+        "categories": categories,
+        "series": [{"name": "نیاز آبی", "data": series_data}],
+    }
+
+
+def get_irrigation_dashboard_recommendation(farm=None):
+    default_item = deepcopy(IRRIGATION_DASHBOARD_RECOMMENDATION)
+    result = _get_latest_result(farm)
+    plan = result.get("plan") or RECOMMEND_RESPONSE_DATA.get("plan", {})
+
+    best_time = plan.get("bestTimeOfDay") or "05:00 - 07:00"
+    frequency = plan.get("frequencyPerWeek")
+    duration = plan.get("durationMinutes")
+    moisture = plan.get("moistureLevel")
+    warning = plan.get("warning")
+
+    subtitle_parts = []
+    if frequency is not None and duration is not None:
+        subtitle_parts.append(f"{frequency} نوبت در هفته، {duration} دقیقه برای هر نوبت")
+    if moisture is not None:
+        subtitle_parts.append(f"رطوبت هدف {moisture}%")
+    if warning:
+        subtitle_parts.append(str(warning))
+
+    default_item["title"] = f"آبیاری: {best_time}"
+    if subtitle_parts:
+        default_item["subtitle"] = ". ".join(subtitle_parts)
+
+    return default_item