from django import forms
from .models import UPPER_CASE
from tenantapp.models import Tenant

class UPPER_CASEForm(forms.ModelForm):
    class Meta:
        model = UPPER_CASE
        fields = ['nome', 'descricao'] # não incluir tenant nesta lista
        widgets = {
            'nome': forms.TextInput(attrs={'class': 'form-control'}),
            'descricao': forms.TextInput(attrs={'class': 'form-control'}),
        }

    def __init__(self, *args, **kwargs):
        user = kwargs.pop('user', None)  # Recebe o user na criação do form
        super().__init__(*args, **kwargs)
        self.is_superadmin = False

        if user and user.is_superadmin():
            self.is_superadmin = True
            self.fields['tenant'] = forms.ModelChoiceField(
                queryset=Tenant.objects.all(),
                widget=forms.Select(attrs={'class': 'form-control'}),
                required=True,
                label='Tenant'
            )