UPDATE
This commit is contained in:
@@ -1,8 +1,7 @@
|
||||
from django.urls import path
|
||||
|
||||
from .views import FertilizationRecommendView, FertilizationRecommendStatusView
|
||||
from .views import FertilizationRecommendView
|
||||
|
||||
urlpatterns = [
|
||||
path("recommend/", FertilizationRecommendView.as_view(), name="fertilization-recommend"),
|
||||
path("recommend/<str:task_id>/status/", FertilizationRecommendStatusView.as_view(), name="fertilization-recommend-status"),
|
||||
]
|
||||
|
||||
+28
-63
@@ -10,31 +10,25 @@ from rest_framework.views import APIView
|
||||
from config.openapi import (
|
||||
build_envelope_serializer,
|
||||
build_response,
|
||||
build_task_queue_data_serializer,
|
||||
build_task_status_data_serializer,
|
||||
)
|
||||
|
||||
from .serializers import FertilizationRecommendRequestSerializer
|
||||
|
||||
|
||||
FertilizationQueueResponseSerializer = build_envelope_serializer(
|
||||
"FertilizationQueueResponseSerializer",
|
||||
build_task_queue_data_serializer("FertilizationQueueDataSerializer"),
|
||||
)
|
||||
FertilizationValidationErrorSerializer = build_envelope_serializer(
|
||||
"FertilizationValidationErrorSerializer",
|
||||
data_required=False,
|
||||
allow_null=True,
|
||||
)
|
||||
FertilizationStatusResponseSerializer = build_envelope_serializer(
|
||||
"FertilizationStatusResponseSerializer",
|
||||
build_task_status_data_serializer("FertilizationStatusDataSerializer"),
|
||||
FertilizationResponseSerializer = build_envelope_serializer(
|
||||
"FertilizationResponseSerializer",
|
||||
data_schema=None,
|
||||
)
|
||||
|
||||
|
||||
class FertilizationRecommendView(APIView):
|
||||
"""
|
||||
توصیه کودهی با Celery.
|
||||
توصیه کودهی به صورت مستقیم.
|
||||
POST با sensor_uuid، plant_name، growth_stage.
|
||||
اطلاعات گیاه از plant app دریافت میشود.
|
||||
نیازی به دریافت نوع آبیاری نیست.
|
||||
@@ -44,21 +38,25 @@ class FertilizationRecommendView(APIView):
|
||||
tags=["Fertilization Recommendation"],
|
||||
summary="درخواست توصیه کودهی",
|
||||
description=(
|
||||
"دادههای سنسور و گیاه را دریافت کرده و یک تسک Celery "
|
||||
"برای تولید توصیه کودهی در صف قرار میدهد. "
|
||||
"دادههای سنسور و گیاه را دریافت کرده و "
|
||||
"توصیه کودهی را مستقیم برمیگرداند. "
|
||||
"اطلاعات گیاه از جدول Plant بارگذاری میشود. "
|
||||
"محاسبات مربوط به نیاز آبی در این endpoint انجام نمیشود و مستقل از توصیه کودهی است."
|
||||
),
|
||||
request=FertilizationRecommendRequestSerializer,
|
||||
responses={
|
||||
202: build_response(
|
||||
FertilizationQueueResponseSerializer,
|
||||
"تسک توصیه کودهی در صف قرار گرفت.",
|
||||
200: build_response(
|
||||
FertilizationResponseSerializer,
|
||||
"توصیه کودهی با موفقیت تولید شد.",
|
||||
),
|
||||
400: build_response(
|
||||
FertilizationValidationErrorSerializer,
|
||||
"پارامتر ورودی نامعتبر است.",
|
||||
),
|
||||
500: build_response(
|
||||
FertilizationValidationErrorSerializer,
|
||||
"خطا در تولید توصیه کودهی.",
|
||||
),
|
||||
},
|
||||
examples=[
|
||||
OpenApiExample(
|
||||
@@ -73,7 +71,7 @@ class FertilizationRecommendView(APIView):
|
||||
],
|
||||
)
|
||||
def post(self, request):
|
||||
from rag.tasks import fertilization_recommendation_task
|
||||
from rag.services.fertilization import get_fertilization_recommendation
|
||||
|
||||
serializer = FertilizationRecommendRequestSerializer(data=request.data)
|
||||
if not serializer.is_valid():
|
||||
@@ -88,53 +86,20 @@ class FertilizationRecommendView(APIView):
|
||||
growth_stage = validated.get("growth_stage")
|
||||
query = validated.get("query")
|
||||
|
||||
task = fertilization_recommendation_task.delay(
|
||||
sensor_uuid=sensor_uuid,
|
||||
plant_name=plant_name,
|
||||
growth_stage=growth_stage,
|
||||
query=query,
|
||||
)
|
||||
try:
|
||||
result = get_fertilization_recommendation(
|
||||
sensor_uuid=sensor_uuid,
|
||||
plant_name=plant_name,
|
||||
growth_stage=growth_stage,
|
||||
query=query,
|
||||
)
|
||||
except Exception as exc:
|
||||
return Response(
|
||||
{"code": 500, "msg": f"خطا در تولید توصیه کودهی: {exc}", "data": None},
|
||||
status=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||||
)
|
||||
|
||||
return Response(
|
||||
{
|
||||
"code": 202,
|
||||
"msg": "تسک توصیه کودهی در صف قرار گرفت.",
|
||||
"data": {
|
||||
"task_id": task.id,
|
||||
"status_url": f"/api/fertilization/recommend/{task.id}/status/",
|
||||
},
|
||||
},
|
||||
status=status.HTTP_202_ACCEPTED,
|
||||
)
|
||||
|
||||
|
||||
class FertilizationRecommendStatusView(APIView):
|
||||
"""وضعیت تسک توصیه کودهی."""
|
||||
|
||||
@extend_schema(
|
||||
tags=["Fertilization Recommendation"],
|
||||
summary="وضعیت تسک توصیه کودهی",
|
||||
description="وضعیت تسک Celery توصیه کودهی را برمیگرداند.",
|
||||
responses={
|
||||
200: build_response(
|
||||
FertilizationStatusResponseSerializer,
|
||||
"وضعیت فعلی تسک توصیه کودهی.",
|
||||
),
|
||||
},
|
||||
)
|
||||
def get(self, request, task_id):
|
||||
from celery.result import AsyncResult
|
||||
|
||||
result = AsyncResult(task_id)
|
||||
data = {"task_id": task_id, "status": result.state}
|
||||
if result.state == "PENDING":
|
||||
data["message"] = "تسک در صف یا یافت نشد."
|
||||
elif result.state == "PROGRESS":
|
||||
data["progress"] = result.info
|
||||
elif result.state == "SUCCESS":
|
||||
data["result"] = result.result
|
||||
elif result.state == "FAILURE":
|
||||
data["error"] = str(result.result)
|
||||
return Response(
|
||||
{"code": 200, "msg": "success", "data": data},
|
||||
{"code": 200, "msg": "success", "data": result},
|
||||
status=status.HTTP_200_OK,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user