from django.db import models

class HealthRecord(models.Model):
    """健康度记录模型"""
    DEVICE_TYPES = (
        ('generator', '发电机组'),
        ('nacelle', '机舱系统'),
        ('converter', '变流器'),
        ('grid', '电网环境')
    )
    
    device_id = models.CharField(max_length=50, verbose_name="设备ID")
    subsystem = models.CharField(max_length=20, choices=DEVICE_TYPES, verbose_name="子系统类型")
    score = models.FloatField(verbose_name="健康度评分")
    weights = models.JSONField(verbose_name="权重分布")
    timestamp = models.DateTimeField(auto_now_add=True, verbose_name="记录时间")

    class Meta:
        db_table = 'health_records'
        indexes = [models.Index(fields=['device_id', 'subsystem'])]