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()