UPDATE
This commit is contained in:
@@ -0,0 +1,8 @@
|
||||
from django.apps import AppConfig
|
||||
|
||||
|
||||
class WaterConfig(AppConfig):
|
||||
default_auto_field = "django.db.models.BigAutoField"
|
||||
name = "water"
|
||||
label = "weather_forecast"
|
||||
verbose_name = "water"
|
||||
@@ -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="WeatherForecastLog",
|
||||
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)),
|
||||
("condition", models.CharField(blank=True, default="", max_length=128)),
|
||||
("temperature", models.FloatField(blank=True, null=True)),
|
||||
("unit", models.CharField(blank=True, default="°C", max_length=16)),
|
||||
("humidity", models.IntegerField(blank=True, null=True)),
|
||||
("wind_speed", models.FloatField(blank=True, null=True)),
|
||||
("wind_unit", models.CharField(blank=True, default="km/h", max_length=16)),
|
||||
("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="weather_forecasts",
|
||||
to="farm_hub.farmhub",
|
||||
),
|
||||
),
|
||||
],
|
||||
options={
|
||||
"db_table": "weather_forecast_logs",
|
||||
"ordering": ["-fetched_at"],
|
||||
},
|
||||
),
|
||||
]
|
||||
@@ -0,0 +1,34 @@
|
||||
"""
|
||||
Static mock data for WATER API.
|
||||
"""
|
||||
|
||||
FARM_WEATHER_CARD = {
|
||||
"condition": "صاف",
|
||||
"temperature": 24,
|
||||
"unit": "°C",
|
||||
"humidity": 45,
|
||||
"windSpeed": 12,
|
||||
"windUnit": "km/h",
|
||||
"chartData": {
|
||||
"labels": ["۶ صبح", "۹ صبح", "۱۲ ظهر", "۳ بعدازظهر", "۶ عصر", "۹ شب", "۱۲ شب"],
|
||||
"series": [[18, 22, 26, 28, 25, 20, 18]],
|
||||
},
|
||||
}
|
||||
|
||||
WATER_NEED_PREDICTION = {
|
||||
"totalNext7Days": 3290,
|
||||
"unit": "m3",
|
||||
"categories": ["روز 1", "روز 2", "روز 3", "روز 4", "روز 5", "روز 6", "روز 7"],
|
||||
"series": [{"name": "نیاز آبی", "data": [420, 450, 480, 460, 490, 510, 480]}],
|
||||
}
|
||||
|
||||
WATER_STRESS_INDEX = {
|
||||
"id": "water_stress_index",
|
||||
"title": "شاخص تنش آبی",
|
||||
"subtitle": "فعلی",
|
||||
"stats": "12%",
|
||||
"avatarColor": "info",
|
||||
"avatarIcon": "tabler-droplet",
|
||||
"chipText": "پایین",
|
||||
"chipColor": "success",
|
||||
}
|
||||
@@ -0,0 +1,32 @@
|
||||
import uuid as uuid_lib
|
||||
|
||||
from django.db import models
|
||||
|
||||
from farm_hub.models import FarmHub
|
||||
|
||||
|
||||
class WeatherForecastLog(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="weather_forecasts",
|
||||
null=True,
|
||||
blank=True,
|
||||
)
|
||||
condition = models.CharField(max_length=128, blank=True, default="")
|
||||
temperature = models.FloatField(null=True, blank=True)
|
||||
unit = models.CharField(max_length=16, blank=True, default="°C")
|
||||
humidity = models.IntegerField(null=True, blank=True)
|
||||
wind_speed = models.FloatField(null=True, blank=True)
|
||||
wind_unit = models.CharField(max_length=16, blank=True, default="km/h")
|
||||
chart_data = models.JSONField(default=dict, blank=True)
|
||||
fetched_at = models.DateTimeField(auto_now_add=True)
|
||||
|
||||
class Meta:
|
||||
db_table = "weather_forecast_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.condition} {self.temperature}{self.unit}"
|
||||
@@ -0,0 +1,48 @@
|
||||
from rest_framework import serializers
|
||||
|
||||
|
||||
class WeatherChartDataSerializer(serializers.Serializer):
|
||||
labels = serializers.ListField(child=serializers.CharField(), required=False)
|
||||
series = serializers.ListField(
|
||||
child=serializers.ListField(child=serializers.FloatField()),
|
||||
required=False,
|
||||
)
|
||||
|
||||
|
||||
class FarmWeatherCardSerializer(serializers.Serializer):
|
||||
condition = serializers.CharField(required=False, allow_blank=True)
|
||||
temperature = serializers.FloatField(required=False)
|
||||
unit = serializers.CharField(required=False, allow_blank=True)
|
||||
humidity = serializers.IntegerField(required=False)
|
||||
windSpeed = serializers.FloatField(required=False)
|
||||
windUnit = serializers.CharField(required=False, allow_blank=True)
|
||||
chartData = WeatherChartDataSerializer(required=False)
|
||||
|
||||
|
||||
class WaterNeedSeriesSerializer(serializers.Serializer):
|
||||
name = serializers.CharField(required=False, allow_blank=True)
|
||||
data = serializers.ListField(child=serializers.FloatField(), required=False)
|
||||
|
||||
|
||||
class WaterNeedPredictionSerializer(serializers.Serializer):
|
||||
totalNext7Days = serializers.FloatField(required=False)
|
||||
unit = serializers.CharField(required=False, allow_blank=True)
|
||||
categories = serializers.ListField(child=serializers.CharField(), required=False)
|
||||
series = WaterNeedSeriesSerializer(many=True, required=False)
|
||||
|
||||
|
||||
class WaterStressIndexSerializer(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 WaterSummarySerializer(serializers.Serializer):
|
||||
farmWeatherCard = FarmWeatherCardSerializer(required=False)
|
||||
waterNeedPrediction = WaterNeedPredictionSerializer(required=False)
|
||||
waterStressIndex = WaterStressIndexSerializer(required=False)
|
||||
@@ -0,0 +1,106 @@
|
||||
from copy import deepcopy
|
||||
|
||||
from irrigation_recommendation.models import IrrigationRecommendationRequest
|
||||
|
||||
from .mock_data import FARM_WEATHER_CARD, WATER_NEED_PREDICTION, WATER_STRESS_INDEX
|
||||
from .models import WeatherForecastLog
|
||||
|
||||
|
||||
def get_farm_weather_card_data(farm=None):
|
||||
if farm is None:
|
||||
return deepcopy(FARM_WEATHER_CARD)
|
||||
|
||||
log = WeatherForecastLog.objects.filter(farm=farm).first()
|
||||
if log is None:
|
||||
return deepcopy(FARM_WEATHER_CARD)
|
||||
|
||||
return {
|
||||
"condition": log.condition or FARM_WEATHER_CARD["condition"],
|
||||
"temperature": log.temperature if log.temperature is not None else FARM_WEATHER_CARD["temperature"],
|
||||
"unit": log.unit or FARM_WEATHER_CARD["unit"],
|
||||
"humidity": log.humidity if log.humidity is not None else FARM_WEATHER_CARD["humidity"],
|
||||
"windSpeed": log.wind_speed if log.wind_speed is not None else FARM_WEATHER_CARD["windSpeed"],
|
||||
"windUnit": log.wind_unit or FARM_WEATHER_CARD["windUnit"],
|
||||
"chartData": deepcopy(log.chart_data or FARM_WEATHER_CARD["chartData"]),
|
||||
}
|
||||
|
||||
|
||||
def _extract_irrigation_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_irrigation_result(farm):
|
||||
if farm is None:
|
||||
return {}
|
||||
|
||||
for request in IrrigationRecommendationRequest.objects.filter(farm=farm):
|
||||
result = _extract_irrigation_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_irrigation_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_water_stress_index_data(farm=None):
|
||||
data = deepcopy(WATER_STRESS_INDEX)
|
||||
result = _get_latest_irrigation_result(farm)
|
||||
moisture_level = (result.get("plan") or {}).get("moistureLevel")
|
||||
|
||||
if moisture_level is None:
|
||||
return data
|
||||
|
||||
stress_value = max(0, round(80 - float(moisture_level)))
|
||||
if stress_value <= 15:
|
||||
data["chipText"] = "پایین"
|
||||
data["chipColor"] = "success"
|
||||
data["avatarColor"] = "info"
|
||||
elif stress_value <= 30:
|
||||
data["chipText"] = "متوسط"
|
||||
data["chipColor"] = "warning"
|
||||
data["avatarColor"] = "warning"
|
||||
else:
|
||||
data["chipText"] = "بالا"
|
||||
data["chipColor"] = "error"
|
||||
data["avatarColor"] = "error"
|
||||
|
||||
data["stats"] = f"{stress_value}%"
|
||||
return data
|
||||
|
||||
|
||||
def get_water_summary_data(farm=None):
|
||||
return {
|
||||
"farmWeatherCard": get_farm_weather_card_data(farm),
|
||||
"waterNeedPrediction": get_water_need_prediction_data(farm),
|
||||
"waterStressIndex": get_water_stress_index_data(farm),
|
||||
}
|
||||
@@ -0,0 +1,10 @@
|
||||
from django.urls import path
|
||||
|
||||
from .views import FarmWeatherCardView, WaterNeedPredictionView, WaterStressIndexView, WaterSummaryView
|
||||
|
||||
urlpatterns = [
|
||||
path("card/", FarmWeatherCardView.as_view(), name="water-card"),
|
||||
path("need-prediction/", WaterNeedPredictionView.as_view(), name="water-need-prediction"),
|
||||
path("stress-index/", WaterStressIndexView.as_view(), name="water-stress-index"),
|
||||
path("summary/", WaterSummaryView.as_view(), name="water-summary"),
|
||||
]
|
||||
+180
@@ -0,0 +1,180 @@
|
||||
"""
|
||||
WATER API views.
|
||||
"""
|
||||
|
||||
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 config.swagger import status_response
|
||||
from external_api_adapter import request as external_api_request
|
||||
from farm_hub.models import FarmHub
|
||||
from .models import WeatherForecastLog
|
||||
from .serializers import FarmWeatherCardSerializer, WaterNeedPredictionSerializer, WaterStressIndexSerializer, WaterSummarySerializer
|
||||
from .services import get_water_need_prediction_data, get_water_stress_index_data, get_water_summary_data
|
||||
|
||||
|
||||
class FarmWeatherCardView(APIView):
|
||||
"""
|
||||
GET endpoint for the farm weather card dashboard data.
|
||||
|
||||
Purpose:
|
||||
Returns current weather conditions and an intraday temperature chart
|
||||
for a given farm. Data is fetched from the AI external adapter.
|
||||
If farm_uuid is provided and the farm exists, the result is persisted
|
||||
in WeatherForecastLog for historical reference.
|
||||
|
||||
Input parameters:
|
||||
- farm_uuid (query, optional): UUID of the farm.
|
||||
|
||||
Response structure:
|
||||
- status: string, always "success".
|
||||
- data: object matching the farmWeatherCard shape — condition,
|
||||
temperature, unit, humidity, windSpeed, windUnit, chartData.
|
||||
"""
|
||||
|
||||
@extend_schema(
|
||||
tags=["WATER"],
|
||||
parameters=[
|
||||
OpenApiParameter(
|
||||
name="farm_uuid",
|
||||
type=OpenApiTypes.UUID,
|
||||
location=OpenApiParameter.QUERY,
|
||||
required=False,
|
||||
description="UUID of the farm to fetch weather data for.",
|
||||
default="11111111-1111-1111-1111-111111111111"),
|
||||
],
|
||||
responses={200: status_response("FarmWeatherCardResponse", data=FarmWeatherCardSerializer())},
|
||||
)
|
||||
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",
|
||||
"/weather-forecast/card",
|
||||
method="GET",
|
||||
query=query,
|
||||
)
|
||||
|
||||
response_data = adapter_response.data if isinstance(adapter_response.data, dict) else {}
|
||||
card_data = response_data.get("result", response_data.get("data", response_data))
|
||||
|
||||
self._persist_log(farm_uuid, card_data)
|
||||
|
||||
return Response(
|
||||
{"status": "success", "data": card_data},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def _persist_log(farm_uuid, card_data):
|
||||
farm = None
|
||||
if farm_uuid:
|
||||
try:
|
||||
farm = FarmHub.objects.get(farm_uuid=farm_uuid)
|
||||
except (FarmHub.DoesNotExist, Exception):
|
||||
pass
|
||||
|
||||
WeatherForecastLog.objects.create(
|
||||
farm=farm,
|
||||
condition=card_data.get("condition", ""),
|
||||
temperature=card_data.get("temperature"),
|
||||
unit=card_data.get("unit", "°C"),
|
||||
humidity=card_data.get("humidity"),
|
||||
wind_speed=card_data.get("windSpeed"),
|
||||
wind_unit=card_data.get("windUnit", "km/h"),
|
||||
chart_data=card_data.get("chartData", {}),
|
||||
)
|
||||
|
||||
|
||||
class WaterNeedPredictionView(APIView):
|
||||
@extend_schema(
|
||||
tags=["WATER"],
|
||||
parameters=[
|
||||
OpenApiParameter(
|
||||
name="farm_uuid",
|
||||
type=OpenApiTypes.UUID,
|
||||
location=OpenApiParameter.QUERY,
|
||||
required=False,
|
||||
description="UUID of the farm to fetch water need prediction for.",
|
||||
default="11111111-1111-1111-1111-111111111111",
|
||||
),
|
||||
],
|
||||
responses={200: status_response("WaterNeedPredictionResponse", data=WaterNeedPredictionSerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
farm = None
|
||||
farm_uuid = request.query_params.get("farm_uuid")
|
||||
if farm_uuid:
|
||||
try:
|
||||
farm = FarmHub.objects.get(farm_uuid=farm_uuid)
|
||||
except (FarmHub.DoesNotExist, Exception):
|
||||
farm = None
|
||||
|
||||
return Response(
|
||||
{"status": "success", "data": get_water_need_prediction_data(farm)},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
|
||||
class WaterStressIndexView(APIView):
|
||||
@extend_schema(
|
||||
tags=["WATER"],
|
||||
parameters=[
|
||||
OpenApiParameter(
|
||||
name="farm_uuid",
|
||||
type=OpenApiTypes.UUID,
|
||||
location=OpenApiParameter.QUERY,
|
||||
required=False,
|
||||
description="UUID of the farm to fetch water stress index for.",
|
||||
default="11111111-1111-1111-1111-111111111111",
|
||||
),
|
||||
],
|
||||
responses={200: status_response("WaterStressIndexResponse", data=WaterStressIndexSerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
farm = None
|
||||
farm_uuid = request.query_params.get("farm_uuid")
|
||||
if farm_uuid:
|
||||
try:
|
||||
farm = FarmHub.objects.get(farm_uuid=farm_uuid)
|
||||
except (FarmHub.DoesNotExist, Exception):
|
||||
farm = None
|
||||
|
||||
return Response(
|
||||
{"status": "success", "data": get_water_stress_index_data(farm)},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
|
||||
class WaterSummaryView(APIView):
|
||||
@extend_schema(
|
||||
tags=["WATER"],
|
||||
parameters=[
|
||||
OpenApiParameter(
|
||||
name="farm_uuid",
|
||||
type=OpenApiTypes.UUID,
|
||||
location=OpenApiParameter.QUERY,
|
||||
required=False,
|
||||
description="UUID of the farm to fetch water summary for.",
|
||||
default="11111111-1111-1111-1111-111111111111",
|
||||
),
|
||||
],
|
||||
responses={200: status_response("WaterSummaryResponse", data=WaterSummarySerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
farm = None
|
||||
farm_uuid = request.query_params.get("farm_uuid")
|
||||
if farm_uuid:
|
||||
try:
|
||||
farm = FarmHub.objects.get(farm_uuid=farm_uuid)
|
||||
except (FarmHub.DoesNotExist, Exception):
|
||||
farm = None
|
||||
|
||||
return Response(
|
||||
{"status": "success", "data": get_water_summary_data(farm)},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
Reference in New Issue
Block a user