from django.db import models
from projeto.models import Projeto


class Equipamento(models.Model):
    TIPO_CHOICES = [
        ('split', 'Split'),
        ('duto', 'Duto'),
        ('exaustor', 'Exaustor'),
        ('chiller', 'Chiller'),
    ]

    projeto = models.ForeignKey(Projeto, on_delete=models.CASCADE, related_name='equipamentos')
    tipo = models.CharField(max_length=50, choices=TIPO_CHOICES)
    pos_x = models.FloatField()
    pos_y = models.FloatField()
    altura_z = models.FloatField(default=0.0)
    rotacao = models.FloatField(default=0.0)
    dados_extras = models.JSONField(default=dict)
    conexoes = models.JSONField(default=list)

    def __str__(self):
        return f"{self.tipo} - ({self.pos_x}, {self.pos_y})"