Ai/location_data/views.py

from rest_framework import status
from drf_spectacular.utils import (
    OpenApiExample,
    OpenApiResponse,
    extend_schema,
    inline_serializer,
)
from rest_framework import serializers as drf_serializers
from rest_framework.response import Response
from rest_framework.views import APIView

from .models import SoilLocation
from .serializers import (
    SoilDataRequestSerializer,
    SoilDataTaskResponseSerializer,
    SoilLocationResponseSerializer,
)
from .tasks import fetch_soil_data_task


class SoilDataView(APIView):
    """
    API خاک: مختصات جغرافیایی را می‌گیرد.
    اگر داده در DB موجود باشد، برگردانده می‌شود؛ در غیر این صورت
    تسک Celery صف می‌شود و task_id برمی‌گردد.
    """

    def _get_request_data(self, request):
        return request.data if request.method == "POST" else request.query_params

    @extend_schema(
        tags=["Soil Data"],
        summary="دریافت داده خاک (GET)",
        description="با ارسال lat و lon، داده خاک از DB یا از طریق تسک Celery برگردانده می‌شود.",
        parameters=[
            {
                "name": "lat",
                "in": "query",
                "required": True,
                "schema": {"type": "number"},
                "description": "عرض جغرافیایی",
            },
            {
                "name": "lon",
                "in": "query",
                "required": True,
                "schema": {"type": "number"},
                "description": "طول جغرافیایی",
            },
        ],
        responses={
            200: OpenApiResponse(description="داده خاک از دیتابیس"),
            202: OpenApiResponse(description="تسک در صف قرار گرفت"),
            400: OpenApiResponse(description="داده نامعتبر"),
        },
    )
    def get(self, request):
        return self._process(request)

    @extend_schema(
        tags=["Soil Data"],
        summary="دریافت داده خاک (POST)",
        description="با ارسال lat و lon در بدنه، داده خاک از DB یا از طریق تسک Celery برگردانده می‌شود.",
        request=SoilDataRequestSerializer,
        responses={
            200: OpenApiResponse(description="داده خاک از دیتابیس"),
            202: OpenApiResponse(description="تسک در صف قرار گرفت"),
            400: OpenApiResponse(description="داده نامعتبر"),
        },
        examples=[
            OpenApiExample(
                "نمونه درخواست",
                value={"lat": 35.6892, "lon": 51.3890},
                request_only=True,
            ),
        ],
    )
    def post(self, request):
        return self._process(request)

    def _process(self, request):
        data = self._get_request_data(request)
        serializer = SoilDataRequestSerializer(data=data)
        if not serializer.is_valid():
            return Response(
                {"code": 400, "msg": "داده نامعتبر.", "data": serializer.errors},
                status=status.HTTP_400_BAD_REQUEST,
            )
        lat = serializer.validated_data["lat"]
        lon = serializer.validated_data["lon"]
        lat_rounded = round(lat, 6)
        lon_rounded = round(lon, 6)

        location = (
            SoilLocation.objects.filter(
                latitude=lat_rounded,
                longitude=lon_rounded,
            )
            .prefetch_related("depths")
            .first()
        )

        if location and location.is_complete:
            data_serializer = SoilLocationResponseSerializer(location)
            return Response(
                {
                    "code": 200,
                    "msg": "success",
                    "data": {
                        "source": "database",
                        **data_serializer.data,
                    },
                },
                status=status.HTTP_200_OK,
            )

        result = fetch_soil_data_task.delay(float(lat_rounded), float(lon_rounded))
        task_data = SoilDataTaskResponseSerializer(
            {
                "task_id": result.id,
                "longitude": float(lon_rounded),
                "latitude": float(lat_rounded),
                "status_url": f"/api/soil-data/tasks/{result.id}/status/",
            }
        ).data
        return Response(
            {
                "code": 202,
                "msg": "تسک در صف. وضعیت را با task_id بررسی کنید.",
                "data": task_data,
            },
            status=status.HTTP_202_ACCEPTED,
        )


class SoilDataTaskStatusView(APIView):
    """وضعیت تسک واکشی خاک. در صورت SUCCESS لیست اطلاعات هر سه عمق برگردانده می‌شود."""

    @extend_schema(
        tags=["Soil Data"],
        summary="وضعیت تسک داده خاک",
        description="وضعیت تسک Celery واکشی داده خاک را برمی‌گرداند.",
        responses={
            200: inline_serializer(
                name="SoilTaskStatusResponse",
                fields={
                    "code": drf_serializers.IntegerField(),
                    "msg": drf_serializers.CharField(),
                    "data": inline_serializer(
                        name="SoilTaskStatusData",
                        fields={
                            "task_id": drf_serializers.CharField(),
                            "status": drf_serializers.CharField(),
                            "message": drf_serializers.CharField(required=False),
                            "progress": drf_serializers.DictField(required=False),
                            "result": drf_serializers.JSONField(required=False),
                            "error": drf_serializers.CharField(required=False),
                        },
                    ),
                },
            ),
        },
    )
    def get(self, request, task_id):
        from celery.result import AsyncResult

        result = AsyncResult(task_id)
        state = result.state
        data = {"task_id": task_id, "status": state}

        if state == "PENDING":
            data["message"] = "تسک در صف یا یافت نشد."
        elif state == "PROGRESS":
            data["progress"] = result.info
        elif state == "SUCCESS":
            task_result = result.result
            if isinstance(task_result, dict) and task_result.get("status") == "completed":
                location_id = task_result.get("location_id")
                location = (
                    SoilLocation.objects.filter(pk=location_id)
                    .prefetch_related("depths")
                    .first()
                )
                if location and location.is_complete:
                    data["result"] = SoilLocationResponseSerializer(location).data
                else:
                    data["result"] = task_result
            else:
                data["result"] = task_result
        elif state == "FAILURE":
            data["error"] = str(result.result)

        return Response(
            {"code": 200, "msg": "success", "data": data},
            status=status.HTTP_200_OK,
        )