From a5c5ae0366285a3d76e73e2ece448a2608c39d16 Mon Sep 17 00:00:00 2001
From: Bernardo Chrispim Baron <bernardo@impulsogov.org>
Date: Fri, 12 May 2023 19:09:06 -0300
Subject: [PATCH] =?UTF-8?q?Utiliza=20cliente=20expl=C3=ADcito=20do=20prefe?=
 =?UTF-8?q?ct?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 implementar_fluxos.py | 36 +++++++++++++++++++++++-------------
 1 file changed, 23 insertions(+), 13 deletions(-)

diff --git a/implementar_fluxos.py b/implementar_fluxos.py
index c36b5f9..0286cd5 100644
--- a/implementar_fluxos.py
+++ b/implementar_fluxos.py
@@ -48,18 +48,26 @@
     "http://127.0.0.1:4200/api",
 )
 
-
-logger.info("Configurando bloco de conexão com registro `{}`...", DOCKER_REGISTRO_URL)
-bloco_docker_registro = DockerRegistry(
-    username=DOCKER_REGISTRO_USUARIO,
-    password=DOCKER_REGISTRO_SENHA,
-    registry_url=DOCKER_REGISTRO_URL,
-    reauth=True,
-)
-bloco_docker_registro.save(
-    name="docker-registro-impulso",
-    overwrite=True,
-)
+    
+logger.info("Iniciando cliente bloco de conexão com registro `{}`...", DOCKER_REGISTRO_URL)
+async with PrefectClient(api=PREFECT_API_URL) as cliente:
+    problemas_api = await cliente.api_healthcheck()
+    if problemas_api:
+        logger.error("Tentativa de estabelecer conexão com a API falhou: " + problemas_api.__repr__)
+        raise problemas_api
+
+    logger.info("Configurando bloco de conexão com registro `{}`...", DOCKER_REGISTRO_URL)
+    bloco_docker_registro = DockerRegistry(
+        username=DOCKER_REGISTRO_USUARIO,
+        password=DOCKER_REGISTRO_SENHA,
+        registry_url=DOCKER_REGISTRO_URL,
+        reauth=True,
+    )
+    await bloco_docker_registro.save(
+        name="docker-registro-impulso",
+        overwrite=True,
+        client=cliente,
+    )
 
 logger.info("Configurando bloco de infraestrutura com Docker...")
 bloco_docker_container = DockerContainer(
@@ -79,9 +87,10 @@
     name="docker-container-impulsoetl",
     stream_output=True,
 )
-bloco_docker_container.save(
+await bloco_docker_container.save(
     name="docker-container-impulsoetl",
     overwrite=True,
+    client=cliente,
 )
 
 logger.info("Lendo módulos disponíveis...")
@@ -122,6 +131,7 @@
                 infrastructure=bloco_docker_container,
                 path=str(modulo_localizacao.parent),
                 version=impulsoetl_versao,
+                client=cliente,
             )
             if not implementacao.schedule:
                 implementacao.schedule = CronSchedule(