bugfix

prefeitura-rio · Sep 10, 2024 · 39f9a36 · 39f9a36
1 parent f7a5542
commit 39f9a36
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pipelines/meteorologia/radar/mendanha/flows.py b/pipelines/meteorologia/radar/mendanha/flows.py
@@ -155,6 +155,7 @@
         formatted_time, saved_with_background_img_path
     )
     upload_file_to_storage(
+        project="datario",
         bucket_name="datario-public",
         destination_blob_name=destination_blob_name,
         source_file_name=source_file_name,

diff --git a/pipelines/meteorologia/radar/mendanha/tasks.py b/pipelines/meteorologia/radar/mendanha/tasks.py
@@ -539,11 +539,13 @@ def create_visualization_with_background(radar_2d, radar_product: str, cbar_titl
 
 
 @task
-def upload_file_to_storage(bucket_name: str, destination_blob_name: str, source_file_name: str):
+def upload_file_to_storage(
+    project: str, bucket_name: str, destination_blob_name: str, source_file_name: str
+):
     """
     Upload files to GCS
     """
-    storage_client = storage.Client(project="datario-public")
+    storage_client = storage.Client(project=project)
     bucket = storage_client.bucket(bucket_name)
     # Cria um blob (o arquivo dentro do bucket)
     blob = bucket.blob(destination_blob_name)