Corrige argumentos e chamada de funções

src/impulsoetl/scripts/geral.py

@@ -14,10 +14,10 @@
 from impulsoetl.bd import tabelas, Sessao
 
 
-#from impulsoetl.brasilapi.cep import obter_cep
-#from impulsoetl.scnes.habilitacoes import obter_habilitacoes
-#from impulsoetl.scnes.vinculos import obter_vinculos
-#from impulsoetl.sim.do import obter_do 
+from impulsoetl.brasilapi.cep import obter_cep
+from impulsoetl.scnes.habilitacoes import obter_habilitacoes
+from impulsoetl.scnes.vinculos import obter_vinculos
+from impulsoetl.sim.do import obter_do 
 from impulsoetl.loggers import habilitar_suporte_loguru, logger
 from impulsoetl.scnes.estabelecimentos_identificados.principal import obter_informacoes_estabelecimentos_identificados
 from impulsoetl.scnes.estabelecimentos_horarios.principal import obter_horarios_estabelecimentos
@@ -29,7 +29,6 @@
 agendamentos = tabelas["configuracoes.capturas_agendamentos"]
 capturas_historico = tabelas["configuracoes.capturas_historico"]
 
-"""
 @flow(
     name="Rodar Agendamentos de Habilitações do SCNES",
     description=(
@@ -247,7 +246,7 @@ def ceps(teste: bool = False) -> None:
     version=__VERSION__,
     validate_parameters=False,
 )
-"""
+
 def cnes_estabelecimentos_identificados(teste: bool = False,)-> None:
 
     habilitar_suporte_loguru()

src/impulsoetl/scripts/impulso_previne.py

@@ -71,6 +71,7 @@ def cadastros_municipios_equipe_validas(
                 periodo_id=agendamento.periodo_id,
                 periodo_codigo=agendamento.periodo_codigo,
                 tabela_destino=agendamento.tabela_destino,
+                operacao_id=operacao_id,
             )
             if teste:
                 sessao.rollback()
@@ -132,6 +133,7 @@ def cadastros_municipios_equipe_homologada(
                 periodo_id=agendamento.periodo_id,
                 periodo_codigo=agendamento.periodo_codigo,
                 tabela_destino=agendamento.tabela_destino,
+                operacao_id=operacao_id,
             )
             if teste:
                 sessao.rollback()
@@ -193,6 +195,7 @@ def cadastros_municipios_equipe_todas(
                 periodo_id=agendamento.periodo_id,
                 periodo_codigo=agendamento.periodo_codigo,
                 tabela_destino=agendamento.tabela_destino,
+                operacao_id=operacao_id,
             )
             if teste:
                 sessao.rollback()
@@ -513,12 +516,14 @@ def indicadores_municipios_equipe_validas(
         )
 
         for agendamento in agendamentos_cadastros:
-            periodo = agendamento.periodo_data_inicio
             obter_indicadores_desempenho(
                 sessao=sessao,
                 visao_equipe=visao_equipe,
-                quadrimestre=periodo,
-                teste=teste,
+                periodo_data=agendamento.periodo_data_fim,
+                periodo_id=agendamento.periodo_id,
+                periodo_codigo=agendamento.periodo_codigo,
+                operacao_id=operacao_id,
+                tabela_destino=agendamento.tabela_destino,
             )
             if teste:
                 sessao.rollback()
@@ -576,12 +581,14 @@ def indicadores_municipios_equipes_homologadas(
         )
 
         for agendamento in agendamentos_cadastros:
-            periodo = agendamento.periodo_data_inicio
             obter_indicadores_desempenho(
                 sessao=sessao,
                 visao_equipe=visao_equipe,
-                quadrimestre=periodo,
-                teste=teste,
+                periodo_data=agendamento.periodo_data_fim,
+                periodo_id=agendamento.periodo_id,
+                periodo_codigo=agendamento.periodo_codigo,
+                operacao_id=operacao_id,
+                tabela_destino=agendamento.tabela_destino,
             )
             if teste:
                 sessao.rollback()
@@ -637,12 +644,14 @@ def indicadores_municipios_equipe_todas(
         )
 
         for agendamento in agendamentos_cadastros:
-            periodo = agendamento.periodo_data_inicio
             obter_indicadores_desempenho(
                 sessao=sessao,
                 visao_equipe=visao_equipe,
-                quadrimestre=periodo,
-                teste=teste,
+                periodo_data=agendamento.periodo_data_fim,
+                periodo_id=agendamento.periodo_id,
+                periodo_codigo=agendamento.periodo_codigo,
+                operacao_id=operacao_id,
+                tabela_destino=agendamento.tabela_destino,
             )
             if teste:
                 sessao.rollback()
@@ -706,6 +715,7 @@ def validacao_producao(
                 periodo_id=agendamento.periodo_id,
                 periodo_codigo=agendamento.periodo_codigo,
                 tabela_destino=agendamento.tabela_destino,
+                operacao_id=operacao_id,
             )
 
             if teste:  # evitar rodar muitas iterações
@@ -780,6 +790,7 @@ def egestor_financiamento(
                     periodo_id=agendamento.periodo_id,
                     tabela_destino=agendamento.tabela_destino,
                     periodo_mes=agendamento.periodo_data_inicio,
+                    operacao_id=operacao_id,
                 )
 
                 if teste:  # evitar rodar muitas iterações
@@ -920,3 +931,4 @@ def relatorio_producao_saude_profissionais_outros(
             sessao.commit()
 
             logger.info("OK.")
+

src/impulsoetl/scripts/saude_mental.py

@@ -13,11 +13,11 @@
 from impulsoetl import __VERSION__
 from impulsoetl.bd import Sessao, tabelas
 from impulsoetl.loggers import habilitar_suporte_loguru, logger
-#from impulsoetl.siasus.bpa_i import obter_bpa_i
-#from impulsoetl.siasus.procedimentos import obter_pa
-#from impulsoetl.siasus.raas_ps import obter_raas_ps
-#from impulsoetl.sihsus.aih_rd import obter_aih_rd
-#from impulsoetl.sinan.violencia import obter_agravos_violencia
+from impulsoetl.siasus.bpa_i import obter_bpa_i
+from impulsoetl.siasus.procedimentos import obter_pa
+from impulsoetl.siasus.raas_ps import obter_raas_ps
+from impulsoetl.sihsus.aih_rd import obter_aih_rd
+from impulsoetl.sinan.violencia import obter_agravos_violencia
 from impulsoetl.sisab.relatorio_producao_resolutividade_por_condicao.principal import obter_relatorio_resolutividade_por_condicao
 from impulsoetl.sisab.relatorio_tipo_equipe_por_tipo_producao.principal import obter_relatorio_tipo_equipe_por_producao
 
@@ -60,7 +60,6 @@ def resolutividade_aps_por_condicao(
         for agendamento in agendamentos_resolutividade_por_condicao:
             obter_relatorio_resolutividade_por_condicao(
                 sessao = sessao,
-                teste = teste,
                 tabela_destino = agendamento.tabela_destino,
                 periodo_id = agendamento.periodo_id,
                 unidade_geografica_id = agendamento.unidade_geografica_id,
@@ -99,7 +98,7 @@ def resolutividade_aps_por_condicao(
     validate_parameters=False,
 )
 def tipo_equipe_por_tipo_producao(
-    teste:bool = True,
+    teste:bool = False,
 ) -> None:
 
     """Número de contatos assistenciais na APS por tipo de produção e equipe.
@@ -132,7 +131,6 @@ def tipo_equipe_por_tipo_producao(
         for agendamento in agendamentos_producao_por_equipe:
             obter_relatorio_tipo_equipe_por_producao(
                 sessao = sessao,
-                teste = teste,
                 tabela_destino = agendamento.tabela_destino,
                 periodo_id = agendamento.periodo_id,
                 unidade_geografica_id = agendamento.unidade_geografica_id,
@@ -154,7 +152,7 @@ def tipo_equipe_por_tipo_producao(
                 break
             sessao.commit()
             logger.info("OK.")
-"""
+
 @flow(
     name="Rodar Agendamentos de Arquivos de Disseminação da RAAS-PS",
     description=(
@@ -459,5 +457,4 @@ def agravos_violencia(
             conector = sessao.connection()
             conector.execute(requisicao_inserir_historico)
             sessao.commit()
-            logger.info("OK.")
-"""
+            logger.info("OK.")