Merge pull request #60 from ImpulsoGov/refatoracao_sisab_cadastros

Correções do ETL SISAB Cadastros Individuais

src/impulsoetl/scripts/impulso_previne.py

@@ -11,30 +11,29 @@
 
 from impulsoetl import __VERSION__
 from impulsoetl.bd import Sessao, tabelas
-from impulsoetl.loggers import habilitar_suporte_loguru, logger
-from impulsoetl.egestor.relatorio_financiamento.principal import (
-    obter_relatorio_financiamento,
-)
-"""
 from impulsoetl.egestor.relatorio_financiamento.principal import (
     obter_relatorio_financiamento,
 )
+from impulsoetl.loggers import habilitar_suporte_loguru, logger
 from impulsoetl.sisab.cadastros_individuais import obter_cadastros_individuais
 from impulsoetl.sisab.indicadores_municipios.principal import (
     obter_indicadores_desempenho,
 )
 from impulsoetl.sisab.parametros_cadastro.principal import obter_parametros
+from impulsoetl.sisab.relatorio_saude_producao.principal import (
+    obter_relatorio_producao_por_profissionais_reduzidos,
+)
+from impulsoetl.sisab.relatorio_saude_producao.principal_outros import (
+    obter_relatorio_producao_por_profissionais_outros,
+)
 from impulsoetl.sisab.relatorio_validacao_producao.principal import (
     obter_validacao_producao,
 )
-"""
-from impulsoetl.sisab.relatorio_saude_producao.principal import obter_relatorio_producao_por_profissionais_reduzidos
-from impulsoetl.sisab.relatorio_saude_producao.principal_outros import obter_relatorio_producao_por_profissionais_outros
 
 agendamentos = tabelas["configuracoes.capturas_agendamentos"]
 capturas_historico = tabelas["configuracoes.capturas_historico"]
 
-"""
+
 @flow(
     name="Rodar Agendamentos de Cadastros das Equipes Válidas",
     description=(
@@ -65,12 +64,13 @@ def cadastros_municipios_equipe_validas(
         )
 
         for agendamento in agendamentos_cadastros:
-            periodo = agendamento.periodo_data_inicio
             obter_cadastros_individuais(
                 sessao=sessao,
                 visao_equipe=visao_equipe,
-                periodo=periodo,
-                teste=teste,
+                periodo_data=agendamento.periodo_data_inicio,
+                periodo_id=agendamento.periodo_id,
+                periodo_codigo=agendamento.periodo_codigo,
+                tabela_destino=agendamento.tabela_destino,
             )
             if teste:
                 sessao.rollback()
@@ -125,12 +125,13 @@ def cadastros_municipios_equipe_homologada(
         )
 
         for agendamento in agendamentos_cadastros:
-            periodo = agendamento.periodo_data_inicio
             obter_cadastros_individuais(
                 sessao=sessao,
                 visao_equipe=visao_equipe,
-                periodo=periodo,
-                teste=teste,
+                periodo_data=agendamento.periodo_data_inicio,
+                periodo_id=agendamento.periodo_id,
+                periodo_codigo=agendamento.periodo_codigo,
+                tabela_destino=agendamento.tabela_destino,
             )
             if teste:
                 sessao.rollback()
@@ -185,12 +186,13 @@ def cadastros_municipios_equipe_todas(
         )
 
         for agendamento in agendamentos_cadastros:
-            periodo = agendamento.periodo_data_inicio
             obter_cadastros_individuais(
                 sessao=sessao,
                 visao_equipe=visao_equipe,
-                periodo=periodo,
-                teste=teste,
+                periodo_data=agendamento.periodo_data_inicio,
+                periodo_id=agendamento.periodo_id,
+                periodo_codigo=agendamento.periodo_codigo,
+                tabela_destino=agendamento.tabela_destino,
             )
             if teste:
                 sessao.rollback()
@@ -726,6 +728,8 @@ def validacao_producao(
             conector.execute(requisicao_inserir_historico)
             sessao.commit()
             logger.info("OK.")
+
+
 @flow(
     name=("Rodar Agendamentos de Relatórios de Financiamento"),
     description=(
@@ -798,9 +802,12 @@ def egestor_financiamento(
                 conector.execute(requisicao_inserir_historico)
                 sessao.commit()
                 logger.info("OK.")
-"""
+
+
 @flow(
-    name=("Rodar Agendamentos do Relatório de Produção do SISAB - Profissionais Reduzidos"),
+    name=(
+        "Rodar Agendamentos do Relatório de Produção do SISAB - Profissionais Reduzidos"
+    ),
     description=(
         "Lê as capturas agendadas para obter o Relatório de Produção de Saúde do SISAB "
         + "para todos os municípios, filtrados por Tipo de Equipe, Categoria Profissional,"
@@ -812,7 +819,6 @@ def egestor_financiamento(
     version=__VERSION__,
     validate_parameters=False,
 )
-
 def relatorio_producao_saude_profissionais_reduzidos(
     teste: bool = True,
 ) -> None:
@@ -835,10 +841,10 @@ def relatorio_producao_saude_profissionais_reduzidos(
                 tabela_destino=agendamento.tabela_destino,
                 periodo_competencia=agendamento.periodo_data_inicio,
                 periodo_id=agendamento.periodo_id,
-                unidade_geografica_id=agendamento.unidade_geografica_id
+                unidade_geografica_id=agendamento.unidade_geografica_id,
             )
 
-            if teste:  
+            if teste:
                 sessao.rollback()
                 break
 
@@ -853,15 +859,14 @@ def relatorio_producao_saude_profissionais_reduzidos(
             conector = sessao.connection()
             conector.execute(requisicao_inserir_historico)
             sessao.commit()
-
-            logger.info("OK.")
 
+            logger.info("OK.")
 
-#relatorio_producao_saude_profissionais_reduzidos()
 
-"""
 @flow(
-    name=("Rodar Agendamentos do Relatório de Produção do SISAB - Profissionais Outros"),
+    name=(
+        "Rodar Agendamentos do Relatório de Produção do SISAB - Profissionais Outros"
+    ),
     description=(
         "Lê as capturas agendadas para obter o Relatório de Produção de Saúde do SISAB "
         + "para todos os municípios, filtrados por Tipo de Equipe, Categoria Profissional,"
@@ -873,7 +878,6 @@ def relatorio_producao_saude_profissionais_reduzidos(
     version=__VERSION__,
     validate_parameters=False,
 )
-
 def relatorio_producao_saude_profissionais_outros(
     teste: bool = True,
 ) -> None:
@@ -896,10 +900,10 @@ def relatorio_producao_saude_profissionais_outros(
                 tabela_destino=agendamento.tabela_destino,
                 periodo_competencia=agendamento.periodo_data_inicio,
                 periodo_id=agendamento.periodo_id,
-                unidade_geografica_id=agendamento.unidade_geografica_id
+                unidade_geografica_id=agendamento.unidade_geografica_id,
             )
 
-            if teste:  
+            if teste:
                 sessao.rollback()
                 break
 
@@ -914,7 +918,5 @@ def relatorio_producao_saude_profissionais_outros(
             conector = sessao.connection()
             conector.execute(requisicao_inserir_historico)
             sessao.commit()
-            
-            logger.info("OK.")
 
-relatorio_producao_saude_profissionais_outros()"""
+            logger.info("OK.")