diff --git a/bitia/__main__.py b/bitia/__main__.py index 03f71448c06736842138f7a5faa052ae54908419..ca99960fa210749b5338c9a8d17e32b1463d97d9 100644 --- a/bitia/__main__.py +++ b/bitia/__main__.py @@ -72,9 +72,7 @@ def list_remote_container( params=dict(pipeline_sha256=pipeline_hash), ) res.raise_for_status() - _json = res.json() - print(_json) - return _json["containers"] + return res.json()["containers"].split(",") @app.command("logs") diff --git a/bitia/session.py b/bitia/session.py index 3dd0c28e47a51d2eac38d6b53bac27da2817bb6e..15cf542e113dc5a8ac4cb3eac9201ecc9bd5b65e 100644 --- a/bitia/session.py +++ b/bitia/session.py @@ -12,8 +12,9 @@ SESSION_PICKLE_FILE = bconfig.bitia_dir() / ".session.pickle" def fetch_logs(container: str, *, server): """Fetch logs from a container.""" logger.info(f"Fetching logs for container `{container}`") - return g_session.post( - f"{server}/logs/?container={container}", + return get( + f"{server}/container/logs", + params=dict(container=container), stream=True, ) diff --git a/tests/test_sanity.py b/tests/test_sanity.py index d1c6b3f4333ce6135fdb8fc8d7b68ceab4fe8433..0caa677de913d1e58cc5c50a5d91b6056c598769 100644 --- a/tests/test_sanity.py +++ b/tests/test_sanity.py @@ -9,10 +9,10 @@ def test_sanity(): def test_run_simple(): - output = bitia.__main__.run_pipeline("ls -ltr /") - print(output) - output = bitia.__main__.run_pipeline("ls -ltr /", rerun=True) - print(output) + for line in bitia.__main__.run_pipeline("ls -ltr /"): + print(111, line.decode().rstrip()) + for line in bitia.__main__.run_pipeline("ls -ltr /", rerun=True): + print(222, line.decode().rstrip()) if __name__ == "__main__":