class ProtocoloContabilFormWizardStep1(forms.ModelForm):
    """ Primeira etapa do wizard de envio de dados contábeis.

    .. sectionauthor:: Elton Pereira de Lima <eltonplima@gmail.com>
    .. codeauthor:: Elton Pereira de Lima <eltonplima@gmail.com>
    """
    protocolo = forms.CharField(required=False)
    ano = forms.ChoiceField()
    mes = forms.ChoiceField()
    municipio = forms.ChoiceField()
    cliente = forms.ChoiceField()

    class Meta:
        model = ProtocoloContabil
        fields = ('ano',
                  'mes',
                  'cliente',
                  'observacao')

    def __init__(self, *args, **kwargs):
        super(ProtocoloContabilFormWizardStep1, self).__init__(*args, **kwargs)

        ano_choices = ConfiguracaoCabecalhoBase().get_anos_com_configuracao().values_list('id', 'nome')
        ano_choices = [('','--------')] + [i for i in ano_choices]
        mes_choices = ConfiguracaoCabecalhoBase().get_meses_com_configuracao().values_list('id', 'nome')
        mes_choices = [('','--------')] + [i for i in mes_choices]
        municipio_choices = ConfiguracaoCabecalhoBase().get_municipios_com_configuracao().values_list('id', 'nome')
        cliente_choices = ConfiguracaoCabecalhoBase().get_clientes_com_configuracao().values_list('id', 'pessoa_base__nome', 'pessoa_base__pessoajuridica__cnpj')
        cliente_choices = [('','--------')] + [(i[0], '%s(%s)' %(i[1], i[2])) for i in cliente_choices]

        self.fields["protocolo"].widget = forms.HiddenInput()
        self.fields["ano"].choices = ano_choices
        self.fields["mes"].choices = mes_choices
        self.fields["cliente"].choices = cliente_choices

    def clean(self):
        cleaned_data = super(ProtocoloContabilFormWizardStep1, self).clean()
        cleaned_data["ano"] = Ano.objects.get(id=int(cleaned_data["ano"]))
        cleaned_data["mes"] = Mes.objects.get(id=int(cleaned_data["mes"]))
        cleaned_data["cliente"] = Cliente.objects.get(id=int(cleaned_data["cliente"]))

        return cleaned_data