Adcionando casos com diferentes regras de realocação

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)