From dcede0089300ecf3504eb35c7e2a848db5d8f2c5 Mon Sep 17 00:00:00 2001 From: EugeneTorap Date: Wed, 6 Nov 2024 22:37:44 +0300 Subject: [PATCH] Add modify_pod_hook to jupyterhub_config.py --- jupyterhub/files/hub/jupyterhub_config.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/jupyterhub/files/hub/jupyterhub_config.py b/jupyterhub/files/hub/jupyterhub_config.py index 7047173b79..d85ae8c69a 100644 --- a/jupyterhub/files/hub/jupyterhub_config.py +++ b/jupyterhub/files/hub/jupyterhub_config.py @@ -23,6 +23,24 @@ ) +def modify_pod_hook(spawner, pod): + if containers := pod.spec.containers: + values = [e.value for e in containers[0].env if e.name == "JUPYTERHUB_USER"] + if not values: + return pod + + split_tenant_env = values[0].split("_")[1:] + tenant_env = "_".join(split_tenant_env) + _volume_mounts = containers[0].volume_mounts + if len(_volume_mounts) >= 2 and _volume_mounts[1].name == "jupyterhub-shared-tenant": + _volume_mounts[1].sub_path = tenant_env + + return pod + + +c.KubeSpawner.modify_pod_hook = modify_pod_hook + + def camelCaseify(s): """convert snake_case to camelCase