mirror of
https://github.com/jupyterhub/the-littlest-jupyterhub.git
synced 2025-12-18 21:54:05 +08:00
9bf4a64826
We don't wanna explicitly map keys from the YAML to traitlet config for the authentication - that's a lot of busywork for no gain. We instead switch to using snake_case everywhere, and dynamically set traitlet config from YAML config!
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
"""
|
|
JupyterHub config for the littlest jupyterhub.
|
|
"""
|
|
import os
|
|
from systemdspawner import SystemdSpawner
|
|
from tljh import user, configurer
|
|
import yaml
|
|
import copy
|
|
|
|
INSTALL_PREFIX = os.environ.get('TLJH_INSTALL_PREFIX')
|
|
USER_ENV_PREFIX = os.path.join(INSTALL_PREFIX, 'user')
|
|
|
|
|
|
class CustomSpawner(SystemdSpawner):
|
|
def start(self):
|
|
"""
|
|
Perform system user activities before starting server
|
|
"""
|
|
# FIXME: Move this elsewhere? Into the Authenticator?
|
|
system_username = 'jupyter-' + self.user.name
|
|
user.ensure_user(system_username)
|
|
user.ensure_user_group(system_username, 'jupyterhub-users')
|
|
if self.user.admin:
|
|
user.ensure_user_group(system_username, 'jupyterhub-admins')
|
|
else:
|
|
user.remove_user_group(system_username, 'jupyterhub-admins')
|
|
return super().start()
|
|
|
|
|
|
c.JupyterHub.spawner_class = CustomSpawner
|
|
|
|
c.JupyterHub.port = 80
|
|
|
|
# Use a high port so users can try this on machines with a JupyterHub already present
|
|
c.JupyterHub.hub_port = 15001
|
|
|
|
c.ConfigurableHTTPProxy.should_start = False
|
|
c.ConfigurableHTTPProxy.api_url = 'http://127.0.0.1:15002'
|
|
|
|
c.SystemdSpawner.extra_paths = [os.path.join(USER_ENV_PREFIX, 'bin')]
|
|
c.SystemdSpawner.default_shell = '/bin/bash'
|
|
# Drop the '-singleuser' suffix present in the default template
|
|
c.SystemdSpawner.unit_name_template = 'jupyter-{USERNAME}'
|
|
|
|
config_overrides_path = os.path.join(INSTALL_PREFIX, 'config.yaml')
|
|
if os.path.exists(config_overrides_path):
|
|
with open(config_overrides_path) as f:
|
|
config_overrides = yaml.safe_load(f)
|
|
else:
|
|
config_overrides = {}
|
|
configurer.apply_config(config_overrides, c)
|