Skip to content
Snippets Groups Projects

Hotfix: fix to assert error post installation

Merged dilawar requested to merge hotfix into main
Files
10
+ 21
8
@@ -54,7 +54,10 @@ def session(func):
@app.command("create-container")
@session
def create_remote_container(
user_input, *, recreate: bool = False, output_lines: T.List[str] = []
user_input,
*,
recreate: bool = False,
output_lines: T.List[str] = [],
):
"""Create container for the pipeline. The container starts running
immediately on the server. Use command `logs` to stream the output.
@@ -71,9 +74,12 @@ def create_remote_container(
stream=True,
)
res.raise_for_status()
for line in track(
res.iter_lines(), description="BiTIA is setting up required infra..."
):
_lines = (
track(res.iter_lines(), description="BiTIA is setting up required infra...")
if not bconfig.get_config("plain", default=False)
else res.iter_lines()
)
for line in _lines:
output_lines.append(line.decode().rstrip())
logger.info(output_lines[-1])
return res
@@ -90,7 +96,9 @@ def list_containers(user_input):
def _list_remote_container(user_input) -> T.List[str]:
pipeline = bpipeline.user_input_to_pipeline(user_input)
logger.debug(f"sha256 of `{pipeline.zipfile}` is `{pipeline.checksum}`")
logger.info(f"Finding container for {user_input}, sha256sum={pipeline.checksum}")
logger.info(
f"Finding container for user input `{user_input}` with sha256sum={pipeline.checksum}"
)
res = bhelper.get(
endpoint="container/list",
server=bconfig.get_server(),
@@ -102,10 +110,12 @@ def _list_remote_container(user_input) -> T.List[str]:
@app.command("logs")
@session
def stream_log(user_input, server: str = bconfig.get_server()):
def stream_log(user_input):
"""Stream logs for the most recent run of a given pipeline."""
pipeline = bpipeline.user_input_to_pipeline(user_input)
logger.info(f"Finding container for {user_input}, sha256sum={pipeline.checksum}")
logger.info(
f"Finding container for user input {user_input} with sha256sum={pipeline.checksum}"
)
res = bhelper.get(
endpoint="logs",
params=dict(pipeline_sha256=pipeline.checksum),
@@ -154,16 +164,19 @@ def main(
verbose: VerbosityLevel = typer.Option(
VerbosityLevel.warning, case_sensitive=False
),
plain: bool = False,
server: T.Optional[str] = None,
):
"""
Callback
"""
bconfig.set_config("plain", plain)
bconfig.set_config("verbosity", verbose.value)
set_logger_level(verbose.value)
if server is not None:
bconfig.set_server(server)
elif os.environ.get("BITIA_SERVER") is not None:
elif os.environ.get("BITIA_SERVER"):
bconfig.set_server(os.environ["BITIA_SERVER"])
else:
# TODO: Read from default config file.
Loading