csc-cia-stne 0.0.58__py3-none-any.whl → 0.0.59__py3-none-any.whl

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

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

@@ -3,7 +3,7 @@ csc_cia_stne/bc_correios.py,sha256=pQAnRrcXEMrx3N1MWydZVIhEQLerh3x8-0B045zZIzk,2
 csc_cia_stne/bc_sta.py,sha256=f75HJ7FLIDSJFLDTvvSvCYo9z0HchzP7rDY5WIdiKXY,16830
 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=JcXKlRBSxf5CKM-alT4ZpqWVWLZQfIYIX42_nENoTgU,11896
+csc_cia_stne/google_drive.py,sha256=lgcOd27vk2Mb_wP_fAWIbec-S3MIBKyh4TpRth6REXc,12788
 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
@@ -25,8 +25,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.58.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
-csc_cia_stne-0.0.58.dist-info/METADATA,sha256=8l3yCTUBCoqJcXuvwv5ioCz_13UvJLiibQVTHSOMYSw,1265
-csc_cia_stne-0.0.58.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-csc_cia_stne-0.0.58.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
-csc_cia_stne-0.0.58.dist-info/RECORD,,
+csc_cia_stne-0.0.59.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.0.59.dist-info/METADATA,sha256=3vFFjntppBcEf6EHTT0_ZtYL8fdblJ6DZOVwZF0byoY,1265
+csc_cia_stne-0.0.59.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+csc_cia_stne-0.0.59.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.0.59.dist-info/RECORD,,