feat / AEB-60 create get route for sensors list

This commit is contained in:
Timofey
2025-10-06 15:56:44 +03:00
parent 0aa1c79690
commit f45c4ee14d
9 changed files with 212 additions and 25 deletions

View File

@@ -1,13 +1,84 @@
from rest_framework import serializers
from django.conf import settings
from sitemanagement.models import Sensor, Metric, Alert
from sitemanagement.models import Sensor, Alert
class NotificationSerializer(serializers.ModelSerializer):
type = serializers.SerializerMethodField()
priority = serializers.SerializerMethodField()
timestamp = serializers.DateTimeField(source='created_at')
acknowledged = serializers.BooleanField(source='resolved', default=False)
class Meta:
model = Alert
fields = ('id', 'type', 'timestamp', 'acknowledged', 'priority')
def get_type(self, obj):
return 'critical' if obj.severity == 'critical' else 'warning'
def get_priority(self, obj):
return 'high' if obj.severity == 'critical' else 'medium'
class DetectorSerializer(serializers.ModelSerializer):
detector_id = serializers.SerializerMethodField()
type = serializers.SerializerMethodField()
name = serializers.SerializerMethodField()
object = serializers.SerializerMethodField()
status = serializers.SerializerMethodField()
zone = serializers.SerializerMethodField()
floor = serializers.SerializerMethodField()
notifications = NotificationSerializer(source='alerts', many=True)
class SensorSerializer(serializers.ModelSerializer):
class Meta:
model = Sensor
fields = '__all__'
fields = ('detector_id', 'type', 'name', 'object', 'status', 'zone', 'floor', 'notifications')
class MetricSerializer(serializers.ModelSerializer):
class Meta:
model = Metric
fields = '__all__'
def get_detector_id(self, obj):
return obj.name or f"{obj.sensor_type.code}-{obj.id}"
def get_type(self, obj):
# маппинг типов сенсоров на нужные значения
sensor_type_mapping = {
'GA': 'fire_detector',
'PE': 'pressure_sensor',
'GLE': 'gas_detector'
}
return sensor_type_mapping.get(obj.sensor_type.code, 'unknown')
def get_name(self, obj):
return f"{obj.sensor_type.name} {obj.serial_number}"
def get_object(self, obj):
# получаем первую зону датчика и её объект
zone = obj.zones.first()
if zone:
return zone.object.title
return None
def get_status(self, obj):
# проверяем наличие активных алертов
latest_alert = obj.alerts.filter(resolved=False).first()
if latest_alert:
return latest_alert.severity # вернет 'warning' или 'critical'
return 'normal'
def get_zone(self, obj):
zone = obj.zones.first()
if zone:
return zone.name
return None
def get_floor(self, obj):
# получаем этаж из зоны
zone = obj.zones.first()
if zone:
return zone.floor
return None # если датчик не привязали к зоне
class DetectorsResponseSerializer(serializers.Serializer):
detectors = serializers.SerializerMethodField()
def get_detectors(self, sensors):
detector_serializer = DetectorSerializer(sensors, many=True)
return {
sensor['detector_id']: sensor
for sensor in detector_serializer.data
}

View File

@@ -1,6 +1,7 @@
from django.urls import path
from .views.UserDataView import UserDataView
from .views.objects_views import ObjectView
from .views.sensors_views import SensorView
from drf_spectacular.views import (
SpectacularAPIView,
SpectacularSwaggerView,
@@ -23,4 +24,5 @@ urlpatterns = [
path("user/", UserDataView.as_view(), name="user-data"),
path("get-objects/", ObjectView.as_view(), name="objects"),
path("get-detectors/", SensorView.as_view(), name="detectors"),
]

View File

@@ -1,10 +0,0 @@
from rest_framework import status
from rest_framework.views import APIView
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from drf_spectacular.utils import extend_schema, OpenApiResponse, OpenApiExample
from api.auth.serializers import UserResponseSerializer
from api.models import UserProfile
from api.utils.decorators import handle_exceptions

View File

@@ -2,13 +2,14 @@ from rest_framework import status
from rest_framework.views import APIView
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from drf_spectacular.utils import extend_schema, OpenApiResponse
from drf_spectacular.utils import extend_schema, OpenApiResponse, OpenApiExample
from api.account.serializers.objects_serializers import ObjectSerializer
from sitemanagement.models import Object
from api.utils.decorators import handle_exceptions
@extend_schema(tags=['Объекты'])
class ObjectView(APIView):
permission_classes = [IsAuthenticated]
serializer_class = ObjectSerializer
@@ -16,7 +17,34 @@ class ObjectView(APIView):
@extend_schema(
summary="Получение всех объектов",
description="Получение всех объектов",
responses={200: OpenApiResponse(response=ObjectSerializer, description="Объекты успешно получены")})
responses={200: OpenApiResponse(response=ObjectSerializer, description="Объекты успешно получены",
examples=[OpenApiExample(
'Успешный ответ',
value={
"id": 1,
"title": "Объект 1",
"description": "Описание объекта 1",
"image": "https://example.com/image.jpg",
"address": "Адрес объекта 1",
"floors": 1,
"area": 100.0,
"zones": [
{
"id": 1,
"name": "Зона 1",
"sensors": [
{
"id": 1,
"name": "Датчик 1",
"serial_number": "GA-123",
"sensor_type": "Тип датчика 1"
}
]
}
]
}
)])})
@handle_exceptions
def get(self, request):
"""Получение всех объектов с их зонами и датчиками"""

View File

@@ -0,0 +1,61 @@
from rest_framework import status
from rest_framework.views import APIView
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from drf_spectacular.utils import extend_schema, OpenApiResponse, OpenApiExample
from api.account.serializers.sensor_serializers import DetectorsResponseSerializer
from sitemanagement.models import Sensor
from api.utils.decorators import handle_exceptions
@extend_schema(tags=['Датчики'])
class SensorView(APIView):
# permission_classes = [IsAuthenticated]
serializer_class = DetectorsResponseSerializer
@extend_schema(
summary="Получение всех датчиков",
description="Получение всех датчиков в формате детекторов",
responses={200: OpenApiResponse(response=DetectorsResponseSerializer, description="Датчики успешно получены",
examples=[OpenApiExample(
'Успешный ответ',
value={
"id": 1,
"type": "fire_detector",
"name": "Датчик 1",
"object": "Объект 1",
"status": "warning",
"zone": "Зона 1",
"floor": 1,
"notifications": [
{
"id": 1,
"type": "warning",
"timestamp": "2024-01-15T14:30:00Z",
"acknowledged": False,
"priority": "high"
}
]
}
)])})
@handle_exceptions
def get(self, request):
"""Получение всех датчиков"""
try:
sensors = Sensor.objects.select_related(
'sensor_type',
'signal_format'
).prefetch_related(
'zones',
'zones__object',
'alerts'
).all()
serializer = DetectorsResponseSerializer(sensors)
return Response(serializer.data, status=status.HTTP_200_OK)
except Sensor.DoesNotExist:
return Response(
{"error": "Датчики не найдены"},
status=status.HTTP_404_NOT_FOUND)