From 9172e98c45ecb2da20c455ce68290830445f2c85 Mon Sep 17 00:00:00 2001 From: Talitha Pumar Date: Fri, 5 Oct 2018 15:45:48 -0300 Subject: [PATCH] Fix #2286 (#2287) * Fix #2286 * more pythonic --- sapl/materia/forms.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/sapl/materia/forms.py b/sapl/materia/forms.py index 77ccca5d6..6f6a544cd 100644 --- a/sapl/materia/forms.py +++ b/sapl/materia/forms.py @@ -245,7 +245,7 @@ class MateriaLegislativaForm(ModelForm): primeiro_autor = True else: primeiro_autor = False - + materia = super(MateriaLegislativaForm, self).save(commit) materia.save() @@ -281,6 +281,20 @@ class UnidadeTramitacaoForm(ModelForm): raise ValidationError(msg) return cleaned_data + def save(self, commit=False): + unidade = super(UnidadeTramitacaoForm, self).save(commit) + cd = self.cleaned_data + + if not cd.get('orgao'): + unidade.orgao = None + if not cd.get('parlamentar'): + unidade.parlamentar = None + if not cd.get('comissao'): + unidade.comissao = None + + unidade.save() + return unidade + class AcompanhamentoMateriaForm(ModelForm):