worker-automate-hub 0.5.5__tar.gz → 0.5.6__tar.gz

{worker_automate_hub-0.5.5 → worker_automate_hub-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: worker-automate-hub
-Version: 0.5.5
+Version: 0.5.6
 Summary: Worker Automate HUB é uma aplicação para automatizar rotinas de RPA nos ambientes Argenta.
 Author: Joel Paim
 Requires-Python: >=3.12,<4.0

{worker_automate_hub-0.5.5 → worker_automate_hub-0.5.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.5"
+version = "0.5.6"
 description = "Worker Automate HUB é uma aplicação para automatizar rotinas de RPA nos ambientes Argenta."
 authors = ["Joel Paim"]
 readme = "README.md"

{worker_automate_hub-0.5.5 → worker_automate_hub-0.5.6}/worker_automate_hub/api/client.py

@@ -491,4 +491,28 @@ async def send_file(uuidRelacao: str, desArquivo: str, tipo: str, file: bytes, f
     except Exception as e:
         err_msg = f"Erro ao enviar arquivo: {str(e)}"
         console.print(f"\n{err_msg}\n", style="bold red")
-        logger.error(err_msg)
+        logger.error(err_msg)
+
+
+async def change_robot_status_true(uuid_robo: str):
+    env_config, _ = load_env_config()
+    try: 
+        headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
+
+        async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(verify_ssl=True)) as session:
+            async with session.get(
+                f"{env_config["API_BASE_URL"]}/automate-hub/robo/enable-robot/{uuid_robo}",
+                headers=headers_basic,
+            ) as response:
+                if response.status == 200:
+                    console.print("Robo alterado com sucesso!", style="bold green")
+                else:
+                    raise Exception(f"{response.status} - {response.text}")
+    except Exception as e:
+        err_msg = f"Erro ao obter nova tarefa: {e}"
+        logger.error(err_msg)
+        console.print(
+            f"{err_msg}\n",
+            style="bold red",
+        )
+        return None

{worker_automate_hub-0.5.5 → worker_automate_hub-0.5.6}/worker_automate_hub/tasks/jobs/conexao_rdp.py

@@ -8,6 +8,7 @@ import pygetwindow as gw
 from rich.console import Console
 import pygetwindow as gw
 from pywinauto import Application
+from worker_automate_hub.api.client import change_robot_status_true 
 from worker_automate_hub.models.dto.rpa_historico_request_dto import RpaHistoricoStatusEnum, RpaRetornoProcessoDTO
 from worker_automate_hub.models.dto.rpa_processo_rdp_dto import RpaProcessoRdpDTO
 from worker_automate_hub.utils.logger import logger
@@ -207,6 +208,8 @@ async def conexao_rdp(task: RpaProcessoRdpDTO) -> RpaRetornoProcessoDTO:
             
             await rdp.clicar_no_titulo()
             
+            await change_robot_status_true(task.uuidRobo)
+
             return RpaRetornoProcessoDTO(
                 sucesso=True,
                 retorno="Conexão RDP estabelecida com sucesso.",