from django.db import models
from core.models import Motorista, Fornecedor

class Coleta(models.Model):
    motorista = models.ForeignKey(Motorista, on_delete=models.CASCADE, related_name='coletas', null=True, blank=True)
    fornecedor = models.ForeignKey(Fornecedor, on_delete=models.CASCADE, related_name='coletas', null=True, blank=True)
    data_coleta = models.DateTimeField(auto_now_add=True)

    # Parâmetros da coleta 
    temperatura = models.DecimalField(max_digits=5, decimal_places=2, null=True, blank=True)
    teor_gordura = models.DecimalField(max_digits=5, decimal_places=2, null=True, blank=True)
    ph = models.DecimalField(max_digits=4, decimal_places=2, null=True, blank=True)
    volume = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)
    condutividade = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)
    agua_adicionada = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)
    proteina = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)  # Associada ao Nitrogênio
    solidos = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)
    lactose = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)
    nitrogenio = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)
    fosforo = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)
    potassio = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)

    def __str__(self):
        return f"Coleta {self.id} - {self.fornecedor.nome if self.fornecedor else 'Sem Fornecedor'}"