worker-automate-hub 0.5.87__tar.gz → 0.5.88__tar.gz

{worker_automate_hub-0.5.87 → worker_automate_hub-0.5.88}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: worker-automate-hub
-Version: 0.5.87
+Version: 0.5.88
 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.87 → worker_automate_hub-0.5.88}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.87"
+version = "0.5.88"
 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.88/worker_automate_hub/api/datalake_service.py

@@ -0,0 +1,70 @@
+from mimetypes import guess_type
+
+import aiohttp
+from rich.console import Console
+
+from worker_automate_hub.config.settings import load_env_config
+from worker_automate_hub.utils.logger import logger
+
+console = Console()
+
+
+async def send_file_to_datalake(
+    file: bytes, filename: str, file_extension: str = None
+) -> None:
+    """
+    Envia um arquivo para a datalake.
+
+    Args:
+        file (bytes): O conteúdo binário do arquivo.
+        filename (str): O nome do arquivo.
+        file_extension (str, optional): A extensão do arquivo. Caso não seja
+            passada, tenta determinar com base no nome do arquivo.
+
+    Raises:
+        aiohttp.ClientResponseError: Caso a API retorne um status de erro.
+        Exception: Caso ocorra um erro genérico durante o processo.
+
+    Returns:
+        None
+    """
+    try:
+        env_config, _ = load_env_config()
+
+        if not file_extension:
+            file_extension = filename.split(".")[-1]
+
+        mime_type, _ = guess_type(filename)
+        if not mime_type:
+            mime_type = "application/octet-stream"
+
+        body = aiohttp.FormData()
+        body.add_field("file", file, filename=filename, content_type=mime_type)
+
+        headers_basic = {"Authorization": f"Basic {env_config['API_AUTHORIZATION']}"}
+
+        async with aiohttp.ClientSession(
+            connector=aiohttp.TCPConnector(ssl=False)
+        ) as session:
+            async with session.post(
+                f"{env_config['API_BASE_URL']}/arquivo/send-file-to-datalake",
+                data=body,
+                headers=headers_basic,
+            ) as response:
+                response.raise_for_status()
+
+                response_text = await response.text()
+
+                log_msg = f"\nSucesso ao enviar arquivo: {filename}\nResposta da API: {response_text}"
+                console.print(log_msg, style="bold green")
+                logger.info(log_msg)
+
+    except aiohttp.ClientResponseError as e:
+        err_msg = f"Erro na resposta da API: {e.status} - {e.message}\nDetalhes: {await e.response.text()}"
+        console.print(f"\n{err_msg}\n", style="bold red")
+        logger.error(err_msg)
+
+    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)