csc-cia-stne 0.0.58__tar.gz → 0.0.59__tar.gz

{csc_cia_stne-0.0.58 → csc_cia_stne-0.0.59}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: csc_cia_stne
-Version: 0.0.58
+Version: 0.0.59
 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.58 → csc_cia_stne-0.0.59}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csc_cia_stne"
-version = "0.0.58"
+version = "0.0.59"
 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.0.58 → csc_cia_stne-0.0.59}/src/csc_cia_stne/google_drive.py

@@ -227,7 +227,7 @@ class GoogleDrive():
             logging.error(f"Ocorreu um erro ao fazer a requisição. Error: {e}")
     
 
-    def get_file(self, file : str, ):
+    def download_file(self, file : str, mimeType : str):
         """
             Obtém o conteúdo de um arquivo armazenado no Google Drive.
 
@@ -253,16 +253,37 @@ class GoogleDrive():
                 - A função assume a existência de um atributo `self.service` configurado para interagir com a API do Google Drive.
         """
         try:
-            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")
-
-            return {"success" : True, "result" : file_data}
+            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}"}
     
         except Exception as e:
             logging.debug(f"Erro ao tentar abrir o arquivo. Erro {e}")
             return {"success" : False, "result" : None}
+    
 
+    def get_base_data(self, id_sheet:str,page:str) -> list:
+        """
+        Retorna os dados da planilha especificada.
+        Parâmetros:
+        - drive_client: Cliente do Google Drive.
+        - id_sheet: ID da planilha.
+        - page: Nome da página da planilha.
+        Retorna:
+        - Uma lista contendo os valores da planilha.
+        Exemplo de uso:
+        >>> drive_client = ...
+        >>> id_sheet = "abc123"
+        >>> page = "Sheet1"
+        >>> data = get_base_data(drive_client, id_sheet, page)
+        """
+        try:
+            sheet = self.service.spreadsheets()
+            result = sheet.values().get(spreadsheetId=id_sheet, range=page).execute()
+            values = result.get('values', [])
+            return {"success" : True, "result" : values}
+        except Exception as e:
+            return {"success" : False, "result" : None, "error" : str(e)}
         
 

{csc_cia_stne-0.0.58 → csc_cia_stne-0.0.59}/src/csc_cia_stne.egg-info/PKG-INFO

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