csc-cia-stne 0.1.15__tar.gz → 0.1.17__tar.gz

{csc_cia_stne-0.1.15 → csc_cia_stne-0.1.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.15
+Version: 0.1.17
 Summary: Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs
 License: MIT
 Keywords: karavela,csc,cia,stone,rpa,botcity,stne

{csc_cia_stne-0.1.15 → csc_cia_stne-0.1.17}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csc_cia_stne"
-version = "0.1.15"
+version = "0.1.17"
 license = { text = "MIT" }
 description = "Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs"
 keywords = ["karavela", "csc", "cia", "stone", "rpa", "botcity", "stne"]

{csc_cia_stne-0.1.15 → csc_cia_stne-0.1.17}/src/csc_cia_stne/gcp_bucket.py

@@ -1,7 +1,7 @@
+import base64
 from google.oauth2 import service_account
 from google.cloud import storage
 from pydantic import ValidationError
-
 from .utilitarios.validations.gcp_bucket import InitParamsValidator,ListFilesValidator,GetFilesValidator,UploadFilesValidator,DeleteFilesValidator
 
 
@@ -112,7 +112,7 @@ class GCPBucket():
             }
 
     # Função para baixar os arquivos de um bucket
-    def get_file(self, bucket_name:str, filename:str, destination:str=None, chunksize:int=256) -> dict:
+    def get_file(self, bucket_name:str, filename:str, destination:str=None, chunksize:int=256, download_as:str='file') -> dict:
         """
         Faz o download de um arquivo de um bucket do Google Cloud Storage para um destino local.
         Args:
@@ -122,6 +122,7 @@ class GCPBucket():
                 Se não for especificado, o arquivo será salvo com o mesmo nome do arquivo no bucket.
             chunksize (int, opcional): Tamanho do chunk em megabytes para o download do arquivo. 
                 O padrão é 256 MB.
+            download_as (str, opcional): Define o formato do download.
         Returns:
             dict: Um dicionário contendo o status da operação. 
                 - 'success' (bool): Indica se a operação foi bem-sucedida.
@@ -133,7 +134,7 @@ class GCPBucket():
 
         try:
 
-            GetFilesValidator(bucket_name=bucket_name, filename=filename, destination=destination, chunksize=chunksize)
+            GetFilesValidator(bucket_name=bucket_name, filename=filename, destination=destination, chunksize=chunksize, download_as=download_as)
 
         except ValidationError as e:
         
@@ -144,7 +145,7 @@ class GCPBucket():
             destination = filename
 
         try:
-
+            
             bucket = self.client.bucket(bucket_name)
             
         except Exception as e:
@@ -161,12 +162,20 @@ class GCPBucket():
 
             blob.chunk_size = chunksize * 1024 * 1024  # 256M
 
-            blob.download_to_filename(destination)
+            if download_as == 'file':
+                blob.download_to_filename(destination)
+                return {
+                    'success':True,
+                    'details':f'Arquivo baixado para o diretório: {str(destination)}'
+                }
 
-            return {
-                'success':True,
-                'details':f'Arquivo baixado para o diretório: {str(destination)}'
-            }
+            elif download_as == 'bytes':
+                b64_str = base64.b64encode(blob.download_as_bytes()).decode()
+                return {
+                    'success':True,
+                    'details':f'Arquivo baixado como bytes em base64',
+                    'data': b64_str
+                }
 
         except Exception as e: