tljh_neurodesk.py

from tljh.hooks import hookimpl as hook
import os, shutil, subprocess

image = "vnmd/neurodesktop"

@hook
def tljh_post_install():  # Setup the neurodesk-notebook
    os    .system("sudo systemctl start docker\n" +
                 f"sudo docker pull {image}")

    if os.path.isdir("/etc/apparmor.d"):
        os.chdir(__import__("CVMFS").__path__[0])
        shutil.copyfile("apparmor", "/etc/apparmor.d/CVMFS")
        os.system(f"""sudo apparmor_parser -r /etc/apparmor.d/CVMFS
            docker{ ' ' if subprocess.getoutput
                ("dpkg-query --showformat='${db:Status-Status}' -W docker-ce") == "installed"
                else '-'
            }compose up -d
        """)

@hook
def tljh_extra_apt_packages(): return ["docker.io", "docker-compose"]

@hook
def tljh_extra_hub_pip_packages(): return ["dockerspawner", "jupyter_client"]

@hook
def tljh_custom_jupyterhub_config(c):
    from jupyter_client.localinterfaces import public_ips
    c.JupyterHub.hub_ip = public_ips()[0]
    c.JupyterHub.spawner_class = "dockerspawner.DockerSpawner"
    c.DockerSpawner.name_template = "{username}-{imagename}"
    c.DockerSpawner.image = image
    c.DockerSpawner.extra_host_config = {
        "security_opt": ["apparmor=unconfined"],
        "cap_add": ["SYS_ADMIN"]
    }
    c.DockerSpawner.extra_create_kwargs = {"user": "root"}
    c.DockerSpawner.volumes = {
        "{prefix}-{username}": "/home/jovyan",
        "/cvmfs": "/cvmfs"
    }