HOT-FIX: Mais bugs em Bancada/Bloco

interlegis · Mar 3, 2021 · cb73555 · cb73555
1 parent 664576b
commit cb73555
Show file tree

Hide file tree

Showing 4 changed files with 55 additions and 30 deletions.
diff --git a/sapl/parlamentares/forms.py b/sapl/parlamentares/forms.py
@@ -324,17 +324,22 @@ def clean(self):
         return cleaned_data
 
     @transaction.atomic
-    def save(self, commit=True):
+    def save(self, commit=False):
         parlamentar = super(ParlamentarCreateForm, self).save(commit)
-        content_type = ContentType.objects.get_for_model(Parlamentar)
-        object_id = parlamentar.pk
-        tipo = TipoAutor.objects.get(content_type=content_type)
-        Autor.objects.create(
-            content_type=content_type,
-            object_id=object_id,
-            tipo=tipo,
-            nome=parlamentar.nome_parlamentar
-        )
+
+        if not self.instance.pk:
+            parlamentar.save()
+            content_type = ContentType.objects.get_for_model(Parlamentar)
+            object_id = parlamentar.pk
+            tipo = TipoAutor.objects.get(content_type=content_type)
+            Autor.objects.create(
+                content_type=content_type,
+                object_id=object_id,
+                tipo=tipo,
+                nome=parlamentar.nome_parlamentar
+            )
+        else:
+            parlamentar.save()
         return parlamentar
 
 
@@ -693,16 +698,19 @@ def clean(self):
     @transaction.atomic
     def save(self, commit=False):
         bloco = super(BlocoForm, self).save(commit)
-        bloco.save()
-        content_type = ContentType.objects.get_for_model(Bloco)
-        object_id = bloco.pk
-        tipo = TipoAutor.objects.get(content_type=content_type)
-        Autor.objects.create(
-            content_type=content_type,
-            object_id=object_id,
-            tipo=tipo,
-            nome=bloco.nome
-        )
+        if not self.instance.pk:
+            bloco.save()
+            content_type = ContentType.objects.get_for_model(Bloco)
+            object_id = bloco.pk
+            tipo = TipoAutor.objects.get(content_type=content_type)
+            Autor.objects.create(
+                content_type=content_type,
+                object_id=object_id,
+                tipo=tipo,
+                nome=bloco.nome
+            )
+        else:
+            bloco.save()
         return bloco
 
 

diff --git a/sapl/parlamentares/views.py b/sapl/parlamentares/views.py
@@ -1282,6 +1282,12 @@ class CreateView(CrudAux.CreateView):
         def get_success_url(self):
             return reverse('sapl.parlamentares:bloco_list')
 
+    class UpdateView(CrudAux.UpdateView):
+        form_class = BlocoForm
+
+        def get_success_url(self):
+            return reverse('sapl.parlamentares:bloco_list')
+
 
 class BlocoMembroCrud(MasterDetailCrud):
     model = BlocoMembro

diff --git a/sapl/sessao/forms.py b/sapl/sessao/forms.py
@@ -266,17 +266,22 @@ def clean(self):
         return self.cleaned_data
 
     @transaction.atomic
-    def save(self, commit=True):
+    def save(self, commit=False):
         bancada = super(BancadaForm, self).save(commit)
-        content_type = ContentType.objects.get_for_model(Bancada)
-        object_id = bancada.pk
-        tipo = TipoAutor.objects.get(content_type=content_type)
-        Autor.objects.create(
-            content_type=content_type,
-            object_id=object_id,
-            tipo=tipo,
-            nome=bancada.nome
-        )
+
+        if not self.instance.pk:
+            bancada.save()
+            content_type = ContentType.objects.get_for_model(Bancada)
+            object_id = bancada.pk
+            tipo = TipoAutor.objects.get(content_type=content_type)
+            Autor.objects.create(
+                content_type=content_type,
+                object_id=object_id,
+                tipo=tipo,
+                nome=bancada.nome
+            )
+        else:
+            bancada.save()
         return bancada
 
 

diff --git a/sapl/sessao/views.py b/sapl/sessao/views.py
@@ -1113,6 +1113,12 @@ class CreateView(CrudAux.CreateView):
         def get_success_url(self):
             return reverse('sapl.sessao:bancada_list')
 
+    class UpdateView(CrudAux.UpdateView):
+        form_class = BancadaForm
+
+        def get_success_url(self):
+            return reverse('sapl.sessao:bancada_list')
+
 
 def recuperar_numero_sessao_view(request):
     try: