sajad-dev/Backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

UPDATE

Browse Source
This commit is contained in:
Mohammad Sajad Pourajam
2026-04-02 23:25:39 +03:30
parent 881f8efa4d
commit bd0d04256c
84 changed files with 2725 additions and 856 deletions
Download Patch File Download Diff File Expand all files Collapse all files
irrigation_recommendation/views.py
+66 -7
View File
@@ -10,7 +10,9 @@ 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 .mock_data import CONFIG_RESPONSE_DATA
from .models import IrrigationRecommendationRequest
from .serializers import (
IrrigationRecommendRequestSerializer,
IrrigationRecommendResponseDataSerializer,
@@ -19,59 +21,116 @@ from .serializers import (
)
class ConfigView(APIView):
class FarmAccessMixin:
@staticmethod
def _get_farm(request, farm_uuid):
if not farm_uuid:
raise serializers.ValidationError({"farm_uuid": ["This field is required."]})
try:
return FarmHub.objects.get(farm_uuid=farm_uuid, owner=request.user)
except FarmHub.DoesNotExist as exc:
raise serializers.ValidationError({"farm_uuid": ["Farm not found."]}) from exc
class ConfigView(FarmAccessMixin, APIView):
@extend_schema(
tags=["Irrigation Recommendation"],
parameters=[
OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=True),
],
responses={200: status_response("IrrigationConfigResponse", data=serializers.JSONField())},
)
def get(self, request):
return Response({"status": "success", "data": CONFIG_RESPONSE_DATA}, status=status.HTTP_200_OK)
farm = self._get_farm(request, request.query_params.get("farm_uuid"))
data = dict(CONFIG_RESPONSE_DATA)
data["farm_uuid"] = str(farm.farm_uuid)
return Response({"status": "success", "data": data}, status=status.HTTP_200_OK)
class RecommendView(APIView):
class RecommendView(FarmAccessMixin, APIView):
@extend_schema(
tags=["Irrigation Recommendation"],
request=IrrigationRecommendRequestSerializer,
responses={200: status_response("IrrigationRecommendResponse", data=IrrigationRecommendResponseDataSerializer())},
)
def post(self, request):
serializer = IrrigationRecommendRequestSerializer(data=request.data)
serializer.is_valid(raise_exception=True)
payload = serializer.validated_data.copy()
farm = self._get_farm(request, payload.get("farm_uuid"))
payload["farm_uuid"] = str(farm.farm_uuid)
adapter_response = external_api_request(
"ai",
"/irrigation/recommend",
method="POST",
payload=request.data,
payload=payload,
)
response_data = adapter_response.data if isinstance(adapter_response.data, dict) else {}
IrrigationRecommendationRequest.objects.create(
farm=farm,
crop_id=payload.get("crop_id", ""),
task_id=str(response_data.get("data", {}).get("task_id") or response_data.get("task_id") or ""),
status=str(response_data.get("data", {}).get("status") or response_data.get("status") or ""),
request_payload=payload,
response_payload=adapter_response.data if isinstance(adapter_response.data, dict) else {"raw": adapter_response.data},
)
return Response(adapter_response.data, status=adapter_response.status_code)
class RecommendTaskCreateView(APIView):
class RecommendTaskCreateView(FarmAccessMixin, APIView):
@extend_schema(
tags=["Irrigation Recommendation"],
request=IrrigationRecommendRequestSerializer,
responses={200: status_response("IrrigationRecommendTaskCreateResponse", data=IrrigationTaskSubmitDataSerializer())},
)
def post(self, request):
serializer = IrrigationRecommendRequestSerializer(data=request.data)
serializer.is_valid(raise_exception=True)
payload = serializer.validated_data.copy()
farm = self._get_farm(request, payload.get("farm_uuid"))
payload["farm_uuid"] = str(farm.farm_uuid)
adapter_response = external_api_request(
"ai",
"/irrigation/recommend",
method="POST",
payload=request.data,
payload=payload,
)
response_data = adapter_response.data if isinstance(adapter_response.data, dict) else {}
IrrigationRecommendationRequest.objects.create(
farm=farm,
crop_id=payload.get("crop_id", ""),
task_id=str(response_data.get("data", {}).get("task_id") or response_data.get("task_id") or ""),
status=str(response_data.get("data", {}).get("status") or response_data.get("status") or ""),
request_payload=payload,
response_payload=adapter_response.data if isinstance(adapter_response.data, dict) else {"raw": adapter_response.data},
)
return Response(adapter_response.data, status=adapter_response.status_code)
class RecommendTaskStatusView(APIView):
class RecommendTaskStatusView(FarmAccessMixin, APIView):
@extend_schema(
tags=["Irrigation Recommendation"],
parameters=[
OpenApiParameter(name="task_id", type=OpenApiTypes.STR, location=OpenApiParameter.PATH),
OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=True),
],
responses={200: status_response("IrrigationRecommendTaskStatusResponse", data=IrrigationTaskStatusDataSerializer())},
)
def get(self, request, task_id):
farm = self._get_farm(request, request.query_params.get("farm_uuid"))
adapter_response = external_api_request(
"ai",
f"/irrigation/recommend/status/{task_id}",
method="GET",
query={"farm_uuid": str(farm.farm_uuid)},
)
response_data = adapter_response.data if isinstance(adapter_response.data, dict) else {}
IrrigationRecommendationRequest.objects.filter(farm=farm, task_id=task_id).update(
status=str(response_data.get("data", {}).get("status") or response_data.get("status") or ""),
response_payload=adapter_response.data if isinstance(adapter_response.data, dict) else {"raw": adapter_response.data},
)
return Response(adapter_response.data, status=adapter_response.status_code)