Backend/farm_alerts/models.py

import uuid as uuid_lib

from django.db import models

from farm_hub.models import FarmHub


SEVERITY_CHOICES = [
    ("info", "Info"),
    ("warning", "Warning"),
    ("error", "Error"),
    ("success", "Success"),
]


class FarmAlert(models.Model):
    uuid = models.UUIDField(default=uuid_lib.uuid4, unique=True, editable=False, db_index=True)
    farm = models.ForeignKey(FarmHub, on_delete=models.CASCADE, related_name="farm_alerts", null=True, blank=True)
    external_alert_id = models.CharField(max_length=255, blank=True, default="", db_index=True)
    title = models.CharField(max_length=255)
    description = models.TextField(blank=True, default="")
    color = models.CharField(max_length=32, default="info", choices=SEVERITY_CHOICES)
    suggested_action = models.TextField(blank=True, default="")
    source_metric_type = models.CharField(max_length=255, blank=True, default="")
    occurred_at = models.DateTimeField(null=True, blank=True)
    payload = models.JSONField(default=dict, blank=True)
    raw_alert = models.JSONField(default=dict, blank=True)
    avatar_icon = models.CharField(max_length=64, blank=True, default="")
    avatar_color = models.CharField(max_length=32, blank=True, default="")
    is_active = models.BooleanField(default=True)
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = "farm_alerts"
        ordering = ["-created_at"]

    def __str__(self):
        return f"{self.title} ({self.color})"


class AnomalyDetection(models.Model):
    uuid = models.UUIDField(default=uuid_lib.uuid4, unique=True, editable=False, db_index=True)
    farm = models.ForeignKey(FarmHub, on_delete=models.CASCADE, related_name="anomalies", null=True, blank=True)
    sensor = models.CharField(max_length=255)
    value = models.CharField(max_length=64)
    expected = models.CharField(max_length=64)
    deviation = models.CharField(max_length=64)
    severity = models.CharField(max_length=32, default="warning", choices=SEVERITY_CHOICES)
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = "farm_anomaly_detections"
        ordering = ["-created_at"]

    def __str__(self):
        return f"{self.sensor}: {self.value}"


class Recommendation(models.Model):
    uuid = models.UUIDField(default=uuid_lib.uuid4, unique=True, editable=False, db_index=True)
    farm = models.ForeignKey(FarmHub, on_delete=models.CASCADE, related_name="recommendations", null=True, blank=True)
    title = models.CharField(max_length=255)
    subtitle = models.TextField(blank=True, default="")
    avatar_icon = models.CharField(max_length=64, blank=True, default="")
    avatar_color = models.CharField(max_length=32, blank=True, default="")
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = "farm_recommendations"
        ordering = ["-created_at"]

    def __str__(self):
        return self.title


class FarmAlertTrackerSnapshot(models.Model):
    farm = models.OneToOneField(
        FarmHub,
        on_delete=models.CASCADE,
        related_name="alert_tracker_snapshot",
    )
    service_id = models.CharField(max_length=64, default="farm_alerts")
    tracker = models.JSONField(default=dict, blank=True)
    headline = models.CharField(max_length=255, blank=True, default="")
    overview = models.TextField(blank=True, default="")
    status_level = models.CharField(max_length=32, default="info", choices=SEVERITY_CHOICES)
    raw_llm_response = models.TextField(blank=True, default="")
    structured_context = models.JSONField(default=dict, blank=True)
    last_ai_synced_at = models.DateTimeField(null=True, blank=True)
    last_source_update_at = models.DateTimeField(null=True, blank=True)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    class Meta:
        db_table = "farm_alert_tracker_snapshots"

    def __str__(self):
        return f"Tracker snapshot for {self.farm_id}"