UPDATE
This commit is contained in:
+120
-52
@@ -1,27 +1,22 @@
|
||||
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 drf_spectacular.utils import extend_schema
|
||||
|
||||
from config.swagger import status_response
|
||||
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 (
|
||||
SoilAnomalyDetectionSerializer,
|
||||
SoilComparisonChartSerializer,
|
||||
SoilKpiSerializer,
|
||||
SoilMoistureHeatmapSerializer,
|
||||
SoilRadarChartSerializer,
|
||||
SoilSummarySerializer,
|
||||
)
|
||||
from .services import (
|
||||
get_anomaly_detection_card_data,
|
||||
get_avg_soil_moisture_data,
|
||||
get_sensor_comparison_chart_data,
|
||||
get_sensor_radar_chart_data,
|
||||
get_soil_moisture_heatmap_data,
|
||||
get_soil_summary_data,
|
||||
)
|
||||
|
||||
|
||||
@@ -35,18 +30,28 @@ def _get_farm_from_request(request):
|
||||
return None
|
||||
|
||||
|
||||
def _extract_adapter_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
|
||||
|
||||
|
||||
class AvgSoilMoistureView(APIView):
|
||||
@extend_schema(
|
||||
tags=["Soil"],
|
||||
parameters=[
|
||||
OpenApiParameter(
|
||||
name="farm_uuid",
|
||||
type=OpenApiTypes.UUID,
|
||||
location=OpenApiParameter.QUERY,
|
||||
required=False,
|
||||
description="UUID of the farm for average soil moisture.",
|
||||
default="11111111-1111-1111-1111-111111111111",
|
||||
),
|
||||
farm_uuid_query_param(required=False, description="UUID of the farm for average soil moisture."),
|
||||
],
|
||||
responses={200: status_response("AvgSoilMoistureResponse", data=SoilKpiSerializer())},
|
||||
)
|
||||
@@ -57,47 +62,48 @@ class AvgSoilMoistureView(APIView):
|
||||
)
|
||||
|
||||
|
||||
class SensorRadarChartView(APIView):
|
||||
@extend_schema(
|
||||
tags=["Soil"],
|
||||
parameters=[
|
||||
OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=False),
|
||||
],
|
||||
responses={200: status_response("SensorRadarChartResponse", data=SoilRadarChartSerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
return Response(
|
||||
{"status": "success", "data": get_sensor_radar_chart_data(_get_farm_from_request(request))},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
|
||||
class SensorComparisonChartView(APIView):
|
||||
@extend_schema(
|
||||
tags=["Soil"],
|
||||
parameters=[
|
||||
OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=False),
|
||||
],
|
||||
responses={200: status_response("SensorComparisonChartResponse", data=SoilComparisonChartSerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
return Response(
|
||||
{"status": "success", "data": get_sensor_comparison_chart_data(_get_farm_from_request(request))},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
|
||||
class SoilAnomalyDetectionView(APIView):
|
||||
@extend_schema(
|
||||
tags=["Soil"],
|
||||
parameters=[
|
||||
OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=False),
|
||||
farm_uuid_query_param(required=True, description="UUID of the farm for soil anomaly detection."),
|
||||
],
|
||||
responses={200: status_response("SoilAnomalyDetectionResponse", data=SoilAnomalyDetectionSerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
farm_uuid = request.query_params.get("farm_uuid")
|
||||
if not farm_uuid:
|
||||
return Response(
|
||||
{"code": 400, "msg": "error", "data": {"farm_uuid": ["This field is required."]}},
|
||||
status=status.HTTP_400_BAD_REQUEST,
|
||||
)
|
||||
|
||||
farm = _get_farm_from_request(request)
|
||||
if farm is None:
|
||||
return Response(
|
||||
{"code": 404, "msg": "error", "data": {"farm_uuid": ["Farm not found."]}},
|
||||
status=status.HTTP_404_NOT_FOUND,
|
||||
)
|
||||
|
||||
adapter_response = external_api_request(
|
||||
"ai",
|
||||
"/api/soile/anomaly-detection/",
|
||||
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,
|
||||
)
|
||||
|
||||
return Response(
|
||||
{"status": "success", "data": get_anomaly_detection_card_data(_get_farm_from_request(request))},
|
||||
{"code": 200, "msg": "success", "data": _extract_adapter_result(adapter_response.data)},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
@@ -106,13 +112,44 @@ class SoilMoistureHeatmapView(APIView):
|
||||
@extend_schema(
|
||||
tags=["Soil"],
|
||||
parameters=[
|
||||
OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=False),
|
||||
farm_uuid_query_param(required=True, description="UUID of the farm for soil moisture heatmap."),
|
||||
],
|
||||
responses={200: status_response("SoilMoistureHeatmapResponse", data=SoilMoistureHeatmapSerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
farm_uuid = request.query_params.get("farm_uuid")
|
||||
if not farm_uuid:
|
||||
return Response(
|
||||
{"code": 400, "msg": "error", "data": {"farm_uuid": ["This field is required."]}},
|
||||
status=status.HTTP_400_BAD_REQUEST,
|
||||
)
|
||||
|
||||
farm = _get_farm_from_request(request)
|
||||
if farm is None:
|
||||
return Response(
|
||||
{"code": 404, "msg": "error", "data": {"farm_uuid": ["Farm not found."]}},
|
||||
status=status.HTTP_404_NOT_FOUND,
|
||||
)
|
||||
|
||||
adapter_response = external_api_request(
|
||||
"ai",
|
||||
"/api/soile/moisture-heatmap/",
|
||||
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,
|
||||
)
|
||||
|
||||
return Response(
|
||||
{"status": "success", "data": get_soil_moisture_heatmap_data(_get_farm_from_request(request))},
|
||||
{"code": 200, "msg": "success", "data": _extract_adapter_result(adapter_response.data)},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
@@ -121,12 +158,43 @@ class SoilSummaryView(APIView):
|
||||
@extend_schema(
|
||||
tags=["Soil"],
|
||||
parameters=[
|
||||
OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=False),
|
||||
farm_uuid_query_param(required=True, description="UUID of the farm for soil health summary."),
|
||||
],
|
||||
responses={200: status_response("SoilSummaryResponse", data=SoilSummarySerializer())},
|
||||
)
|
||||
def get(self, request):
|
||||
farm_uuid = request.query_params.get("farm_uuid")
|
||||
if not farm_uuid:
|
||||
return Response(
|
||||
{"code": 400, "msg": "error", "data": {"farm_uuid": ["This field is required."]}},
|
||||
status=status.HTTP_400_BAD_REQUEST,
|
||||
)
|
||||
|
||||
farm = _get_farm_from_request(request)
|
||||
if farm is None:
|
||||
return Response(
|
||||
{"code": 404, "msg": "error", "data": {"farm_uuid": ["Farm not found."]}},
|
||||
status=status.HTTP_404_NOT_FOUND,
|
||||
)
|
||||
|
||||
adapter_response = external_api_request(
|
||||
"ai",
|
||||
"/api/soile/health-summary/",
|
||||
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,
|
||||
)
|
||||
|
||||
return Response(
|
||||
{"status": "success", "data": get_soil_summary_data(_get_farm_from_request(request))},
|
||||
{"code": 200, "msg": "success", "data": _extract_adapter_result(adapter_response.data)},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user