From cf12c320cf6e5f171a2d1d9ac4a6f29929e0537c Mon Sep 17 00:00:00 2001
From: ulyssesBML <ulysses3353@gmail.com>
Date: Fri, 30 Aug 2019 12:36:26 -0300
Subject: [PATCH] =?UTF-8?q?Adcionando=20casos=20com=20diferentes=20regras?=
 =?UTF-8?q?=20de=20realoca=C3=A7=C3=A3o?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 scripts/deduplica_comissao.py | 64 +++++++++++++++++++++++++++++++----
 1 file changed, 57 insertions(+), 7 deletions(-)

diff --git a/scripts/deduplica_comissao.py b/scripts/deduplica_comissao.py
index 29e1313b6..94b29cb58 100644
--- a/scripts/deduplica_comissao.py
+++ b/scripts/deduplica_comissao.py
@@ -1,6 +1,7 @@
 
 from sapl.comissoes.models import Comissao, Composicao, Reuniao
 from sapl.materia.models import DespachoInicial, Relatoria, UnidadeTramitacao
+from sapl.utils import intervalos_tem_intersecao
 
 from difflib import SequenceMatcher
 
@@ -25,26 +26,75 @@ def detecta_duplicados():
     return lst_duplicados
 
 
-def muda_autor(principal, secundaria):
-    for autor in secundaria.autor.all():
-        autor.delete()    
+def realoca_autor(principal, secundaria):
+        autor_principal = principal.autor.first() 
+        clone = secundaria.autor.first()
+
+        for autoria in clone.autoria_set.all():
+            autoria.autor_id = autor_principal
+            autoria.save()
+        
+        for proposicao in clone.proposicao_set.all():
+            proposicao.autor_id = autor_principal
+            proposicao.save()
+        
+        for autorianorma in clone.autorianorma_set.all():
+            autorianorma.autor_id = autor_principal
+            autorianorma.save()
+        
+        for documentoadministrativo in clone.documentoadministrativo_set.all():
+            documentoadministrativo.autor_id = autor[0]
+            documentoadministrativo.save()
+        
+        for protocolo in clone.protocolo_set.all():
+            protocolo.autor_id = autor_principal
+            protocolo.save()
+
+            clone.delete()
+
+def redireciona_apaga(principal,secundaria,apaga):
+    if(apaga):
+        secundaria.delete()
+    else:
+        secundaria.comissao = principal
+        secundaria.save()
+
 
 def muda_models_dependentes(principal,secundaria):
     for model in models_dependentes:
-        for obj in model.objects.filter(comissao=secundaria):
-            obj.comissao = principal
-            obj.save()
+        for obj_secundario in model.objects.filter(comissao=secundaria):
+            for obj_principal in model.objects.filter(comissao=principal):
+                apaga = False
+
+                if model == Composicao:
+                    apaga = intervalos_tem_intersecao(obj_principal.periodo.data_inicio, 
+                                         obj_principal.periodo.data_fim,
+                                         obj_secundario.periodo.data_inicio,
+                                         obj_secundario.periodo.data_fim)
+                
+                elif model == DespachoInicial:
+                    apaga = obj_principal.materia == obj_secundario.materia
+                
+                elif model == Reuniao:
+                    apaga = obj_principal.numero == obj_secundario.numero
+                
+                else:
+                    apaga = False
+
+                redireciona_apaga(obj_principal, obj_secundario, apaga)
 
 def junta_dulpicados(duplicados):
     principal = duplicados[-1]
     for secundaria in duplicados[:-1]:
         muda_models_dependentes(principal,secundaria)
-        muda_autor(principal, secundaria)
+        realoca_autor(principal, secundaria)
         secundaria.delete()       
 
 
 def main():
     lst_duplicados = detecta_duplicados()
+    print('Duplicados encomtrados:\n')
+    print(lst_duplicados)
     for c in lst_duplicados:
         junta_dulpicados(c)