From f51bd99c3444880f5dcd179948e01cad7435a16c Mon Sep 17 00:00:00 2001 From: Marcio Mazza Date: Fri, 31 Aug 2018 16:09:01 -0300 Subject: [PATCH] =?UTF-8?q?Usa=20slug=20de=20arquivo=20de=20configura?= =?UTF-8?q?=C3=A7=C3=A3o=20ao=20ressucitar=20deps?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sapl/legacy/scripts/ressucita_dependencias.py | 24 ++++++++++++++----- 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/sapl/legacy/scripts/ressucita_dependencias.py b/sapl/legacy/scripts/ressucita_dependencias.py index cd3f1a9b4..3fa9decd7 100644 --- a/sapl/legacy/scripts/ressucita_dependencias.py +++ b/sapl/legacy/scripts/ressucita_dependencias.py @@ -1,12 +1,13 @@ from textwrap import dedent -import texttable import yaml -from unipath import Path -from sapl.legacy.migracao_dados import (DIR_REPO, PROPAGACOES_DE_EXCLUSAO, - exec_legado, +import texttable +from sapl.legacy.migracao_dados import (PROPAGACOES_DE_EXCLUSAO, exec_legado, get_arquivo_ajustes_pre_migracao) +from sapl.legacy_migration_settings import (DIR_DADOS_MIGRACAO, DIR_REPO, + NOME_BANCO_LEGADO) +from unipath import Path def stripsplit(ll): @@ -49,6 +50,9 @@ fks_legado = ''' materia_legislativa tip_id_basica tipo_materia_legislativa despacho_inicial cod_comissao comissao relatoria cod_comissao comissao + autor cod_comissao comissao + composicao_mesa cod_cargo cargo_mesa + comissao tip_comissao tipo_comissao ''' fks_legado = stripsplit(fks_legado) fks_legado = {(o, c): t for (o, c, t) in fks_legado} @@ -346,8 +350,16 @@ def get_ressucitar(slug): return get_sqls_desexcluir_criar(preambulo, desexcluir, criar, slug) -def adiciona_ressucitar(slug): +def get_slug(): + arq = DIR_DADOS_MIGRACAO.child('siglas_para_slugs.yaml') + with open(arq, 'r') as arq: + siglas_para_slugs = yaml.load(arq) + sigla = NOME_BANCO_LEGADO[-3:] + return siglas_para_slugs[sigla] + + +def adiciona_ressucitar(): arq_ajustes_pre_migracao = get_arquivo_ajustes_pre_migracao() conteudo = arq_ajustes_pre_migracao.read_file() - sqls = get_ressucitar(slug) + sqls = get_ressucitar(get_slug()) arq_ajustes_pre_migracao.write_file('{}\n{}'.format(conteudo, sqls))