From 6675447d8aff3ba0cbd7d51c9c8d6d918517ce25 Mon Sep 17 00:00:00 2001
From: Dilawar Singh <dilawar@subcom.tech>
Date: Wed, 5 Oct 2022 14:26:34 +0530
Subject: [PATCH] chore: Fixes to the new API.
---
bitia/__main__.py | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/bitia/__main__.py b/bitia/__main__.py
index 8855fb7..fa01842 100644
--- a/bitia/__main__.py
+++ b/bitia/__main__.py
@@ -96,8 +96,8 @@ def submit_pipeline(user_input: str, server: str = DEFAULT_SERVER):
def run_pipeline(user_input: str, server: str = DEFAULT_SERVER):
"""Run a pipeline"""
pipeline_zip = user_input_to_pipeline(user_input)
- containers = submit_job(pipeline_zip, server=server)
- return _log_container(containers, server=server)
+ res = submit_job(pipeline_zip, server=server)
+ return _log_container(res["container"], server=server)
@app.command()
@@ -106,10 +106,10 @@ def version():
print(bitia.version())
-def _log_container(containers: T.List[T.Any], server: str):
- for container in containers:
- for line in _fetch_logs(container["Name"], server=server):
- print(line.decode().rstrip())
+def _log_container(container: str, server: str):
+ logger.info(f"Fetching logs from {container}")
+ for line in _fetch_logs(container, server=server):
+ print(line.decode().rstrip())
def _check_server_status(server: str) -> int:
@@ -240,7 +240,7 @@ def _fetch_logs(container: str, *, server):
logger.info(f"Fetching logs for container `{container}`")
return g_session.post(
f"{server}/logs",
- json=dict(container=container, follow=True, tail=-1),
+ json=dict(container=container, follow=True),
stream=True,
)
--
GitLab