Resolvendo code climate

sapl/base/forms.py

@@ -1436,11 +1436,9 @@ class PartidoForm(FileFieldCheckMixin, ModelForm):
 
 class PartidoUpdateForm(PartidoForm):
 
-    opcoes = (('nao','Não'),
+    opcoes = YES_NO_CHOICES
-              ('sim','Sim')
-            )
 
-    historico = forms.ChoiceField(initial='nao', choices=opcoes)
+    historico = forms.ChoiceField(initial=False, choices=opcoes)
 
 
     class Meta:
@@ -1475,7 +1473,7 @@ class PartidoUpdateForm(PartidoForm):
         super(PartidoUpdateForm,self).clean()
         cleaned_data = self.cleaned_data
 
-        is_historico = cleaned_data['historico'] == 'sim' 
+        is_historico = (cleaned_data['historico'] == 'True')
 
         if is_historico: 
             if not cleaned_data['data_criacao'] or not cleaned_data['data_extincao']:
@@ -1485,7 +1483,6 @@ class PartidoUpdateForm(PartidoForm):
             if self.instance.data_extincao and  self.instance.data_extincao < cleaned_data['data_extincao']:
                  raise ValidationError("Data de fim de historico deve ser anterior a data de extinção do partido.")
 
-    
             if self.instance.pk:
                 partido = Partido.objects.get(pk=self.instance.pk)
                 historico = HistoricoPartido.objects.filter(partido=partido).order_by('-inicio_historico')
@@ -1499,26 +1496,28 @@ class PartidoUpdateForm(PartidoForm):
         return cleaned_data
 
     def save(self,commit=False):
-       partido = self.instance
+        partido = self.instance
-       is_historico = self.cleaned_data['historico'] 
+    
-           
+        cleaned_data = self.cleaned_data
-       if is_historico == "nao":
+        is_historico = (cleaned_data['historico'] == 'True')
-           partido.save(commit)
+            
-       elif is_historico == "sim":
+        if not is_historico:
-           sigla = self.cleaned_data['sigla']
+            partido.save(commit)
-           nome = self.cleaned_data['nome']
+        else:
-           inicio_historico = self.cleaned_data['data_criacao']
+            sigla = self.cleaned_data['sigla']
-           fim_historico = self.cleaned_data['data_extincao']
+            nome = self.cleaned_data['nome']
-           logo_partido = self.cleaned_data['logo_partido']
+            inicio_historico = self.cleaned_data['data_criacao']
-           historico_partido = HistoricoPartido(sigla=sigla,
+            fim_historico = self.cleaned_data['data_extincao']
-                                               nome=nome,
+            logo_partido = self.cleaned_data['logo_partido']
-                                               inicio_historico=inicio_historico,
+            historico_partido = HistoricoPartido(sigla=sigla,
-                                               fim_historico=fim_historico,
+                                                nome=nome,
-                                               logo_partido=logo_partido,
+                                                inicio_historico=inicio_historico,
-                                               partido=partido,
+                                                fim_historico=fim_historico,
-                                               )
+                                                logo_partido=logo_partido,
-           historico_partido.save()
+                                                partido=partido,
-       return partido
+                                                )
+            historico_partido.save()
+        return partido
 
 class RelatorioHistoricoTramitacaoAdmFilterSet(django_filters.FilterSet):
 
@@ -1557,3 +1556,4 @@ class RelatorioHistoricoTramitacaoAdmFilterSet(django_filters.FilterSet):
                      row1, row2, row3,
                      form_actions(label='Pesquisar'))
         )
+