csc-cia-stne 0.0.37__py3-none-any.whl → 0.0.39__py3-none-any.whl

csc_cia_stne/__init__.py

@@ -10,6 +10,8 @@ from .bc_correios import BC_Correios
 from .gcp_bigquery import BigQuery
 from .email import Email
 from .provio import Provio
+from .google_drive import GoogleDrive
+from .slack import Slack
 
 # Define os itens disponíveis para importação
 __all__ = [
@@ -22,7 +24,9 @@ __all__ = [
     "Util",
     "logger",
     "Provio",
-    "Email"
+    "Email", 
+    "GoogleDrive",
+    "Slack"
 ]
 
 _diretorio_inicial = os.getcwd()

csc_cia_stne/google_drive.py

@@ -1,16 +1,13 @@
-import os, time, logging, json
-from .karavela import Karavela
+import os, logging
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaFileUpload
 from googleapiclient.errors import HttpError
 from io import BytesIO
-from googleapiclient.discovery import build
 from google.oauth2 import service_account
-from .utilitarios.validations.GoogleDriveValidator import InitParamsValidator, CreateFolderValidator, ListFolderValidator
+from .utilitarios.validations.GoogleDriveValidator import InitParamsValidator, CreateFolderValidator, ListFolderValidator, UploadValidator
 from pydantic import ValidationError
 
 
-
 class GoogleDrive():
     
     """
@@ -20,8 +17,16 @@ class GoogleDrive():
         Args:
             key (str): Chave de autenticação para acessar a API do Google Drive.
     """
+    _instance = None  # Atributo de classe para armazenar a única instância.
+
+    def __new__(cls, *args, **kwargs):
+        # Verifica se já existe uma instância da classe.
+        if cls._instance is None:
+            # Cria e armazena a instância na primeira chamada.
+            cls._instance = super().__new__(cls)
+        return cls._instance
     
-    def __init__(self, key, with_subject : str, scopes : list = ["https://www.googleapis.com/auth/drive"], version : str = "v3"):
+    def __init__(self, token : dict, with_subject : str, scopes : list = ["https://www.googleapis.com/auth/drive"], version : str = "v3"):
         """
         Inicializa uma instância da classe GoogleDrive.
         Parâmetros:
@@ -36,43 +41,20 @@ class GoogleDrive():
         google_drive = GoogleDrive(key="chave_secreta", with_subject="assunto", scopes=["https://www.googleapis.com/auth/drive"], version="v3")
         ```
         """
-        # código de inicialização da instância
         
         try:
-            InitParamsValidator(key=key, with_subject=with_subject, scopes=scopes)
+            InitParamsValidator(token=token, with_subject=with_subject, scopes=scopes, version=version)
         except ValidationError as e:
             raise ValueError("Erro na validação dos dados de input da inicialização da instância:", e.errors())
-
-        self.__token = json.loads(Karavela.get_secret(key))
+        self.__token = token
         self.version = version
         self.scopes = scopes
-        self.__auth = self.__autentica(with_subject)
+        self.with_subject = with_subject
         self.page_size = 1000
-        self.service = self.create_service()
+        self.service = self.__create_service()
 
-    def timer_decorator(func):
-        """
-        Decorator que calcula o tempo de execução de uma função.
-        Args:
-            func: A função a ser decorada.
-        Returns:
-            A função decorada.
-        Exemplo de uso:
-            @timer_decorator
-            def minha_funcao():
-                # código da função
-            minha_funcao()
-        """
-        
-        def wrapper(*args, **kwargs):
-            inicio = time.time()
-            resultado = func(*args, **kwargs)
-            fim = time.time()
-            logging.debug(f"Tempo de execução de {func.__name__}: {fim - inicio:.2f} segundos")
-            return resultado
-        return wrapper
-    
-    def create_service(self):
+
+    def __create_service(self):
         """
         Cria um serviço do Google Drive.
         Returns:
@@ -89,14 +71,15 @@ class GoogleDrive():
         """
         
         try:
-            
-            cred = build(f"drive", f"{self.version}", credentials=self.__auth)
-            logging.debug(f"Serviço do Google drive criado")
-            return cred
+            auth = self.__autentica(self.with_subject)
+            service = build(f"drive", f"{self.version}", credentials=auth)
+            logging.info(f"Serviço do Google drive criado")
+            return service
         except Exception as e:
-            logging.debug(f"Erro ao criar o serviço do Google drive: Error: {e}")
+            logging.info(f"Erro ao criar o serviço do Google drive: Error: {e}")
             return False
-    
+
+
     def __autentica(self, with_subject : str):
         """
         Autentica o usuário com as credenciais fornecidas e retorna as credenciais delegadas para o assunto especificado.
@@ -121,7 +104,7 @@ class GoogleDrive():
             logging.debug(f"Erro ao tentar autenticar. Verifique o erro: {e}")
             return False
 
-    @timer_decorator
+
     def upload(self, folder_id: str, name: str, file_path: str, mimetype: str):
         """
             Faz o upload de um arquivo para o Google Drive em uma pasta especificada.
@@ -131,11 +114,27 @@ class GoogleDrive():
                 name (str): Nome do arquivo que será carregado.
                 file_path (str): Caminho completo do arquivo no sistema local.
                 mimetype (str): Tipo MIME do arquivo a ser carregado.
+                    exemplos:   text/plain
+                                text/html
+                                image/jpeg
+                                image/png
+                                audio/mpeg
+                                audio/ogg
+                                audio/*
+                                video/mp4
+                                application/octet-stream
 
             Returns:
                 dict: Informações sobre o arquivo carregado, incluindo o ID do arquivo.
                 None: Caso o caminho do arquivo não seja encontrado.
         """
+        try:
+            UploadValidator(folder_id=folder_id, name=name, file_path=file_path, mimetype=mimetype)
+        except ValidationError as e:
+            raise ValueError("Erro na validação dos dados para realizar o upload do arquivo", e.errors())
+        
+
+        
         file_metadata = {"name": name, "parents": [folder_id]}
         if not os.path.exists(file_path):
             logging.debug(f"Pasta {file_path} não encontrada")           
@@ -156,6 +155,7 @@ class GoogleDrive():
             logging.debug(f"Erro ao realizar o upload do arquivo {name} no google drive. Erro: {e}")
             return {"success" : False, "result" : None, "error" : str(e)}
    
+
     def create_folder(self, name: str, parent_folder_id: str):
         """
             Cria uma pasta no Google Drive dentro de uma pasta existente.
@@ -185,8 +185,9 @@ class GoogleDrive():
         except Exception as e:
             logging.debug(f"Não foi possível criar a pasta {name}")
             return {"success" : False, "result": None, "error" : str(e)}
-    
-    def list(self, query: str = "", spaces: str = "drive", fields: str = "nextPageToken, files(id, name)"):
+
+
+    def list_items_folder(self, query: str = "", spaces: str = "drive", fields: str = "nextPageToken, files(id, name)"):
         """
             Lista os arquivos e pastas no Google Drive com base nos critérios fornecidos.
 
@@ -194,6 +195,7 @@ class GoogleDrive():
                 query (str, optional): Critério de busca para os arquivos ou pastas no Google Drive. 
                                     Consulte https://developers.google.com/drive/api/v3/ref-search-terms.
                                     Defaults to "".
+                                    exemplo: query = 'ID_DA_PASTA_NO_GOOGLE_DRIVE' in parents and trashed=false"
                 spaces (str, optional): Especifica os locais de armazenamento a serem consultados. Pode ser 'drive', 
                                         'appDataFolder', ou 'photos'. Defaults to 'drive'.
                 fields (str, optional): Campos a serem retornados na resposta. Consulte a documentação para os campos disponíveis.
@@ -221,8 +223,10 @@ class GoogleDrive():
         except HttpError as hr:
             logging.debug(f"Ocorreu um de http. Erro: {hr}")
             return {"success" : False, "result" : None, "error" : str(hr)}
+        except Exception as e:
+            logging.error(f"Ocorreu um erro ao fazer a requisição. Error: {e}")
     
-    @timer_decorator
+
     def get_file(self, file : str, ):
         """
             Obtém o conteúdo de um arquivo armazenado no Google Drive.
@@ -249,8 +253,7 @@ class GoogleDrive():
                 - A função assume a existência de um atributo `self.service` configurado para interagir com a API do Google Drive.
         """
         try:
-            # file_metadata = self.service.files().get(fileId=file_id, fields="name, mimeType").execute()
-            logging.debug(f"Lendo o arquivo {file.get("name")}")
+            logging.debug(f"Lendo o arquivo {file.get('name')}")
             request = self.service.files().get_media(fileId=file.get("id"))
             file_data = BytesIO(request.execute())
             logging.debug("Leitura do arquivo finalizada")

csc_cia_stne/provio.py

@@ -90,7 +90,7 @@ class Provio:
                     skip += 500
                     requisicao += 1
                     if verbose:
-                        logging.info(f"Exportando relatório: Requisição #{str(requisicao).zfill(3)} - {len(todos_os_dados)} registros exportados")
+                        logging.info(f"Exportando relatório: Requisição #{str(requisicao).zfill(3)} - {len(todos_os_dados)} registros no total")
             
                 else: