update bak

jupyterhub_config.py.bak

@@ -1,68 +1,108 @@
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+
+# Configuration file for JupyterHub
 import os
-from pathlib import Path
-from dockerspawner import DockerSpawner
 
-c = get_config()
+c = get_config()  # noqa: F821
-c.Application.log_level = 'DEBUG'
 
-# 基本的JupyterHub配置
+# We rely on environment variables to configure JupyterHub so that we
-c.JupyterHub.cookie_secret_file = os.path.expanduser('~/.jupyterhub/jupyterhub_cookie_secret')
+# avoid having to rebuild the JupyterHub container every time we change a
-db_file = os.path.expanduser('~/.jupyterhub/jupyterhub.sqlite')
+# configuration parameter.
-c.JupyterHub.db_url = f'sqlite:///{db_file}'
-c.ConfigurableHTTPProxy.pid_file = os.path.expanduser('~/.jupyterhub/jupyterhub-proxy.pid')
 
-# Authenticator 设置
+# from dockerspawner import DockerSpawner
-c.JupyterHub.authenticator_class = 'jupyterhub.auth.PAMAuthenticator'
-c.PAMAuthenticator.encoding = 'utf8'
-c.Authenticator.admin_users = set()
-c.Authenticator.allowed_users = set()
-c.LocalAuthenticator.create_system_users = True
 
-# Spawner 设置
+# class MyDockerSpawner(DockerSpawner):
-c.Spawner.ip = '127.0.0.1'
+#     def start(self):
-c.Spawner.cmd = ['jupyter-labhub']
+#         # 启动父类的start方法
-c.Spawner.default_url = '/lab'
+#         self.user_options['environment']['JUPYTER_ENABLE_NBEXTENSIONS'] = 'true'
-c.LocalProcessSpawner.shell_cmd = ["bash", "-l", "-c"]
+#         self.user_options['cmd'] = [
-c.Spawner.notebook_dir = '~'
+#             'bash',
-c.Spawner.args = ['--allow-root', "--KernelSpecManager.ensure_native_kernel=False", '--NotebookApp.allow_origin_pat=https://.*vscode-cdn\\.net', '--NotebookApp.iopub_data_rate_limit=10000000']
+#             '-c',
+#             'pip install nglview jupyter_packaging -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com && jupyter nbextension enable nglview --py --sys-prefix && jupyter labextension install nglview-js-widgets && jupyter labextension install @jupyter-widgets/jupyterlab-manager && start-singleuser.sh'
+#         ]
+#         return super().start()
         
-# 环境变量保持
+# Spawn single-user servers as Docker containers
-c.Spawner.env_keep = ['PATH', 'PYTHONPATH', 'LD_LIBRARY_PATH', 'ENV1', 'ENV2']
+c.Authenticator.allow_all = True
+c.JupyterHub.spawner_class = "dockerspawner.DockerSpawner"
 
-# JupyterHub 服务配置
+# Spawn containers from this image
-c.JupyterHub.ip = '0.0.0.0'
+c.DockerSpawner.image = os.environ["DOCKER_NOTEBOOK_IMAGE"]
-c.JupyterHub.port = 9000
-c.JupyterHub.shutdown_on_logout = True
-c.JupyterHub.statsd_prefix = 'jupyterhub'
-c.JupyterHub.page_title = 'JupyterHub Service'
 
-# Dockerspawner 配置（如果需要启用）
+# Connect containers to this Docker network
-c.JupyterHub.spawner_class = DockerSpawner
+network_name = os.environ["DOCKER_NETWORK_NAME"]
-c.DockerSpawner.allowed_images='*'
+c.DockerSpawner.use_internal_ip = True
+c.DockerSpawner.network_name = network_name
 
-# Docker 守护进程的地址
+# Explicitly set notebook directory because we'll be mounting a volume to it.
-c.DockerSpawner.docker_host = 'unix:///var/run/docker.sock'
+# Most `jupyter/docker-stacks` *-notebook images run the Notebook server as
+# user `jovyan`, and set the notebook directory to `/home/jovyan/work`.
+# We follow the same convention.
+notebook_dir = os.environ.get("DOCKER_NOTEBOOK_DIR", "/home/jovyan/work")
+c.DockerSpawner.notebook_dir = notebook_dir
 
-# 使用的 Docker 镜像
+# Mount the real user's Docker volume on the host to the notebook user's
-c.DockerSpawner.image = 'quay.io/jupyter/scipy-notebook'
+# notebook directory in the container
+# c.DockerSpawner.volumes = {"jupyterhub-user-{username}": notebook_dir}
+# Mount the real user's Docker volume on the host to the notebook user's
+# notebook directory in the container
+c.DockerSpawner.volumes = {
+    "jupyterhub-user-{username}": notebook_dir,
+    "/mnt/mydrive": "/home/jovyan/work/mydrive",
+    "/mnt/mydrive/project/docker-jupyterhub/id_rsa": "/home/jovyan/.ssh/id_rsa",
+}
 
-# 删除容器当它停止时
+
+# Remove containers once they are stopped
 c.DockerSpawner.remove = True
 
-# 设置网络（如果您有特定的 Docker 网络配置）
+# For debugging arguments passed to spawned containers
-# c.DockerSpawner.network_name = 'jupyterhub'
+c.DockerSpawner.debug = True
+# c.Application.log_level = 'DEBUG'
 
-# JupyterHub 的连接地址，用于 DockerSpawner 内部通信
+# User containers will access hub by container name on the Docker network
-# 如果 JupyterHub 运行在同一 Docker 网络中，可以使用 Docker 容器名称
+c.JupyterHub.hub_ip = 'jupyterhub'
-# c.JupyterHub.hub_connect_ip = 'jupyterhub'
+c.JupyterHub.hub_port = 8080
 
-# 其他配置...
+# Persist hub data on volume mounted inside container
+c.JupyterHub.cookie_secret_file = "/data/jupyterhub_cookie_secret"
+c.JupyterHub.db_url = "sqlite:////data/jupyterhub.sqlite"
 
-# 注意：下面这行配置是不必要的，因为您已经使用 Unix 套接字
+# Authenticate users with Native Authenticator
-# c.DockerSpawner.docker_host = 'tcp://docker-daemon-host:2375'
+c.JupyterHub.authenticator_class = "nativeauthenticator.NativeAuthenticator"
-# 如果使用TLS（根据需要取消注释）
+
-# os.environ['DOCKER_TLS_VERIFY'] = '1'
+# Allow anyone to sign-up without approval
-# os.environ['DOCKER_CERT_PATH'] = '/path/to/certificates'
+c.NativeAuthenticator.open_signup = True
+
+# Allowed admins
+admin = os.environ.get("JUPYTERHUB_ADMIN")
+if admin:
+    c.Authenticator.admin_users = [admin]
+
+# c.DockerSpawner.extra_create_kwargs.update({
+#    "environment": {"JUPYTER_ENABLE_LAB": "yes"}
+# })
+
+# 启动jupyter时候增加跨域支持, 否则反向代理的时候出现问题
+# --NotebookApp.iopub_data_rate_limit=10000000 给nglview使用
+c.DockerSpawner.extra_create_kwargs.update({
+    "environment": {"NOTEBOOK_ARGS": "--NotebookApp.allow_origin='*' --NotebookApp.iopub_data_rate_limit=10000000"}
+})
+
+
+# 要支持正则匹配的域名请求，可以通过设置 allow_origin_pat 参数来实现。这个参数允许你使用正则表达式来匹配允许跨域请求的域名。例如，如果你想允许所有以 .example.com 结尾的域名进行跨域请求，可以在 jupyterhub_config.py 文件中添加如下配置：
+
+# c.Spawner.environment = {
+#     'JUPYTERHUB_CORS': '{"allow_origin_pat": "https?://.*\\.example\\.com"}'
+# }
+
+# GPU 和网络配置
+c.DockerSpawner.extra_host_config = {
+    'runtime': 'nvidia'
+}
+c.DockerSpawner.environment = {
+    'NVIDIA_DRIVER_CAPABILITIES': 'compute,utility',
+    'NVIDIA_VISIBLE_DEVICES': 'all'
+}
 
-# 其他配置（根据需要添加）
-# ...