from django import forms
from .models import Telhado, TIPO_TELHADO_CHOICES, TIPO_DIFERENCIAL_INSOLACAO_TELHADO

class TelhadoForm(forms.ModelForm):
    tipo_telhado = forms.ChoiceField(choices=TIPO_TELHADO_CHOICES, required=True)
    diferencial_insolacao_telhado = forms.ChoiceField(choices=TIPO_DIFERENCIAL_INSOLACAO_TELHADO, required=False)
    area_telhado = forms.FloatField(required=False, widget=forms.NumberInput(attrs={'step': '0.01'}))

    class Meta:
        model = Telhado
        fields = ['area_telhado', 'tipo_telhado', 'coeficiente_telhado', 'diferencial_insolacao_telhado']

        labels = {
            'area_telhado': 'Área do Telhado (m²)',
            'tipo_telhado': 'Tipo de Telhado',
            'coeficiente_telhado': 'Coeficiente de Transmissão de Calor do Telhado (W/m²·°C)',
            'diferencial_insolacao_telhado': 'Diferencial de Temperatura de Insolação',
        }

        widgets = {
            'coeficiente_telhado': forms.NumberInput(attrs={'step': '0.01'}),
        }