csc-cia-stne 0.0.67__py3-none-any.whl → 0.0.68__py3-none-any.whl

csc_cia_stne/google_drive.py

@@ -1,22 +1,27 @@
-import os, logging
+import os
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaFileUpload
 from googleapiclient.errors import HttpError
 from io import BytesIO
 from google.oauth2 import service_account
-from .utilitarios.validations.GoogleDriveValidator import InitParamsValidator, CreateFolderValidator, ListFolderValidator, UploadValidator
+from .utilitarios.validations.GoogleDriveValidator import (
+    InitParamsValidator,
+    CreateFolderValidator,
+    ListFolderValidator,
+    UploadValidator,
+)
 from pydantic import ValidationError
 
 
-class GoogleDrive():
-    
+class GoogleDrive:
     """
-        Classe responsável por gerenciar operações no Google Drive, como upload de arquivos, criação de pastas
-        e listagem de conteúdo. Utiliza a API do Google Drive para realizar essas operações.
+    Classe responsável por gerenciar operações no Google Drive, como upload de arquivos, criação de pastas
+    e listagem de conteúdo. Utiliza a API do Google Drive para realizar essas operações.
 
-        Args:
-            key (str): Chave de autenticação para acessar a API do Google Drive.
+    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):
@@ -25,8 +30,14 @@ class GoogleDrive():
             # Cria e armazena a instância na primeira chamada.
             cls._instance = super().__new__(cls)
         return cls._instance
-    
-    def __init__(self, token : dict, 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:
@@ -41,11 +52,16 @@ class GoogleDrive():
         google_drive = GoogleDrive(key="chave_secreta", with_subject="assunto", scopes=["https://www.googleapis.com/auth/drive"], version="v3")
         ```
         """
-        
+
         try:
-            InitParamsValidator(token=token, with_subject=with_subject, scopes=scopes, version=version)
+            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())
+            raise ValueError(
+                "Erro na validação dos dados de input da inicialização da instância:",
+                e.errors(),
+            )
         self.__token = token
         self.version = version
         self.scopes = scopes
@@ -53,7 +69,6 @@ class GoogleDrive():
         self.page_size = 1000
         self.service = self.__create_service()
 
-
     def __create_service(self):
         """
         Cria um serviço do Google Drive.
@@ -69,18 +84,15 @@ class GoogleDrive():
             else:
                 # Tratar o erro de criação do serviço
         """
-        
+
         try:
             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.info(f"Erro ao criar o serviço do Google drive: Error: {e}")
             return False
 
-
-    def __autentica(self, with_subject : str):
+    def __autentica(self, with_subject: str):
         """
         Autentica o usuário com as credenciais fornecidas e retorna as credenciais delegadas para o assunto especificado.
         Args:
@@ -93,85 +105,95 @@ class GoogleDrive():
             # Autenticar com o assunto "user@example.com"
             credentials = self.__autentica("user@example.com")
         """
-        
 
         try:
-            credentials = service_account.Credentials.from_service_account_info(self.__token, scopes=self.scopes)
+            credentials = service_account.Credentials.from_service_account_info(
+                self.__token, scopes=self.scopes
+            )
             delegated_credencial = credentials.with_subject(with_subject)
             return delegated_credencial
 
         except Exception as e:
-            logging.debug(f"Erro ao tentar autenticar. Verifique o erro: {e}")
             return False
 
-
     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.
-
-            Args:
-                folder_id (str): ID da pasta no Google Drive onde o arquivo será armazenado.
-                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.
+        Faz o upload de um arquivo para o Google Drive em uma pasta especificada.
+
+        Args:
+            folder_id (str): ID da pasta no Google Drive onde o arquivo será armazenado.
+            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)
+            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())
-        
+            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")           
-            return {"success" : False, "result" : None, "error" : "Diretório ou arquivo não encontrado"}
-        
+            return {
+                "success": False,
+                "result": None,
+                "error": "Diretório ou arquivo não encontrado",
+            }
+
         try:
-            logging.debug(f"Realizando o upload do arquivo")
             media = MediaFileUpload(file_path, mimetype=mimetype, resumable=True)
             file = (
                 self.service.files()
-                .create(body=file_metadata, media_body=media, fields="id", supportsAllDrives=True)
+                .create(
+                    body=file_metadata,
+                    media_body=media,
+                    fields="id",
+                    supportsAllDrives=True,
+                )
                 .execute()
             )
 
-            logging.debug(f"Upload realizado com sucesso")
-            return {"success" : True, "result" : file}
+            return {"success": True, "result": file}
         except Exception as e:
-            logging.debug(f"Erro ao realizar o upload do arquivo {name} no google drive. Erro: {e}")
-            return {"success" : False, "result" : None, "error" : str(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.
+        Cria uma pasta no Google Drive dentro de uma pasta existente.
 
-            Args:
-                name (str): Nome da pasta a ser criada.
-                parent_folder_id (int): ID da pasta pai onde a nova pasta será criada.
+        Args:
+            name (str): Nome da pasta a ser criada.
+            parent_folder_id (int): ID da pasta pai onde a nova pasta será criada.
 
-            Returns:
-                str: ID da pasta criada.
+        Returns:
+            str: ID da pasta criada.
         """
         try:
             CreateFolderValidator(name=name, parent_folder_id=parent_folder_id)
         except ValidationError as e:
-            raise ValueError("Erro na validação dos dados de input da inicialização da instância:", e.errors())
-            
+            raise ValueError(
+                "Erro na validação dos dados de input da inicialização da instância:",
+                e.errors(),
+            )
+
         try:
             folder_metadata = {
                 "name": name,
@@ -179,91 +201,101 @@ class GoogleDrive():
                 "mimeType": "application/vnd.google-apps.folder",
             }
             folder = (
-                self.service.files().create(body=folder_metadata, fields="id", supportsAllDrives=True).execute()
+                self.service.files()
+                .create(body=folder_metadata, fields="id", supportsAllDrives=True)
+                .execute()
             )
-            return {"success" : True, "result": folder}
+            return {"success": True, "result": folder}
         except Exception as e:
-            logging.debug(f"Não foi possível criar a pasta {name}")
-            return {"success" : False, "result": None, "error" : str(e)}
+            return {"success": False, "result": None, "error": str(e)}
+
+    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.
 
+        Args:
+            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.
+                                    Defaults to "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.
-
-            Args:
-                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.
-                                        Defaults to "nextPageToken, files(id, name)".
-
-            Returns:
-                dict: Dicionário contendo o resultado da busca.
+        Returns:
+            dict: Dicionário contendo o resultado da busca.
         """
         try:
             ListFolderValidator(query=query, fields=fields, spaces=spaces)
         except ValidationError as e:
-            raise ValueError("Erro na validação dos dados de input da lista:", e.errors())
-        try: 
+            raise ValueError(
+                "Erro na validação dos dados de input da lista:", e.errors()
+            )
+        try:
             results = (
                 self.service.files()
-                .list(q=query,
-                      spaces=spaces,
-                      pageSize=self.page_size,
-                      fields=fields,
-                      supportsAllDrives=True,
-                      includeItemsFromAllDrives=True)
+                .list(
+                    q=query,
+                    spaces=spaces,
+                    pageSize=self.page_size,
+                    fields=fields,
+                    supportsAllDrives=True,
+                    includeItemsFromAllDrives=True,
+                )
                 .execute()
             )
-            return {"success" : True, "result" : results}
+            return {"success": True, "result": results}
         except HttpError as hr:
-            logging.debug(f"Ocorreu um de http. Erro: {hr}")
-            return {"success" : False, "result" : None, "error" : str(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}")
-    
+            return {"success": False, "result": None, "error": str(e)}
 
-    def download_file(self, file : str, mimeType : str):
+    def download_file(self, file: str, mimeType: str):
         """
-            Obtém o conteúdo de um arquivo armazenado no Google Drive.
+        Obtém o conteúdo de um arquivo armazenado no Google Drive.
 
-            Esta função acessa o Google Drive usando a API e lê os dados do arquivo especificado, retornando-os como um objeto binário de memória (`BytesIO`).
+        Esta função acessa o Google Drive usando a API e lê os dados do arquivo especificado, retornando-os como um objeto binário de memória (`BytesIO`).
 
-            Parâmetros:
-                - file (str): Dicionário contendo informações do arquivo no Google Drive, incluindo as chaves:
-                    - `"name"`: Nome do arquivo.
-                    - `"id"`: ID do arquivo.
+        Parâmetros:
+            - file (str): Dicionário contendo informações do arquivo no Google Drive, incluindo as chaves:
+                - `"name"`: Nome do arquivo.
+                - `"id"`: ID do arquivo.
 
-            Retorna:
-                - BytesIO: Objeto em memória contendo os dados do arquivo.
-                - None: Caso ocorra um erro ao tentar abrir ou ler o arquivo.
+        Retorna:
+            - BytesIO: Objeto em memória contendo os dados do arquivo.
+            - None: Caso ocorra um erro ao tentar abrir ou ler o arquivo.
 
-            Logs:
-                - Registra mensagens indicando o início e o término da leitura do arquivo.
-                - Em caso de falha, registra o erro ocorrido.
+        Logs:
+            - Registra mensagens indicando o início e o término da leitura do arquivo.
+            - Em caso de falha, registra o erro ocorrido.
 
-            Exceções:
-                - Qualquer erro durante o processo será capturado e registrado no log. A função retornará `None` nesses casos.
+        Exceções:
+            - Qualquer erro durante o processo será capturado e registrado no log. A função retornará `None` nesses casos.
 
-            Dependências:
-                - A função assume a existência de um atributo `self.service` configurado para interagir com a API do Google Drive.
+        Dependências:
+            - A função assume a existência de um atributo `self.service` configurado para interagir com a API do Google Drive.
         """
         try:
-            request = self.service.files().export_media(fileId=file.get("id"), mimeType=mimeType)
-            with open (file["name"], "wb") as f:
+            request = self.service.files().export_media(
+                fileId=file.get("id"), mimeType=mimeType
+            )
+            with open(file["name"], "wb") as f:
                 f.write(request.execute())
-            return {"success" : True, "result" : f"Arquivo baixado com sucesso. Diretório: {file}"}
-    
+            return {
+                "success": True,
+                "result": f"Arquivo baixado com sucesso. Diretório: {file}",
+            }
+
         except Exception as e:
-            logging.debug(f"Erro ao tentar abrir o arquivo. Erro {e}")
-            return {"success" : False, "result" : None}
-    
+            return {"success": False, "result": None}
 
-    def get_base_data(self, id_sheet:str,page:str) -> list:
+    def get_base_data(self, id_sheet: str, page: str) -> list:
         """
         Retorna os dados da planilha especificada.
         Parâmetros:
@@ -281,9 +313,7 @@ class GoogleDrive():
         try:
             sheet = self.service.spreadsheets()
             result = sheet.values().get(spreadsheetId=id_sheet, range=page).execute()
-            values = result.get('values', [])
-            return {"success" : True, "result" : values}
+            values = result.get("values", [])
+            return {"success": True, "result": values}
         except Exception as e:
-            return {"success" : False, "result" : None, "error" : str(e)}
-        
-
+            return {"success": False, "result": None, "error": str(e)}

{csc_cia_stne-0.0.67.dist-info → csc_cia_stne-0.0.68.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: csc_cia_stne
-Version: 0.0.67
+Version: 0.0.68
 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.0.67.dist-info → csc_cia_stne-0.0.68.dist-info}/RECORD

@@ -3,7 +3,7 @@ csc_cia_stne/bc_correios.py,sha256=pQAnRrcXEMrx3N1MWydZVIhEQLerh3x8-0B045zZIzk,2
 csc_cia_stne/bc_sta.py,sha256=KWHTytM3msAFayt8y5_97RwWHE8Dv8ssfwyh6zU7xI0,25836
 csc_cia_stne/email.py,sha256=RK_TzWBVnUfpP-s5NvjTJJjzhICy8e2fME9EuaiySMY,8162
 csc_cia_stne/gcp_bigquery.py,sha256=jYxvqrWDOPkxc05U4aef7V5lL8ptqsE93lfn0dLFyvc,7385
-csc_cia_stne/google_drive.py,sha256=lgcOd27vk2Mb_wP_fAWIbec-S3MIBKyh4TpRth6REXc,12788
+csc_cia_stne/google_drive.py,sha256=0gZMW69jEvBMcUl3s9qaL5-q2IXOEAghnfhXp6-6ooY,12357
 csc_cia_stne/karavela.py,sha256=jJCYX43D49gGuzmwwK6bN9XVnv2dXdp9iHnnV5H1LMQ,4794
 csc_cia_stne/logger_json.py,sha256=CXxSCOFGMymDi8XE9SKnPKjW4D0wJLqDLnxqePS26i8,3187
 csc_cia_stne/logger_rich.py,sha256=WlMqxH1lMy-tzK0I4NpBRgSzPdHc2-YSU71N3SsKM5A,8338
@@ -26,8 +26,8 @@ csc_cia_stne/utilitarios/validations/GoogleDriveValidator.py,sha256=PBo-AV2bjR__
 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/web_validator.py,sha256=HYKYSpDv1RvRjZIuwTPt-AbEz-9392MxM_O329iYuSA,5722
-csc_cia_stne-0.0.67.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
-csc_cia_stne-0.0.67.dist-info/METADATA,sha256=5yKmOI6aa-7_32-o8vylgl_9mvx4UttDvEERbSw_v04,1340
-csc_cia_stne-0.0.67.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-csc_cia_stne-0.0.67.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
-csc_cia_stne-0.0.67.dist-info/RECORD,,
+csc_cia_stne-0.0.68.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.0.68.dist-info/METADATA,sha256=Rtr1P2z_LC8R8NSeqCtJhMZFF_AYCi0vYjPwMSRGBUo,1340
+csc_cia_stne-0.0.68.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+csc_cia_stne-0.0.68.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.0.68.dist-info/RECORD,,