def post_pvc(namespace):
body = request.get_json()
log.info(f"Got body: {body}")
notebook = helpers.load_param_yaml(
utils.NOTEBOOK_TEMPLATE_YAML,
name=body["name"],
namespace=namespace,
serviceAccount="default-editor",
)
defaults = utils.load_spawner_ui_config()
form.set_notebook_image(notebook, body, defaults)
form.set_notebook_image_pull_policy(notebook, body, defaults)
form.set_server_type(notebook, body, defaults)
form.set_notebook_cpu(notebook, body, defaults)
form.set_notebook_memory(notebook, body, defaults)
form.set_notebook_gpus(notebook, body, defaults)
form.set_notebook_tolerations(notebook, body, defaults)
form.set_notebook_affinity(notebook, body, defaults)
form.set_notebook_configurations(notebook, body, defaults)
# Workspace Volume
workspace_vol = form.get_workspace_vol(body, defaults)
if not body.get("noWorkspace", False) and workspace_vol["type"] == "New":
# Create the PVC
ws_pvc = utils.pvc_from_dict(workspace_vol, namespace)
log.info("Creating Workspace Volume: %s", ws_pvc.to_dict())
api.create_pvc(ws_pvc, namespace)
if not body.get("noWorkspace", False) and workspace_vol["type"] != "None":
form.add_notebook_volume(
notebook,
workspace_vol["name"],
workspace_vol["name"],
"/home/jovyan",
)
# Add the Data Volumes
for vol in form.get_data_vols(body, defaults):
if vol["type"] == "New":
# Create the PVC
dtvol_pvc = utils.pvc_from_dict(vol, namespace)
log.info("Creating Data Volume: %s", dtvol_pvc)
api.create_pvc(dtvol_pvc, namespace=namespace)
form.add_notebook_volume(notebook, vol["name"], vol["name"],
vol["path"])
# shm
form.set_notebook_shm(notebook, body, defaults)
log.info("Creating Notebook: %s", notebook)
api.create_notebook(notebook, namespace)
return api.success_response("message", "Notebook created successfully.")
def load_notebook_template(**kwargs):
"""
kwargs: the parameters to be replaced in the yaml
Reads the yaml for the web app's custom resource, replaces the variables
and returns it as a python dict.
"""
return helpers.load_param_yaml(NOTEBOOK_TEMPLATE_YAML, **kwargs)
def load_pvcviewer_yaml_template(**kwargs):
"""
kwargs: the parameters to be replaced in the yaml
Reads the yaml for the web app's custom resource, replaces the variables
and returns it as a python dict.
"""
return helpers.load_param_yaml(PVCVIEWER_YAML, **kwargs)
def post_notebook(namespace):
body = request.get_json()
log.info("Got body: %s" % body)
notebook = helpers.load_param_yaml(
utils.NOTEBOOK_TEMPLATE_YAML,
name=body["name"],
namespace=namespace,
serviceAccount="default-editor",
)
defaults = utils.load_spawner_ui_config()
form.set_notebook_image(notebook, body, defaults)
form.set_notebook_image_pull_policy(notebook, body, defaults)
form.set_server_type(notebook, body, defaults)
form.set_notebook_cpu(notebook, body, defaults)
form.set_notebook_memory(notebook, body, defaults)
form.set_notebook_gpus(notebook, body, defaults)
form.set_notebook_tolerations(notebook, body, defaults)
form.set_notebook_affinity(notebook, body, defaults)
form.set_notebook_configurations(notebook, body, defaults)
form.set_notebook_environment(notebook, body, defaults)
# Workspace Volume
ws_pvc = None
workspace_vol = form.get_workspace_vol(body, defaults)
if not body.get("noWorkspace", False) and workspace_vol["type"] != "None":
ws_pvc = rok_common.rok_pvc_from_dict(workspace_vol, namespace)
rok_common.add_workspace_volume_annotations(ws_pvc, workspace_vol)
form.add_notebook_volume(
notebook,
ws_pvc.metadata.name,
ws_pvc.metadata.name,
"/home/jovyan",
)
# Add the Data Volumes
dtvol_pvcs = []
for vol in form.get_data_vols(body, defaults):
dtvol_pvc = rok_common.rok_pvc_from_dict(vol, namespace)
dtvol_pvcs.append(dtvol_pvc)
rok_common.add_data_volume_annotations(dtvol_pvc, vol)
form.add_notebook_volume(
notebook,
dtvol_pvc.metadata.name,
dtvol_pvc.metadata.name,
vol["path"],
)
# shm
form.set_notebook_shm(notebook, body, defaults)
# Create the Notebook before creating the PVCs
log.info("Creating Notebook: %s", notebook)
notebook = api.create_notebook(notebook, namespace)
# Create the PVCs with owner references to the Notebook
if ws_pvc is not None:
rok_common.add_owner_reference(ws_pvc, notebook)
log.info("Creating Workspace Volume: %s", ws_pvc.to_dict())
api.create_pvc(ws_pvc, namespace)
for dtvol_pvc in dtvol_pvcs:
rok_common.add_owner_reference(dtvol_pvc, notebook)
log.info("Creating Data Volume %s:", dtvol_pvc)
api.create_pvc(dtvol_pvc, namespace=namespace)
return api.success_response("message", "Notebook created successfully.")
def post_pvc(namespace):
body = request.get_json()
log.info("Got body: %s" % body)
notebook = helpers.load_param_yaml(
utils.NOTEBOOK_TEMPLATE_YAML,
name=body["name"],
namespace=namespace,
serviceAccount="default-editor",
)
defaults = utils.load_spawner_ui_config()
form.set_notebook_image(notebook, body, defaults)
form.set_notebook_image_pull_policy(notebook, body, defaults)
form.set_server_type(notebook, body, defaults)
form.set_notebook_cpu(notebook, body, defaults)
form.set_notebook_memory(notebook, body, defaults)
form.set_notebook_gpus(notebook, body, defaults)
form.set_notebook_tolerations(notebook, body, defaults)
form.set_notebook_affinity(notebook, body, defaults)
form.set_notebook_configurations(notebook, body, defaults)
form.set_notebook_shm(notebook, body, defaults)
# Notebook volumes
api_volumes = []
api_volumes.extend(
form.get_form_value(body, defaults, "datavols", "dataVolumes"))
workspace = form.get_form_value(body,
defaults,
"workspace",
"workspaceVolume",
optional=True)
if workspace:
api_volumes.append(workspace)
# ensure that all objects can be created
api.create_notebook(notebook, namespace, dry_run=True)
for api_volume in api_volumes:
pvc = volumes.get_new_pvc(api_volume)
if pvc is None:
continue
api.create_pvc(pvc, namespace, dry_run=True)
# create the new PVCs and set the Notebook volumes and mounts
for api_volume in api_volumes:
pvc = volumes.get_new_pvc(api_volume)
if pvc is not None:
logging.info("Creating PVC: %s", pvc)
pvc = api.create_pvc(pvc, namespace)
v1_volume = volumes.get_pod_volume(api_volume, pvc)
mount = volumes.get_container_mount(api_volume, v1_volume["name"])
notebook = volumes.add_notebook_volume(notebook, v1_volume)
notebook = volumes.add_notebook_container_mount(notebook, mount)
log.info("Creating Notebook: %s", notebook)
api.create_notebook(notebook, namespace)
return api.success_response("message", "Notebook created successfully.")