mirror of
https://github.com/jupyterhub/the-littlest-jupyterhub.git
synced 2025-12-18 21:54:05 +08:00
51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
from jupyterhub_configurator.mixins import ConfiguratorSpawnerMixin
|
|
from systemdspawner import SystemdSpawner
|
|
from traitlets import Dict, List, Unicode
|
|
|
|
from tljh import configurer, user
|
|
from tljh.normalize import generate_system_username
|
|
|
|
|
|
class CustomSpawner(SystemdSpawner):
|
|
"""
|
|
SystemdSpawner with user creation on spawn.
|
|
|
|
FIXME: Remove this somehow?
|
|
"""
|
|
|
|
user_groups = Dict(key_trait=Unicode(), value_trait=List(Unicode()), config=True)
|
|
|
|
def start(self):
|
|
"""
|
|
Perform system user activities before starting server
|
|
"""
|
|
# FIXME: Move this elsewhere? Into the Authenticator?
|
|
system_username = generate_system_username("jupyter-" + self.user.name)
|
|
|
|
# FIXME: This is a hack. Allow setting username directly instead
|
|
self.username_template = system_username
|
|
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")
|
|
if self.user_groups:
|
|
for group, users in self.user_groups.items():
|
|
if self.user.name in users:
|
|
user.ensure_user_group(system_username, group)
|
|
return super().start()
|
|
|
|
|
|
cfg = configurer.load_config()
|
|
# Use the jupyterhub-configurator mixin only if configurator is enabled
|
|
# otherwise, any bugs in the configurator backend will stop new user spawns!
|
|
if cfg["services"]["configurator"]["enabled"]:
|
|
# Dynamically create the Spawner class using `type`(https://docs.python.org/3/library/functions.html?#type),
|
|
# based on whether or not it should inherit from ConfiguratorSpawnerMixin
|
|
UserCreatingSpawner = type(
|
|
"UserCreatingSpawner", (ConfiguratorSpawnerMixin, CustomSpawner), {}
|
|
)
|
|
else:
|
|
UserCreatingSpawner = type("UserCreatingSpawner", (CustomSpawner,), {})
|