Adds IQR calculation and labels based on these values.

prefeitura-rio · Jan 16, 2025 · 42adcba · 42adcba
1 parent d74bcc0
commit 42adcba
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 2 deletions.
diff --git a/models/marts/subgeral/monitora_reg/_mart_monitora_reg_schema.yml b/models/marts/subgeral/monitora_reg/_mart_monitora_reg_schema.yml
@@ -195,4 +195,7 @@ models:
           - not_null:
               name: mart_monitora_reg__monitor_vagas_ambulatoriais__cnes_dados__not_null
           - dbt_utils.not_constant:
-              name: mart_monitora_reg__monitor_vagas_ambulatoriais__cnes_dados__not_constant
+              name: mart_monitora_reg__monitor_vagas_ambulatoriais__cnes_dados__not_constant
+
+      - name: oferta_status
+        description: "Classifica o balanço de oferta como Adequado, Alto, Baixo ou Muito Baixo utilizando o Intervalo Interquartil (IQR), com base em ano_competencia, mes_competencia e id_procedimento."
diff --git a/models/marts/subgeral/monitora_reg/mart_monitora_reg__monitor_vagas_ambulatoriais.sql b/models/marts/subgeral/monitora_reg/mart_monitora_reg__monitor_vagas_ambulatoriais.sql
@@ -167,7 +167,42 @@ with
             = safe_cast(prof.mes_competencia as int64)
             and safe_cast(ofer.id_cbo_2002 as int64)
             = safe_cast(prof.id_cbo_2002 as int64)
+    ),
+
+    iqr as (
+        select
+            ano_competencia,
+            mes_competencia,
+            id_procedimento,
+            approx_quantiles(vagas_diferenca_ofertado_esperado, 4)[offset(1)] as q1,
+            approx_quantiles(vagas_diferenca_ofertado_esperado, 4)[offset(3)] as q3
+        from mva
+        group by ano_competencia, mes_competencia, id_procedimento
+    ),
+
+    mva_with_status as (
+        select
+            mva.*,
+
+            case
+                when mva.vagas_diferenca_ofertado_esperado is null
+                then 'N/A'
+                when mva.vagas_diferenca_ofertado_esperado < q1 - 1.5 * (q3 - q1)
+                then 'MUITO BAIXA'
+                when mva.vagas_diferenca_ofertado_esperado < q1
+                then 'BAIXA'
+                when mva.vagas_diferenca_ofertado_esperado <= q3
+                then 'ADEQUADA'
+                else 'ALTA'
+            end as status_oferta
+
+        from mva
+        left join
+            iqr
+            on mva.ano_competencia = iqr.ano_competencia
+            and mva.mes_competencia = iqr.mes_competencia
+            and mva.id_procedimento = iqr.id_procedimento
     )
 
 select *
-from mva
+from mva_with_status