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

csc_cia_stne/google_drive.py

@@ -256,9 +256,9 @@ class GoogleDrive:
         except Exception as e:
             return {"success": False, "result": None, "error": str(e)}
 
-    def download_file(self, file: str, mimeType: str):
+    def download_google_files(self, file: str, mimeType: str, path: 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. Aceito somente para extensões Google
 
         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`).
 
@@ -281,15 +281,63 @@ class GoogleDrive:
         Dependências:
             - A função assume a existência de um atributo `self.service` configurado para interagir com a API do Google Drive.
         """
+        if not os.path.exists(path):
+            os.makedirs(path)
         try:
             request = self.service.files().export_media(
                 fileId=file.get("id"), mimeType=mimeType
             )
+            file_path = f"{path}{file["name"]}"
             with open(file["name"], "wb") as f:
                 f.write(request.execute())
             return {
                 "success": True,
-                "result": f"Arquivo baixado com sucesso. Diretório: {file}",
+                "result": file_path,
+            }
+
+        except Exception as e:
+            return {"success": False, "result": None}
+
+    def download_others_files(self, file: str,  path: str):
+        """
+        Obtém o conteúdo de um arquivo armazenado nos seguintes formatos:
+        .xlsx, .pdf, .jpg, etc.
+
+        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.
+            - mimeType (str): Tipo do arquivo, por exemplo: xlsx = application/vnd.openxmlformats-officedocument.spreadsheetml.sheet
+            - path (str): diretório onde será salvo 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.
+
+        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.
+        """
+        if not os.path.exists(path):
+            os.makedirs(path)
+        try:
+            request = self.service.files().get_media(
+                fileId=file.get("id")
+            )
+            file_path = f"{path}{file["name"]}"
+            with open(file_path, "wb") as f:
+                f.write(request.execute())
+            return {
+                "success": True,
+                "result": file_path,
             }
 
         except Exception as e:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: csc_cia_stne
-Version: 0.0.68
+Version: 0.0.69
 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.68.dist-info → csc_cia_stne-0.0.69.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=0gZMW69jEvBMcUl3s9qaL5-q2IXOEAghnfhXp6-6ooY,12357
+csc_cia_stne/google_drive.py,sha256=OeBIliKZUxSBcOQwBfAdaQMKChWVWaEudMUDuc1rjU4,14399
 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.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,,
+csc_cia_stne-0.0.69.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.0.69.dist-info/METADATA,sha256=HFuGB3cNOVMxx1_KC1uwY1GPJ-94OunEZSoobehMqjI,1340
+csc_cia_stne-0.0.69.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+csc_cia_stne-0.0.69.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.0.69.dist-info/RECORD,,