92 lines
3.2 KiB
Python
92 lines
3.2 KiB
Python
|
from uuid import UUID
|
||
|
|
|
||
|
|
from celery.result import AsyncResult
|
||
|
|
from drf_spectacular.utils import OpenApiResponse, extend_schema, inline_serializer
|
||
|
|
from rest_framework import serializers as drf_serializers
|
||
|
|
from rest_framework import status
|
||
|
|
from rest_framework.response import Response
|
||
|
|
from rest_framework.views import APIView
|
||
|
|
|
||
|
|
from .tasks import generate_dashboard_data_task
|
||
|
|
|
||
|
|
|
||
|
|
class DashboardDataGenerateView(APIView):
|
||
|
|
@extend_schema(
|
||
|
|
tags=["Dashboard Data"],
|
||
|
|
summary="Generate dashboard data",
|
||
|
|
request=inline_serializer(
|
||
|
|
name="DashboardDataGenerateRequest",
|
||
|
|
fields={
|
||
|
|
"sensor_id": drf_serializers.UUIDField(required=False),
|
||
|
|
"snesor_id": drf_serializers.UUIDField(required=False),
|
||
|
|
},
|
||
|
|
),
|
||
|
|
responses={
|
||
|
|
202: inline_serializer(
|
||
|
|
name="DashboardDataGenerateResponse",
|
||
|
|
fields={
|
||
|
|
"code": drf_serializers.IntegerField(),
|
||
|
|
"msg": drf_serializers.CharField(),
|
||
|
|
"data": inline_serializer(
|
||
|
|
name="DashboardDataGenerateResponseData",
|
||
|
|
fields={
|
||
|
|
"task_id": drf_serializers.CharField(),
|
||
|
|
"status_url": drf_serializers.CharField(),
|
||
|
|
},
|
||
|
|
),
|
||
|
|
},
|
||
|
|
),
|
||
|
|
400: OpenApiResponse(description="Invalid input"),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
def post(self, request):
|
||
|
|
sensor_id = request.data.get("sensor_id") or request.data.get("snesor_id")
|
||
|
|
if not sensor_id:
|
||
|
|
return Response(
|
||
|
|
{"code": 400, "msg": "پارامتر sensor_id الزامی است.", "data": None},
|
||
|
|
status=status.HTTP_400_BAD_REQUEST,
|
||
|
|
)
|
||
|
|
try:
|
||
|
|
sensor_id = str(UUID(str(sensor_id)))
|
||
|
|
except (TypeError, ValueError):
|
||
|
|
return Response(
|
||
|
|
{"code": 400, "msg": "sensor_id باید UUID معتبر باشد.", "data": None},
|
||
|
|
status=status.HTTP_400_BAD_REQUEST,
|
||
|
|
)
|
||
|
|
|
||
|
|
task = generate_dashboard_data_task.delay(sensor_id)
|
||
|
|
return Response(
|
||
|
|
{
|
||
|
|
"code": 202,
|
||
|
|
"msg": "dashboard task queued",
|
||
|
|
"data": {
|
||
|
|
"task_id": task.id,
|
||
|
|
"status_url": f"/api/dashboard-data/{task.id}/status/",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
status=status.HTTP_202_ACCEPTED,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class DashboardDataStatusView(APIView):
|
||
|
|
@extend_schema(
|
||
|
|
tags=["Dashboard Data"],
|
||
|
|
summary="Dashboard task status",
|
||
|
|
)
|
||
|
|
def get(self, request, task_id):
|
||
|
|
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},
|
||
|
|
status=status.HTTP_200_OK,
|
||
|
|
)
|