csc-cia-stne 0.0.58__py3-none-any.whl → 0.0.60__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/utilitarios/functions/__init__.py

@@ -4,6 +4,7 @@ from .func_b64 import b64decode, b64encode
 from .func_converters import convert_bquery_result_to_json
 from .func_settings import get_config
 from .func_get_secret import get_secret
+from .func_rdp import obter_ip_conexao_rdp, conectar_rdp
 
 __all__ = [
     "titulo", 
@@ -12,5 +13,7 @@ __all__ = [
     "b64decode", 
     "convert_bquery_result_to_json",
     "get_config",
-    "get_secret"
+    "get_secret",
+    "obter_ip_conexao_rdp",
+    "conectar_rdp"
     ]

csc_cia_stne/utilitarios/functions/func_rdp.py

@@ -0,0 +1,103 @@
+import platform
+import subprocess
+import time
+import pyautogui
+import psutil
+import logging
+
+def obter_ip_conexao_rdp():
+    """Retorna o ip da conexão RDP
+
+    Returns:
+        str: ip
+    """
+    
+    for proc in psutil.process_iter(['pid', 'name', 'cmdline']):
+    
+        if proc.info['name'] == 'mstsc.exe' and proc.info['cmdline']:
+    
+            for arg in proc.info['cmdline']:
+    
+                if arg.startswith("/v:"):  # O argumento '/v:' contém o IP
+    
+                    return arg.replace("/v:", "").strip()
+    
+    return None
+
+def conectar_rdp(host, usuario, senha):
+    """Conecta via RDP em uma máquina remota
+    
+    Args:
+    
+            host (str): ip/host destino
+        usuario (str): usuário
+        senha (str): senha
+    
+    Returns:
+        bool: True/False
+    """
+    
+    sistema = platform.system()
+
+    if sistema == "Windows":
+
+        try:
+            
+            # Inicia o mstsc
+            cmd_conexao = f"start mstsc /v:{host} /f"
+            subprocess.run(cmd_conexao, shell=True)
+
+            # Aguarda um tempo para verificar se a conexão foi feita
+            time.sleep(3)  # Dá tempo para o usuário tentar logar
+            pyautogui.typewrite(senha)
+            pyautogui.press('enter')
+            time.sleep(2)
+            pyautogui.press('left')
+            pyautogui.press('enter')
+            time.sleep(5)
+            return True
+        
+        except Exception as e:
+            
+            logging.error(f"Falha ao tentar logar via RDP (Windows)\nErro: {str(e)}")
+            return False
+    
+    elif sistema == "Linux":
+        
+        # Comando para executar a conexao com o xfreerdp
+        # Para instalar: sudo apt install freerdp2-x11
+        comando_rdp = f"""xfreerdp /u:{usuario} /p:{senha} /v:{host} /size:1920x1080"""
+
+        # Executar o comando e capturar saída
+        try:
+            
+            processo_rdp = subprocess.Popen(
+                comando_rdp,
+                shell=True,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True
+            )
+        
+        except Exception as e:
+        
+            raise Exception(f"Falha ao executar o comando de conexão RDP no Linux. Você possui o xfreerdp instalado? (sudo apt install freerdp2-x11)\nErro: {str(e)}")
+        
+        # Aguarda 10 segundos, para aparecer o banner azul
+        time.sleep(10)
+
+        # Se a conexão foi bem-sucedida, retornar True
+        if processo_rdp.poll() is None:
+            
+            # Clica no 'enter', no banner azul
+            pyautogui.press('enter')
+            return True
+
+        else:
+
+            return False
+        
+    else:
+
+        raise Exception("Sistema operacional não suportado (Somente 'Windows' ou 'Linux').")
+

{csc_cia_stne-0.0.58.dist-info → csc_cia_stne-0.0.60.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: csc_cia_stne
-Version: 0.0.58
+Version: 0.0.60
 Summary: Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs
 License: MIT
 Keywords: karavela,csc,cia,stone,rpa,botcity,stne
@@ -26,6 +26,7 @@ Requires-Dist: webdriver-manager
 Requires-Dist: botcity-framework-web
 Requires-Dist: email-validator
 Requires-Dist: botcity-maestro-sdk
+Requires-Dist: psutil
 
 Essa biblioteca é desenvolvida e atualizada pelo time **CSC-CIA** da **Stone**
 

{csc_cia_stne-0.0.58.dist-info → csc_cia_stne-0.0.60.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
@@ -13,10 +13,11 @@ csc_cia_stne/slack.py,sha256=33_UNF7M529eIuWjmzSJFEZ4RmVNkFkuVxvxwsKY1tQ,8126
 csc_cia_stne/stne_admin.py,sha256=vnGSEzcmqWE42vg71oEuoRg6ENaGsZsXFOjxduSH4KU,23561
 csc_cia_stne/web.py,sha256=_pc6BzPy2x0RvqtZsByjtKcSuUqlVTevfmmmKbVWLhA,15417
 csc_cia_stne/utilitarios/__init__.py,sha256=0FrfH9XklutPS4QlbGjNYPulNfO2LFQoFXRnJQm-pKE,283
-csc_cia_stne/utilitarios/functions/__init__.py,sha256=rdEp4-g0hGu6QF0q6t9VtyedCpYqvQLf7xXlYKxnrrs,422
+csc_cia_stne/utilitarios/functions/__init__.py,sha256=nUcjSI23FxfX18AQ8Q_Gimyxme35JTyfCYNcIypPfNU,527
 csc_cia_stne/utilitarios/functions/func_b64.py,sha256=XGU34BIQQXWXBS0yM2B4A2wDlcrMl1unIJXjq4lpLnk,1254
 csc_cia_stne/utilitarios/functions/func_converters.py,sha256=EY1zvlBaRX7G1MceVSiRXwwKDQDZwUO9iECBL0fe5iU,481
 csc_cia_stne/utilitarios/functions/func_get_secret.py,sha256=khTtUTE-acdA9lIM8l7weejDSqoTYlf59ypBdC_F_lw,2150
+csc_cia_stne/utilitarios/functions/func_rdp.py,sha256=6nUKxa_sHqpz1kvdnOEyKkVopZY3QEByqZ6Sfw64gOs,2791
 csc_cia_stne/utilitarios/functions/func_recriar_pastas.py,sha256=4whZpB3aJQaCPJ3osMAQpKrzEhqYtJbljGWlx_OvKIM,826
 csc_cia_stne/utilitarios/functions/func_settings.py,sha256=XwlfqdcfocXQ8kTsDKZ6GsAtpzr0_u44AOTIMtdem7U,2059
 csc_cia_stne/utilitarios/functions/func_titulo.py,sha256=bH4tYxovTARF-g2kWUK_GIzzXt8egbVdp6mWD6fc_3I,5345
@@ -25,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.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.60.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.0.60.dist-info/METADATA,sha256=sG1SMPQd88mTGRsZsYElFnIPoJ6wiwpWM9aOgw-9534,1287
+csc_cia_stne-0.0.60.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+csc_cia_stne-0.0.60.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.0.60.dist-info/RECORD,,