commit inicial

prefeitura-rio · Jun 17, 2024 · 6f8006f · 6f8006f
1 parent 33fd24b
commit 6f8006f
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/pipelines/rj_smtr/constants.py b/pipelines/rj_smtr/constants.py
@@ -168,6 +168,32 @@ class constants(Enum):  # pylint: disable=c0103
     SUBSIDIO_SPPO_DASHBOARD_STAGING_DATASET_ID = "dashboard_subsidio_sppo_staging"
     SUBSIDIO_SPPO_DASHBOARD_TABLE_ID = "sumario_servico_dia"
     SUBSIDIO_SPPO_DATA_CHECKS_PARAMS = {
+        "check_trips_processing": {
+            "query": """SELECT
+                        s.*,
+                        i.feed_start_date AS feed_start_date_valido,
+                        FROM (
+                            SELECT
+                                *
+                            FROM
+                                rj-smtr.projeto_subsidio_sppo.subsidio_data_versao_efetiva
+                            WHERE
+                                DATA >= "2024-04-01" -- DATA_SUBSIDIO_V6_INICIO (Feature trajetos alternativos)
+                                AND DATA BETWEEN DATE("{start_timestamp}")
+                                    AND DATE("{end_timestamp}")
+                                ) AS s
+                        LEFT JOIN
+                            rj-smtr.gtfs.feed_info AS i
+                        ON
+                        (DATA BETWEEN i.feed_start_date
+                            AND i.feed_end_date
+                            OR (DATA >= i.feed_start_date
+                            AND i.feed_end_date IS NULL))
+                        WHERE
+                            i.feed_start_date != s.feed_start_date
+                        """,
+            "order_columns": ["data"],
+        },
         "check_gps_capture": {
             "query": """WITH
             t AS (
@@ -565,6 +591,9 @@ class constants(Enum):  # pylint: disable=c0103
             "Todos os dados de status dos veículos foram devidamente tratados": {
                 "test": "check_sppo_veiculo_dia",
             },
+            "Todos os dados de viagens foram tratados com o respectivo feed correto": {
+                "test": "check_trips_processing",
+            },
         }
     }
     SUBSIDIO_SPPO_DATA_CHECKS_POS_LIST = {