bugfix

prefeitura-rio · Sep 3, 2024 · 7941b1c · 7941b1c
1 parent 46240c4
commit 7941b1c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pipelines/meteorologia/radar/mendanha/tasks.py b/pipelines/meteorologia/radar/mendanha/tasks.py
@@ -28,7 +28,7 @@
     list_files_storage,
 )
 
-from pipelines.utils.utils import log  # , to_partitions
+from prefeitura_rio.pipelines_utils.logging import log
 from prefeitura_rio.pipelines_utils.gcs import get_gcs_client
 
 
@@ -57,7 +57,8 @@ def get_filenames_storage(
     # Listar e ordenar arquivos de cada volume
     volume_files = {}
 
-    sorted_files = list_files_storage(bucket, prefix="vol")
+    sorted_files = list_files_storage(bucket, prefix=vol_a)
+    log(f"{len(sorted_files)} files found in vol_a")
     volume_files[vol_a] = sorted_files
 
     # Identificar o último arquivo em vol_a
@@ -69,7 +70,7 @@ def get_filenames_storage(
     # data_partição e assim ter que ler menos nomes de arquivos
 
     for vol in volumes[1:]:
-        sorted_files = list_files_storage(bucket, prefix="vol")
+        sorted_files = list_files_storage(bucket, prefix=vol)
         volume_files[vol] = sorted_files
 
     # Encontrar os arquivos subsequentes em vol_b, vol_c e vol_d