Ajusta localização dos logs da função de extração do ETL relatorio_sa…

…ude_producao
ImpulsoGov · Jul 26, 2023 · 8fe22e5 · 8fe22e5
1 parent 632d6b2
commit 8fe22e5
Showing 1 changed file with 28 additions and 24 deletions.
diff --git a/src/impulsoetl/sisab/relatorio_saude_producao/extracao.py b/src/impulsoetl/sisab/relatorio_saude_producao/extracao.py
@@ -9,41 +9,48 @@
 from prefect import task
 
 from impulsoetl.loggers import logger, habilitar_suporte_loguru
-from impulsoetl.sisab.utilitarios_sisab_relatorio_producao import extrair_producao_por_municipio
-from impulsoetl.sisab.utilitarios_sisab_relatorio_producao import transformar_producao_por_municipio
+from impulsoetl.sisab.utilitarios_sisab_relatorio_producao import (
+    extrair_producao_por_municipio,
+)
+from impulsoetl.sisab.utilitarios_sisab_relatorio_producao import (
+    transformar_producao_por_municipio,
+)
 
 
 CATEGORIA_PROFISSIONAL_REDUZIDA = [
-    'Cirurgião dentista',
-    'Enfermeiro',
-    'Fisioterapeuta',
-    'Médico',
-    'Psicólogo',
-    'Técnico e auxiliar de enfermagem',
-    'Técnico e auxiliar de saúde bucal',
-    ]
-
-def obter_relatorio_reduzido(
-    periodo_competencia: date)-> pd.DataFrame():
+    "Cirurgião dentista",
+    "Enfermeiro",
+    "Fisioterapeuta",
+    "Médico",
+    "Psicólogo",
+    "Técnico e auxiliar de enfermagem",
+    "Técnico e auxiliar de saúde bucal",
+]
+
+
+def obter_relatorio_reduzido(periodo_competencia: date) -> pd.DataFrame():
 
     df_consolidado = pd.DataFrame()
-
+
+    logger.info("Iniciando extraçção do relatório...")
+
     try:
         df_parcial = extrair_producao_por_municipio(
             tipo_producao="Atendimento individual",
             competencias=[periodo_competencia],
             selecoes_adicionais={
-                "Problema/Condição Avaliada": "Selecionar Todos", 
-                "Conduta":"Selecionar Todos",
-                "Categoria do Profissional":CATEGORIA_PROFISSIONAL_REDUZIDA, 
+                "Problema/Condição Avaliada": "Selecionar Todos",
+                "Conduta": "Selecionar Todos",
+                "Categoria do Profissional": CATEGORIA_PROFISSIONAL_REDUZIDA,
             },
+        ).pipe(transformar_producao_por_municipio)
 
-            ).pipe(transformar_producao_por_municipio)
-
         print(df_parcial)
 
         df_consolidado = df_consolidado.append(df_parcial)
 
+        logger.info("Extração concluída")
+
     except Exception as e:
         logger.error(e)
         pass
@@ -60,14 +67,11 @@ def obter_relatorio_reduzido(
     retries=2,
     retry_delay_seconds=120,
 )
-def extrair_relatorio(
-    periodo_competencia: date)-> pd.DataFrame():
+
+def extrair_relatorio(periodo_competencia: date) -> pd.DataFrame():
 
     habilitar_suporte_loguru()
-    logger.info("Iniciando extraçção do relatório...")
 
     df_extraido = obter_relatorio_reduzido(periodo_competencia)
-
-    logger.info("Extração concluída")
 
     return df_extraido