From 26ee44c40caf8a97bc48c51da19a18889affd365 Mon Sep 17 00:00:00 2001
From: Marcio Mazza <marciomazza@gmail.com>
Date: Tue, 28 Aug 2018 15:16:52 -0300
Subject: [PATCH] Adiciona sqls p ressucitar deps diretamente ao arq de ajustes

---
 sapl/legacy/migracao_dados.py                 |  8 ++++++--
 sapl/legacy/scripts/ressucita_dependencias.py | 14 +++++++++++---
 2 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/sapl/legacy/migracao_dados.py b/sapl/legacy/migracao_dados.py
index f417f87f2..dec526ecb 100644
--- a/sapl/legacy/migracao_dados.py
+++ b/sapl/legacy/migracao_dados.py
@@ -811,6 +811,11 @@ def roda_comando_shell(cmd):
     assert res == 0, 'O comando falhou: {}'.format(cmd)
 
 
+def get_arquivo_ajustes_pre_migracao():
+    return DIR_DADOS_MIGRACAO.child(
+        'ajustes_pre_migracao', '{}.sql'.format(sigla_casa))
+
+
 def migrar_dados(apagar_do_legado=False):
     try:
         ocorrencias.clear()
@@ -829,8 +834,7 @@ def migrar_dados(apagar_do_legado=False):
         exec_legado('SET SESSION sql_mode = "NO_AUTO_VALUE_ON_ZERO";')
 
         # executa ajustes pré-migração, se existirem
-        arq_ajustes_pre_migracao = DIR_DADOS_MIGRACAO.child(
-            'ajustes_pre_migracao', '{}.sql'.format(sigla_casa))
+        arq_ajustes_pre_migracao = get_arquivo_ajustes_pre_migracao()
         if arq_ajustes_pre_migracao.exists():
             exec_legado(arq_ajustes_pre_migracao.read_file())
 
diff --git a/sapl/legacy/scripts/ressucita_dependencias.py b/sapl/legacy/scripts/ressucita_dependencias.py
index dce0e1a0b..040476c23 100644
--- a/sapl/legacy/scripts/ressucita_dependencias.py
+++ b/sapl/legacy/scripts/ressucita_dependencias.py
@@ -5,7 +5,8 @@ import yaml
 from unipath import Path
 
 from sapl.legacy.migracao_dados import (DIR_REPO, PROPAGACOES_DE_EXCLUSAO,
-                                        exec_legado)
+                                        exec_legado,
+                                        get_arquivo_ajustes_pre_migracao)
 
 
 def stripsplit(ll):
@@ -338,6 +339,13 @@ def get_sqls_desexcluir_criar(desexcluir, criar, slug):
         return TEMPLATE_RESSUCITADOS.format(links, sqls)
 
 
-def print_ressucitar(slug):
+def get_ressucitar(slug):
     desexcluir, criar = get_dependencias_a_ressucitar(slug)
-    print(get_sqls_desexcluir_criar(desexcluir, criar, slug))
+    return get_sqls_desexcluir_criar(desexcluir, criar, slug)
+
+
+def adiciona_ressucitar(slug):
+    arq_ajustes_pre_migracao = get_arquivo_ajustes_pre_migracao()
+    conteudo = arq_ajustes_pre_migracao.read_file()
+    sqls = get_ressucitar(slug)
+    arq_ajustes_pre_migracao.write_file('{}\n{}'.format(conteudo, sqls))