diff --git a/parlamentares/views.py b/parlamentares/views.py index 9eb0a1a09..1a90a8e44 100644 --- a/parlamentares/views.py +++ b/parlamentares/views.py @@ -879,6 +879,14 @@ class FiliacaoView(FormMixin, GenericView): de outro período de filiação" return self.error_message(parlamentar, form, mensagem) + if (data_desfiliacao and + data_filiacao <= data_init and + data_desfiliacao >= data_fim): + mensagem = "A data de filiação e\ + desfiliação não podem estar no intervalo\ + de outro período de filiação" + return self.error_message(parlamentar, form, mensagem) + # Salva a nova filiação caso tudo esteja correto else: filiacao = form.save(commit=False) @@ -938,6 +946,10 @@ class FiliacaoEditView(FormMixin, GenericView): parlamentar=parlamentar, data_desfiliacao=None) + if 'Excluir' in request.POST: + filiacao.delete() + return self.form_valid(form) + # Vê se o candidato já se filiou alguma vez a algum partido if not candidato_filiado: filiacao = form.save(commit=False) @@ -986,6 +998,15 @@ class FiliacaoEditView(FormMixin, GenericView): return self.error_message(parlamentar, form, mensagem) + if (data_desfiliacao and + data_filiacao <= data_init and + data_desfiliacao >= data_fim): + mensagem = "A data de filiação e\ + desfiliação não podem estar no intervalo\ + de outro período de filiação" + return self.error_message(parlamentar, + form, + mensagem) if 'Salvar' in request.POST: filiacao.save()