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

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

@@ -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 → csc_cia_stne-0.0.60}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csc_cia_stne"
-version = "0.0.58"
+version = "0.0.60"
 license = { text = "MIT" }
 description = "Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs"
 keywords = ["karavela", "csc", "cia", "stone", "rpa", "botcity", "stne"]
@@ -31,5 +31,6 @@ dependencies = [
     "webdriver-manager",
     "botcity-framework-web",
     "email-validator",
-    "botcity-maestro-sdk"
+    "botcity-maestro-sdk",
+    "psutil"
 ]

{csc_cia_stne-0.0.58 → csc_cia_stne-0.0.60}/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.60}/src/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-0.0.60/src/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 → csc_cia_stne-0.0.60}/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.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 → csc_cia_stne-0.0.60}/src/csc_cia_stne.egg-info/SOURCES.txt

@@ -26,6 +26,7 @@ src/csc_cia_stne/utilitarios/functions/__init__.py
 src/csc_cia_stne/utilitarios/functions/func_b64.py
 src/csc_cia_stne/utilitarios/functions/func_converters.py
 src/csc_cia_stne/utilitarios/functions/func_get_secret.py
+src/csc_cia_stne/utilitarios/functions/func_rdp.py
 src/csc_cia_stne/utilitarios/functions/func_recriar_pastas.py
 src/csc_cia_stne/utilitarios/functions/func_settings.py
 src/csc_cia_stne/utilitarios/functions/func_titulo.py

{csc_cia_stne-0.0.58 → csc_cia_stne-0.0.60}/src/csc_cia_stne.egg-info/requires.txt

@@ -18,3 +18,4 @@ webdriver-manager
 botcity-framework-web
 email-validator
 botcity-maestro-sdk
+psutil