143 lines
12 KiB
Python
143 lines
12 KiB
Python
from rest_framework import serializers, status
|
|
from rest_framework.pagination import PageNumberPagination
|
|
from rest_framework.permissions import AllowAny, IsAuthenticated
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
from drf_spectacular.utils import OpenApiExample, OpenApiParameter, OpenApiTypes, extend_schema
|
|
|
|
from config.swagger import code_response, farm_uuid_query_param
|
|
from farm_hub.models import FarmHub
|
|
from notifications.serializers import FarmNotificationSerializer
|
|
from soil.serializers import SoilComparisonChartSerializer, SoilRadarChartSerializer
|
|
|
|
from .authentication import SensorExternalAPIKeyAuthentication
|
|
from .sensor_serializers import Sensor7In1SummarySerializer, SensorComparisonChartQuerySerializer, SensorComparisonChartResponseSerializer, SensorRadarChartQuerySerializer, SensorRadarChartResponseSerializer, SensorValuesListQuerySerializer, SensorValuesListResponseSerializer
|
|
from .serializers import SensorCatalogSerializer, SensorExternalRequestLogQuerySerializer, SensorExternalRequestLogSerializer, SensorExternalRequestSerializer
|
|
from .services import FarmDataForwardError, create_sensor_external_notification, forward_sensor_payload_to_farm_data, get_farm_sensor_map_for_logs, get_primary_soil_sensor, get_sensor_7_in_1_comparison_chart_data, get_sensor_7_in_1_radar_chart_data, get_sensor_7_in_1_summary_data, get_sensor_comparison_chart_data, get_sensor_external_request_logs_for_farm, get_sensor_radar_chart_data, get_sensor_values_list_data
|
|
|
|
|
|
class SensorCatalogListView(APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
@extend_schema(tags=["Sensor Catalog"], responses={200: code_response("SensorCatalogListResponse", data=SensorCatalogSerializer(many=True))})
|
|
def get(self, request):
|
|
from .models import SensorCatalog
|
|
return Response({"code": 200, "msg": "success", "data": SensorCatalogSerializer(SensorCatalog.objects.order_by("code"), many=True).data}, status=status.HTTP_200_OK)
|
|
|
|
|
|
class Sensor7In1SummaryView(APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
required_feature_code = "sensor-7-in-1"
|
|
|
|
@staticmethod
|
|
def _get_farm(request):
|
|
farm_uuid = request.query_params.get("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
|
|
|
|
@staticmethod
|
|
def _get_primary_sensor(*, farm):
|
|
sensor = get_primary_soil_sensor(farm=farm)
|
|
if sensor is None:
|
|
raise serializers.ValidationError({"farm_uuid": ["No sensor found for this farm."]})
|
|
return sensor
|
|
|
|
@extend_schema(tags=["Sensor 7 in 1"], parameters=[farm_uuid_query_param(required=True, description="UUID of the farm for sensor 7 in 1 summary.")], responses={200: code_response("Sensor7In1SummaryResponse", data=Sensor7In1SummarySerializer())})
|
|
def get(self, request):
|
|
farm = self._get_farm(request)
|
|
return Response({"code": 200, "msg": "OK", "data": get_sensor_7_in_1_summary_data(farm)}, status=status.HTTP_200_OK)
|
|
|
|
|
|
class Sensor7In1RadarChartView(Sensor7In1SummaryView):
|
|
@extend_schema(tags=["Sensor 7 in 1"], parameters=[farm_uuid_query_param(required=True, description="UUID of the farm for sensor 7 in 1 radar chart.")], responses={200: code_response("Sensor7In1RadarChartResponse", data=SoilRadarChartSerializer())})
|
|
def get(self, request):
|
|
farm = self._get_farm(request)
|
|
return Response({"code": 200, "msg": "OK", "data": get_sensor_7_in_1_radar_chart_data(farm)}, status=status.HTTP_200_OK)
|
|
|
|
|
|
class Sensor7In1ComparisonChartView(Sensor7In1SummaryView):
|
|
@extend_schema(tags=["Sensor 7 in 1"], parameters=[farm_uuid_query_param(required=True, description="UUID of the farm for sensor 7 in 1 comparison chart.")], responses={200: code_response("Sensor7In1ComparisonChartResponse", data=SoilComparisonChartSerializer())})
|
|
def get(self, request):
|
|
farm = self._get_farm(request)
|
|
return Response({"code": 200, "msg": "OK", "data": get_sensor_7_in_1_comparison_chart_data(farm)}, status=status.HTTP_200_OK)
|
|
|
|
|
|
class SensorComparisonChartView(Sensor7In1SummaryView):
|
|
@extend_schema(tags=["Sensor 7 in 1"], parameters=[farm_uuid_query_param(required=True, description="UUID of the farm."), OpenApiParameter(name="range", type=OpenApiTypes.STR, location=OpenApiParameter.QUERY, required=False, description="Chart range, supported values: 7d, 30d. Defaults to 7d.")], responses={200: SensorComparisonChartResponseSerializer})
|
|
def get(self, request):
|
|
serializer = SensorComparisonChartQuerySerializer(data=request.query_params)
|
|
serializer.is_valid(raise_exception=True)
|
|
farm = self._get_farm(request)
|
|
sensor = self._get_primary_sensor(farm=farm)
|
|
return Response(get_sensor_comparison_chart_data(farm=farm, physical_device_uuid=sensor.physical_device_uuid, range_value=serializer.validated_data["range"]), status=status.HTTP_200_OK)
|
|
|
|
|
|
class SensorValuesListView(Sensor7In1SummaryView):
|
|
@extend_schema(tags=["Sensor 7 in 1"], parameters=[farm_uuid_query_param(required=True, description="UUID of the farm."), OpenApiParameter(name="range", type=OpenApiTypes.STR, location=OpenApiParameter.QUERY, required=False, description="Values list range, supported values: 1h, 24h, 7d. Defaults to 7d.")], responses={200: SensorValuesListResponseSerializer})
|
|
def get(self, request):
|
|
serializer = SensorValuesListQuerySerializer(data=request.query_params)
|
|
serializer.is_valid(raise_exception=True)
|
|
farm = self._get_farm(request)
|
|
sensor = self._get_primary_sensor(farm=farm)
|
|
return Response(get_sensor_values_list_data(farm=farm, physical_device_uuid=sensor.physical_device_uuid, range_value=serializer.validated_data["range"]), status=status.HTTP_200_OK)
|
|
|
|
|
|
class SensorRadarChartView(Sensor7In1SummaryView):
|
|
@extend_schema(tags=["Sensor 7 in 1"], parameters=[farm_uuid_query_param(required=True, description="UUID of the farm."), OpenApiParameter(name="range", type=OpenApiTypes.STR, location=OpenApiParameter.QUERY, required=False, description="Radar chart range, supported values: today, 7d, 30d. Defaults to 7d.")], responses={200: SensorRadarChartResponseSerializer})
|
|
def get(self, request):
|
|
serializer = SensorRadarChartQuerySerializer(data=request.query_params)
|
|
serializer.is_valid(raise_exception=True)
|
|
farm = self._get_farm(request)
|
|
sensor = self._get_primary_sensor(farm=farm)
|
|
return Response(get_sensor_radar_chart_data(farm=farm, physical_device_uuid=sensor.physical_device_uuid, range_value=serializer.validated_data["range"]), status=status.HTTP_200_OK)
|
|
|
|
|
|
class SensorExternalRequestLogPagination(PageNumberPagination):
|
|
page_size = 20
|
|
page_size_query_param = "page_size"
|
|
max_page_size = 100
|
|
|
|
|
|
class SensorExternalAPIView(APIView):
|
|
authentication_classes = [SensorExternalAPIKeyAuthentication]
|
|
permission_classes = [AllowAny]
|
|
|
|
@extend_schema(tags=["Sensor External API"], request=SensorExternalRequestSerializer, examples=[OpenApiExample("Sensor External API Request", value={"uuid": "22222222-2222-2222-2222-222222222222", "payload": {"moisture_percent": 32.5, "temperature_c": 21.3, "ph": 6.7, "ec_ds_m": 1.1, "nitrogen_mg_kg": 42, "phosphorus_mg_kg": 18, "potassium_mg_kg": 210}}, request_only=True)], parameters=[OpenApiParameter(name="X-API-Key", type=OpenApiTypes.STR, location=OpenApiParameter.HEADER, required=True, default="12345", description="API key for sensor external API.")], responses={201: code_response("SensorExternalAPIResponse", data=FarmNotificationSerializer()), 401: code_response("SensorExternalAPIUnauthorizedResponse"), 404: code_response("SensorExternalAPIDeviceNotFoundResponse"), 503: code_response("SensorExternalAPIUnavailableResponse")})
|
|
def post(self, request):
|
|
serializer = SensorExternalRequestSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
try:
|
|
notification = create_sensor_external_notification(physical_device_uuid=serializer.validated_data["uuid"], payload=serializer.validated_data.get("payload"))
|
|
forward_sensor_payload_to_farm_data(physical_device_uuid=serializer.validated_data["uuid"], payload=serializer.validated_data.get("payload"))
|
|
except ValueError as exc:
|
|
if "not migrated" in str(exc):
|
|
return Response({"code": 503, "msg": "Required tables are not ready. Run migrations."}, status=status.HTTP_503_SERVICE_UNAVAILABLE)
|
|
return Response({"code": 404, "msg": "Physical device not found."}, status=status.HTTP_404_NOT_FOUND)
|
|
except FarmDataForwardError as exc:
|
|
return Response({"code": 503, "msg": str(exc)}, status=status.HTTP_503_SERVICE_UNAVAILABLE)
|
|
return Response({"code": 201, "msg": "success", "data": FarmNotificationSerializer(notification).data}, status=status.HTTP_201_CREATED)
|
|
|
|
|
|
class SensorExternalRequestLogListAPIView(APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
pagination_class = SensorExternalRequestLogPagination
|
|
|
|
@extend_schema(tags=["Sensor External API"], parameters=[OpenApiParameter(name="farm_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=True, default="11111111-1111-1111-1111-111111111111"), OpenApiParameter(name="page", type=OpenApiTypes.INT, location=OpenApiParameter.QUERY, required=True), OpenApiParameter(name="page_size", type=OpenApiTypes.INT, location=OpenApiParameter.QUERY, required=True), OpenApiParameter(name="physical_device_uuid", type=OpenApiTypes.UUID, location=OpenApiParameter.QUERY, required=False), OpenApiParameter(name="sensor_type", type=OpenApiTypes.STR, location=OpenApiParameter.QUERY, required=False), OpenApiParameter(name="date_from", type=OpenApiTypes.DATE, location=OpenApiParameter.QUERY, required=False), OpenApiParameter(name="date_to", type=OpenApiTypes.DATE, location=OpenApiParameter.QUERY, required=False)], responses={200: code_response("SensorExternalRequestLogListResponse", data=SensorExternalRequestLogSerializer(many=True), extra_fields={"count": serializers.IntegerField(), "next": serializers.CharField(allow_null=True), "previous": serializers.CharField(allow_null=True)}), 401: code_response("SensorExternalRequestLogListUnauthorizedResponse"), 503: code_response("SensorExternalRequestLogListUnavailableResponse")})
|
|
def get(self, request):
|
|
serializer = SensorExternalRequestLogQuerySerializer(data=request.query_params)
|
|
serializer.is_valid(raise_exception=True)
|
|
try:
|
|
queryset = get_sensor_external_request_logs_for_farm(farm_uuid=serializer.validated_data["farm_uuid"], physical_device_uuid=serializer.validated_data.get("physical_device_uuid"), sensor_type=serializer.validated_data.get("sensor_type"), date_from=serializer.validated_data.get("date_from"), date_to=serializer.validated_data.get("date_to"))
|
|
except ValueError:
|
|
return Response({"code": 503, "msg": "Required tables are not ready. Run migrations."}, status=status.HTTP_503_SERVICE_UNAVAILABLE)
|
|
paginator = self.pagination_class()
|
|
paginator.page_size = serializer.validated_data["page_size"]
|
|
page = paginator.paginate_queryset(queryset, request, view=self)
|
|
farm_sensor_map = get_farm_sensor_map_for_logs(logs=page)
|
|
data = SensorExternalRequestLogSerializer(page, many=True, context={"farm_sensor_map": farm_sensor_map}).data
|
|
return Response({"code": 200, "msg": "success", "count": paginator.page.paginator.count, "next": paginator.get_next_link(), "previous": paginator.get_previous_link(), "data": data}, status=status.HTTP_200_OK)
|