def update_environment_build_status(
status: str,
session: requests.sessions.Session,
environment_build_uuid,
) -> Any:
"""Update environment build status."""
data = {"status": status}
if data["status"] == "STARTED":
data["started_time"] = datetime.utcnow().isoformat()
elif data["status"] in ["SUCCESS", "FAILURE"]:
data["finished_time"] = datetime.utcnow().isoformat()
url = f"{CONFIG_CLASS.ORCHEST_API_ADDRESS}/environment-builds/{environment_build_uuid}"
with session.put(url, json=data) as response:
return response.json()
def update_jupyter_image_build_status(
status: str,
session: requests.sessions.Session,
jupyter_image_build_uuid,
) -> Any:
"""Update Jupyter build status."""
data = {"status": status}
if data["status"] == "STARTED":
data["started_time"] = datetime.utcnow().isoformat()
elif data["status"] in ["SUCCESS", "FAILURE"]:
data["finished_time"] = datetime.utcnow().isoformat()
url = (f"{CONFIG_CLASS.ORCHEST_API_ADDRESS}/jupyter-builds/"
f"{jupyter_image_build_uuid}")
with session.put(url, json=data) as response:
return response.json()
