78 lines
3.3 KiB
Python
78 lines
3.3 KiB
Python
from drf_spectacular.utils import OpenApiExample, extend_schema
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from config.openapi import build_envelope_serializer, build_response
|
|
|
|
from .serializers import FarmAlertsRequestSerializer
|
|
from .services import get_farm_alerts_tracker
|
|
|
|
|
|
FarmAlertsValidationErrorSerializer = build_envelope_serializer(
|
|
"FarmAlertsValidationErrorSerializer",
|
|
data_required=False,
|
|
allow_null=True,
|
|
)
|
|
FarmAlertsTrackerResponseSerializer = build_envelope_serializer(
|
|
"FarmAlertsTrackerResponseSerializer",
|
|
data_schema=None,
|
|
)
|
|
|
|
|
|
class FarmAlertsTrackerView(APIView):
|
|
@extend_schema(
|
|
tags=["Farm Alerts"],
|
|
summary="ارزیابی tracker هشدارهای مزرعه",
|
|
description=(
|
|
"با دریافت farm_uuid، هشدارهای مزرعه را تحلیل می کند، "
|
|
"کانتکست مزرعه و لیست alertهای ورودی را به RAG می فرستد، "
|
|
"و notificationهای سطح خطر/هشدار/اطلاع رسانی را در دیتابیس ذخیره می کند."
|
|
),
|
|
request=FarmAlertsRequestSerializer,
|
|
responses={
|
|
200: build_response(FarmAlertsTrackerResponseSerializer, "خروجی tracker هشدارهای مزرعه."),
|
|
400: build_response(FarmAlertsValidationErrorSerializer, "پارامتر ورودی نامعتبر است."),
|
|
500: build_response(FarmAlertsValidationErrorSerializer, "خطا در تولید خروجی tracker هشدارها."),
|
|
},
|
|
examples=[
|
|
OpenApiExample(
|
|
"نمونه درخواست tracker",
|
|
value={
|
|
"farm_uuid": "11111111-1111-1111-1111-111111111111",
|
|
"alerts": [
|
|
{
|
|
"alert_id": "soil-moisture-001",
|
|
"level": "warning",
|
|
"title": "افت رطوبت خاک",
|
|
"message": "رطوبت خاک کمتر از حد مطلوب گزارش شده است.",
|
|
"suggested_action": "آبیاری اصلاحی بررسی شود.",
|
|
"source_metric_type": "moisture",
|
|
}
|
|
],
|
|
},
|
|
request_only=True,
|
|
),
|
|
],
|
|
)
|
|
def post(self, request):
|
|
serializer = FarmAlertsRequestSerializer(data=request.data)
|
|
if not serializer.is_valid():
|
|
return Response(
|
|
{"code": 400, "msg": "داده نامعتبر.", "data": serializer.errors},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
validated = serializer.validated_data
|
|
try:
|
|
result = get_farm_alerts_tracker(
|
|
farm_uuid=validated["farm_uuid"],
|
|
query=validated.get("query"),
|
|
alerts=validated.get("alerts"),
|
|
)
|
|
except Exception as exc:
|
|
return Response(
|
|
{"code": 500, "msg": f"خطا در تولید tracker هشدارها: {exc}", "data": None},
|
|
status=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
)
|
|
return Response({"code": 200, "msg": "success", "data": result}, status=status.HTTP_200_OK)
|