csc-cia-stne 0.1.14__tar.gz → 0.1.16__tar.gz

{csc_cia_stne-0.1.14 → csc_cia_stne-0.1.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.14
+Version: 0.1.16
 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.14 → csc_cia_stne-0.1.16}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csc_cia_stne"
-version = "0.1.14"
+version = "0.1.16"
 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.14 → csc_cia_stne-0.1.16}/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:
         

{csc_cia_stne-0.1.14 → csc_cia_stne-0.1.16}/src/csc_cia_stne/utilitarios/validations/gcp_bucket.py

@@ -1,7 +1,9 @@
 from __future__ import annotations
-from pydantic import BaseModel, field_validator, model_validator, ValidationError
-from typing import Optional, Mapping, Any
+from pydantic import BaseModel, field_validator, model_validator, ValidationError, Field, ConfigDict
+from typing import Optional, Mapping, Any, Annotated
 
+NonEmptyStr = Annotated[str, Field(min_length=1)]
+PositiveInt = Annotated[int, Field(gt=0)]
 
 class InitParamsValidator(BaseModel):
     """
@@ -22,10 +24,6 @@ class InitParamsValidator(BaseModel):
             )
         return self
 
-
-
-
-
 class ListFilesValidator(BaseModel):
     """
     Classe ListFilesValidator
@@ -47,6 +45,23 @@ class ListFilesValidator(BaseModel):
         return value
     
 class GetFilesValidator(BaseModel):
+
+    model_config = ConfigDict(strict=True)  # não faz coerção: "1024" não vira 1024
+
+    bucket_name: NonEmptyStr
+    filename: NonEmptyStr
+    destination: NonEmptyStr
+    chunksize: PositiveInt
+    download_as: NonEmptyStr = 'file'  # 'file' ou 'bytes'
+
+    # Opcional: se você quiser mensagens personalizadas além das do Field
+    @field_validator("bucket_name", "filename", "destination", "download_as")
+    def _not_blank(cls, v, info):
+        if not v.strip():
+            raise ValueError(f"O parametro '{info.field_name}' não pode ser vazio/whitespace.")
+        return v
+    
+class _GetFilesValidator(BaseModel):
     """
     Classe GetFilesValidator
     Valida os parâmetros necessários para operações relacionadas a arquivos em um bucket GCP.
@@ -131,6 +146,7 @@ class UploadFilesValidator(BaseModel):
                 f"O parametro '{info.field_name}' deve ser uma string não vazia; recebido {type(value)}"
             )
         return value.strip()    
+
 class DeleteFilesValidator(BaseModel):
     """
     Classe DeleteFilesValidator

{csc_cia_stne-0.1.14 → csc_cia_stne-0.1.16}/src/csc_cia_stne.egg-info/PKG-INFO

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