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

    .. sectionauthor:: Elton Pereira de Lima
    .. codeauthor:: Elton Pereira de Lima

    """
    protocolo = forms.CharField(required=False)
    ano = forms.ChoiceField()
    mes = forms.ChoiceField()
    municipio = forms.ChoiceField()
    cliente = forms.ChoiceField()

    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')
        municipio_choices = [('','--------')] + [i for i in municipio_choices]
        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


class ProtocoloContabilFormWizardStep2(forms.Form):
    """ Segunda etapa do wizard de envio de dados contábeis.

    .. sectionauthor:: Elton Pereira de Lima
    .. codeauthor:: Elton Pereira de Lima
    """
    def __init__(self, *args, **kwargs):
        data = kwargs.pop('data')
        super(ProtocoloContabilFormWizardStep2, self).__init__(*args, **kwargs)
        if not isinstance(data, QueryDict):
            ano = data.pop('ano')
            mes = data.pop('mes')
            cliente = data.pop('cliente')
            qs = ProtocoloContabilCategoriaArquivo.get_categorias(mes,ano,cliente)

            # É hora de criar os campos dinamicamente.
            for i in xrange(qs.count()):
                self.fields["tipo_%s" % i] = forms.CharField(label=unicode(qs[i]))
                self.fields['relatorio_%s' % i] = forms.FileField(required=False)