worker-automate-hub 0.5.87__py3-none-any.whl → 0.5.88__py3-none-any.whl

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)

{worker_automate_hub-0.5.87.dist-info → worker_automate_hub-0.5.88.dist-info}/METADATA

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

@@ -2,6 +2,7 @@ worker_automate_hub/__init__.py,sha256=LV28uQvBfpPIqudGIMJmVB8E941MjXHcu8DMoX5n8
 worker_automate_hub/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/api/ahead_service.py,sha256=QbNrZf9q7fS0s-S5fZVytqC7dINi9u2f6aB6SDrGVVA,2231
 worker_automate_hub/api/client.py,sha256=uYmKPEdt2zUS7tKYJpk4NqXAGS413rpR4m33Eq3_5IU,19840
+worker_automate_hub/api/datalake_service.py,sha256=UZTxbJFZSQ0Po2N5lWg4gRZBr0Zr0BGR5f6uGjmWCOQ,2387
 worker_automate_hub/api/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/api/helpers/api_helpers.py,sha256=SkheO2fXexeh-a4shr8Qzsz_kZhuSG0DJ7kbODctRbM,3696
 worker_automate_hub/api/rdp_service.py,sha256=L7orr60FkJr6zjETtA4me8tRYCW9m1g-i5pq6AILUFo,1647
@@ -76,7 +77,7 @@ worker_automate_hub/utils/updater.py,sha256=en2FCGhI8aZ-JNP3LQm64NJDc4awCNW7UhbV
 worker_automate_hub/utils/util.py,sha256=ZMbWjvZ6vbDgkosFAt2xUJGoAJOIZhxNfJyYplvyN3Q,162310
 worker_automate_hub/utils/utils_nfe_entrada.py,sha256=iYpOs7fb7C3bY61QHySr00xlDNzrrCP0zaexOpBPuaQ,31930
 worker_automate_hub/worker.py,sha256=CT-poyP1ZYvubArYsnnNd9oJ53SPaDwgr6p6keS3nI4,6248
-worker_automate_hub-0.5.87.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
-worker_automate_hub-0.5.87.dist-info/METADATA,sha256=Zr2zxQd6u1VHpUZNWhXz_w7LQnm8HzWvHrB6cV0krvA,2894
-worker_automate_hub-0.5.87.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-worker_automate_hub-0.5.87.dist-info/RECORD,,
+worker_automate_hub-0.5.88.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
+worker_automate_hub-0.5.88.dist-info/METADATA,sha256=8gTgzQet0ffK8xespG6aZSX2MNQvsBfe_eFRvlhph4Y,2894
+worker_automate_hub-0.5.88.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+worker_automate_hub-0.5.88.dist-info/RECORD,,