csc-cia-stne 0.1.14__py3-none-any.whl → 0.1.16__py3-none-any.whl

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/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.dist-info → csc_cia_stne-0.1.16.dist-info}/METADATA

@@ -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.dist-info → csc_cia_stne-0.1.16.dist-info}/RECORD

@@ -4,7 +4,7 @@ csc_cia_stne/bc_sta.py,sha256=sE-aU-ZVSAqryQrT1-nor9eAFM5npNAKF1QSm-ChhGU,28945
 csc_cia_stne/email.py,sha256=y4xyPAe6_Mga5Wf6qAsDzYgn0f-zf2KshfItlWe58z8,8481
 csc_cia_stne/ftp.py,sha256=M9WCaq2hm56jGyszNaPinliFaZS0BNrT7VrVPMjkMg4,10988
 csc_cia_stne/gcp_bigquery.py,sha256=foq8azvvv_f7uikMDslX9RcUIrx7RAS-Sn0AGW0QFQc,7231
-csc_cia_stne/gcp_bucket.py,sha256=_Q3LS3Uy9tyi6LFHBedu38S1GS2CrI5iJCxRLu0rNaU,11007
+csc_cia_stne/gcp_bucket.py,sha256=vMALWiW7IoBCuJAR8bUCpOV6BuBzI9AhRRk3b72OdMk,11515
 csc_cia_stne/google_drive.py,sha256=j1fvHpgU76_VqCtSEe05JXLOgNwNNfrP6GRKq6eU4b4,17074
 csc_cia_stne/karavela.py,sha256=jJCYX43D49gGuzmwwK6bN9XVnv2dXdp9iHnnV5H1LMQ,4794
 csc_cia_stne/logger_json.py,sha256=CXxSCOFGMymDi8XE9SKnPKjW4D0wJLqDLnxqePS26i8,3187
@@ -31,15 +31,15 @@ csc_cia_stne/utilitarios/validations/GoogleDriveValidator.py,sha256=Q1oZTae4hDJ2
 csc_cia_stne/utilitarios/validations/ServiceNowValidator.py,sha256=yleKUIo1ZfyloP9fDPNjv3JJXdLcocT81WIgRSYmqEA,14423
 csc_cia_stne/utilitarios/validations/__init__.py,sha256=O_qyEU2ji3u6LHUXZCXvUFsMpoMWL625qqHTXyXivTA,106
 csc_cia_stne/utilitarios/validations/ftp.py,sha256=LNYyS2c8eou3ML23bBtn87xJFIKSrb8w-2acG0Zugog,4988
-csc_cia_stne/utilitarios/validations/gcp_bucket.py,sha256=E94UCoph9FPo0y2y6UnGOO-zqJFfaHqRAXHWXee7FtM,6229
+csc_cia_stne/utilitarios/validations/gcp_bucket.py,sha256=aDwytjpur6vDKGDMzrqtzDi4bclQ-ahiXuURhmDYZpk,6979
 csc_cia_stne/utilitarios/validations/waccess.py,sha256=8yfOrmIPUSLzbCt6P0F6vj3FkSgU_RgrJpJlRhlLyV0,7352
 csc_cia_stne/utilitarios/validations/web_validator.py,sha256=HYKYSpDv1RvRjZIuwTPt-AbEz-9392MxM_O329iYuSA,5722
 csc_cia_stne/utilitarios/web_screen/__init__.py,sha256=5QcOPXKd95SvP2DoZiHS0gaU68GlyFD9pQ9kae_9D1Q,100
 csc_cia_stne/utilitarios/web_screen/web_screen_abstract.py,sha256=PjL8Vgfj_JdKidia7RFyCkro3avYLQu4RZRos41sh3w,3241
 csc_cia_stne/utilitarios/web_screen/web_screen_botcity.py,sha256=Xi5YJjl2pcxlX3OimqcBWRNXZEpAE7asyUjDJ4Oho5U,12297
 csc_cia_stne/utilitarios/web_screen/web_screen_selenium.py,sha256=JLIcPJE9ZX3Pd6zG6oTRMqqUAY063UzLY3ReRlxmiSM,15581
-csc_cia_stne-0.1.14.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
-csc_cia_stne-0.1.14.dist-info/METADATA,sha256=Pb4ndmVY2lwJTOZG-E2TyvZemsu4RwDTsQQrPhCPaPo,1464
-csc_cia_stne-0.1.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-csc_cia_stne-0.1.14.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
-csc_cia_stne-0.1.14.dist-info/RECORD,,
+csc_cia_stne-0.1.16.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.1.16.dist-info/METADATA,sha256=Namt43MLWsc_Ebf_VhYyrjRLBkUVc8XGDi7HsYzYfJ8,1464
+csc_cia_stne-0.1.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+csc_cia_stne-0.1.16.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.1.16.dist-info/RECORD,,