100 lines
4.0 KiB
Python
100 lines
4.0 KiB
Python
|
from rest_framework import serializers
|
||
|
|
|
||
|
|
from notifications.serializers import FarmNotificationSerializer
|
||
|
|
|
||
|
|
|
||
|
|
ALLOWED_TRACKER_FIELDS = {"farm_uuid", "alerts"}
|
||
|
|
|
||
|
|
|
||
|
|
class FarmAlertInputSerializer(serializers.Serializer):
|
||
|
|
alert_id = serializers.CharField(required=False, allow_blank=True)
|
||
|
|
level = serializers.CharField(required=False, allow_blank=True)
|
||
|
|
title = serializers.CharField(required=False, allow_blank=True)
|
||
|
|
message = serializers.CharField(required=False, allow_blank=True)
|
||
|
|
suggested_action = serializers.CharField(required=False, allow_blank=True)
|
||
|
|
source_metric_type = serializers.CharField(required=False, allow_blank=True)
|
||
|
|
timestamp = serializers.DateTimeField(required=False, allow_null=True)
|
||
|
|
payload = serializers.JSONField(required=False)
|
||
|
|
|
||
|
|
|
||
|
|
class FarmAlertsTrackerRequestSerializer(serializers.Serializer):
|
||
|
|
farm_uuid = serializers.UUIDField(required=True, help_text="UUID مزرعه.")
|
||
|
|
alerts = FarmAlertInputSerializer(many=True, required=False, default=list)
|
||
|
|
|
||
|
|
def validate(self, attrs):
|
||
|
|
initial_keys = set(getattr(self, "initial_data", {}).keys())
|
||
|
|
extra_fields = initial_keys - ALLOWED_TRACKER_FIELDS
|
||
|
|
if extra_fields:
|
||
|
|
raise serializers.ValidationError(
|
||
|
|
{field: ["This field is not allowed."] for field in sorted(extra_fields)}
|
||
|
|
)
|
||
|
|
return attrs
|
||
|
|
|
||
|
|
|
||
|
|
class AlertTrackerAIResponseSerializer(serializers.Serializer):
|
||
|
|
farm_uuid = serializers.UUIDField(read_only=True)
|
||
|
|
service_id = serializers.CharField()
|
||
|
|
tracker = serializers.JSONField()
|
||
|
|
headline = serializers.CharField(allow_blank=True)
|
||
|
|
overview = serializers.CharField(allow_blank=True)
|
||
|
|
status_level = serializers.CharField()
|
||
|
|
notifications = FarmNotificationSerializer(many=True)
|
||
|
|
raw_llm_response = serializers.CharField(allow_blank=True)
|
||
|
|
structured_context = serializers.JSONField()
|
||
|
|
|
||
|
|
|
||
|
|
class AlertStatSerializer(serializers.Serializer):
|
||
|
|
title = serializers.CharField()
|
||
|
|
count = serializers.CharField()
|
||
|
|
avatarColor = serializers.CharField()
|
||
|
|
avatarIcon = serializers.CharField()
|
||
|
|
|
||
|
|
|
||
|
|
class AlertTrackerSerializer(serializers.Serializer):
|
||
|
|
totalAlerts = serializers.IntegerField()
|
||
|
|
radialBarValue = serializers.IntegerField()
|
||
|
|
alertStats = AlertStatSerializer(many=True)
|
||
|
|
|
||
|
|
|
||
|
|
class AlertTimelineItemSerializer(serializers.Serializer):
|
||
|
|
title = serializers.CharField()
|
||
|
|
description = serializers.CharField()
|
||
|
|
time = serializers.CharField()
|
||
|
|
color = serializers.CharField()
|
||
|
|
|
||
|
|
|
||
|
|
class AlertTimelineSerializer(serializers.Serializer):
|
||
|
|
alerts = AlertTimelineItemSerializer(many=True)
|
||
|
|
|
||
|
|
|
||
|
|
class AnomalyItemSerializer(serializers.Serializer):
|
||
|
|
sensor = serializers.CharField()
|
||
|
|
value = serializers.CharField()
|
||
|
|
expected = serializers.CharField()
|
||
|
|
deviation = serializers.CharField()
|
||
|
|
severity = serializers.CharField()
|
||
|
|
|
||
|
|
|
||
|
|
class AnomalyDetectionSerializer(serializers.Serializer):
|
||
|
|
anomalies = AnomalyItemSerializer(many=True)
|
||
|
|
|
||
|
|
|
||
|
|
class RecommendationItemSerializer(serializers.Serializer):
|
||
|
|
title = serializers.CharField()
|
||
|
|
subtitle = serializers.CharField()
|
||
|
|
avatarIcon = serializers.CharField()
|
||
|
|
avatarColor = serializers.CharField()
|
||
|
|
|
||
|
|
|
||
|
|
class RecommendationsListSerializer(serializers.Serializer):
|
||
|
|
recommendations = RecommendationItemSerializer(many=True)
|
||
|
|
|
||
|
|
|
||
|
|
class CreateAlertSerializer(serializers.Serializer):
|
||
|
|
farm_uuid = serializers.UUIDField(required=False, allow_null=True, help_text="UUID مزرعه برای اتصال alert به مزرعه.")
|
||
|
|
title = serializers.CharField(max_length=255, help_text="عنوان هشدار.")
|
||
|
|
description = serializers.CharField(required=False, default="", allow_blank=True, help_text="توضیح هشدار.")
|
||
|
|
color = serializers.ChoiceField(choices=["info", "warning", "error", "success"], default="info", help_text="سطح یا رنگ هشدار.")
|
||
|
|
avatar_icon = serializers.CharField(required=False, default="", allow_blank=True, help_text="آیکون هشدار.")
|
||
|
|
avatar_color = serializers.CharField(required=False, default="", allow_blank=True, help_text="رنگ آواتار هشدار.")
|