[hotfix] processa_ordem_servico (#207)

* commit inicial * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Insere tipo_os * remove formato dos arquivos * remove quebra de linha nos nomes das colunas do anexo da faixa horária * Incorpora io.BytesIO * remove log * Refatora tratamento tipo_os * refatora tratamento tipo_os e anexo da faixa horaria * remove log * fix processa_ordem_servico * Adiciona changelogs e altera run_config do flow gtfs_captura_nova * ajusta processa_ordem_servico --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Guilherme Botelho <[email protected]>
prefeitura-rio · Sep 11, 2024 · 0c45750 · 0c45750
1 parent ca41313
commit 0c45750
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pipelines/migration/br_rj_riodejaneiro_gtfs/utils.py b/pipelines/migration/br_rj_riodejaneiro_gtfs/utils.py
@@ -250,6 +250,9 @@ def processa_ordem_servico(
         ""
     ) + quadro["servico"].str.extract(r"([0-9]+)", expand=False).fillna("")
 
+    if "tipo_os" not in quadro.columns:
+        quadro["tipo_os"] = "Regular"
+
     quadro = quadro[list(set(columns.values()))]
     quadro = quadro.replace("—", 0)
     quadro = quadro.reindex(columns=list(set(columns.values())))
@@ -278,9 +281,6 @@ def processa_ordem_servico(
     quadro["extensao_ida"] = quadro["extensao_ida"] / 1000
     quadro["extensao_volta"] = quadro["extensao_volta"] / 1000
 
-    if "tipo_os" not in quadro.columns:
-        quadro["tipo_os"] = "Regular"
-
     quadro_geral = pd.concat([quadro_geral, quadro])
 
     # Verificações