Adiciona sqls p ressucitar deps diretamente ao arq de ajustes

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())
 

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))