From 0b5baa9e77df0596ca384db366011a71bdb62c70 Mon Sep 17 00:00:00 2001 From: Marcio Mazza Date: Fri, 16 Apr 2021 15:36:10 +0200 Subject: [PATCH] Solve yaml.load warning --- sapl/legacy/migracao_dados.py | 6 +++--- sapl/legacy/scripts/ressuscita_dependencias.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/sapl/legacy/migracao_dados.py b/sapl/legacy/migracao_dados.py index fe1db34f3..8651078e5 100644 --- a/sapl/legacy/migracao_dados.py +++ b/sapl/legacy/migracao_dados.py @@ -127,7 +127,7 @@ def get_renames(): model_renames = {} includes = {} for app in appconfs: - app_rename_data = yaml.load( + app_rename_data = yaml.safe_load( pkg_resources.resource_string(app.module.__name__, "legacy.yaml") ) for model_name, renames in app_rename_data.items(): @@ -1118,7 +1118,7 @@ def migrar_dados(primeira_migracao=False, apagar_do_legado=False): if arq_ajustes_sql.exists(): exec_legado(arq_ajustes_sql.read_file()) if arq_ajustes_reverter.exists(): - revert_delete_producao(yaml.load(arq_ajustes_reverter.read_file())) + revert_delete_producao(yaml.safe_load(arq_ajustes_reverter.read_file())) uniformiza_banco(primeira_migracao) @@ -1504,7 +1504,7 @@ def get_arquivo_resolve_registro_votacao(): def get_como_resolver_registro_votacao_ambiguo(): path = get_arquivo_resolve_registro_votacao() if path.exists(): - return yaml.load(path.read_file()) + return yaml.safe_load(path.read_file()) else: return {} diff --git a/sapl/legacy/scripts/ressuscita_dependencias.py b/sapl/legacy/scripts/ressuscita_dependencias.py index 6befefe79..a652d3dcb 100644 --- a/sapl/legacy/scripts/ressuscita_dependencias.py +++ b/sapl/legacy/scripts/ressuscita_dependencias.py @@ -230,7 +230,7 @@ Para facilitar sua conferência, seguem os links para as proposições envolvida def get_fks_faltando(): - ocorrencias = yaml.load( + ocorrencias = yaml.safe_load( Path(DIR_REPO.child("ocorrencias.yaml").read_file()) ) return ocorrencias.get("fk", []) @@ -515,7 +515,7 @@ def get_ressuscitar(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) + siglas_para_slugs = yaml.safe_load(arq) sigla = NOME_BANCO_LEGADO[-3:] return siglas_para_slugs[sigla]