Move plugin manager to utils

Fix typo
This commit is contained in:
johannes.kaufmann
2019-10-28 08:48:52 +01:00
parent e87a3846d8
commit 6c00595073
3 changed files with 26 additions and 18 deletions

View File

@@ -4,13 +4,12 @@ JupyterHub config for the littlest jupyterhub.
from glob import glob from glob import glob
import os import os
import pluggy
from systemdspawner import SystemdSpawner from systemdspawner import SystemdSpawner
from tljh import configurer, user, hooks from tljh import configurer, user
from tljh.config import INSTALL_PREFIX, USER_ENV_PREFIX, CONFIG_DIR from tljh.config import INSTALL_PREFIX, USER_ENV_PREFIX, CONFIG_DIR
from tljh.normalize import generate_system_username from tljh.normalize import generate_system_username
from tljh.yaml import yaml from tljh.utils import get_plugin_manager
from jupyterhub_traefik_proxy import TraefikTomlProxy from jupyterhub_traefik_proxy import TraefikTomlProxy
from traitlets import Dict, Unicode, List from traitlets import Dict, Unicode, List
@@ -68,11 +67,8 @@ configurer.apply_config(tljh_config, c)
# Let TLJH hooks modify `c` if they want # Let TLJH hooks modify `c` if they want
# Set up plugin infrastructure
pm = pluggy.PluginManager('tljh')
pm.add_hookspecs(hooks)
pm.load_setuptools_entrypoints('tljh')
# Call our custom configuration plugin # Call our custom configuration plugin
pm = get_plugin_manager()
pm.hook.tljh_custom_jupyterhub_config(c=c) pm.hook.tljh_custom_jupyterhub_config(c=c)
# Load arbitrary .py config files if they exist. # Load arbitrary .py config files if they exist.

View File

@@ -8,14 +8,8 @@ import pwd
import subprocess import subprocess
from os.path import expanduser from os.path import expanduser
import pluggy
from tljh import hooks
# Set up plugin infrastructure # Set up plugin infrastructure
pm = pluggy.PluginManager('tljh') from tljh.utils import get_plugin_manager
pm.add_hookspecs(hooks)
pm.load_setuptools_entrypoints('tljh')
def ensure_user(username): def ensure_user(username):
@@ -31,6 +25,8 @@ def ensure_user(username):
# User doesn't exist, time to create! # User doesn't exist, time to create!
pass pass
pm = get_plugin_manager()
subprocess.check_call([ subprocess.check_call([
'useradd', 'useradd',
'--create-home', '--create-home',

View File

@@ -1,11 +1,15 @@
""" """
Miscelaneous functions useful in at least two places unrelated to each other Miscellaneous functions useful in at least two places unrelated to each other
""" """
import subprocess
import logging import logging
import subprocess
# Copied into bootstrap/bootstrap.py. Make sure these two copies are exactly the same! # Copied into bootstrap/bootstrap.py. Make sure these two copies are exactly the same!
import pluggy
from tljh import hooks
def run_subprocess(cmd, *args, **kwargs): def run_subprocess(cmd, *args, **kwargs):
""" """
Run given cmd with smart output behavior. Run given cmd with smart output behavior.
@@ -33,4 +37,16 @@ def run_subprocess(cmd, *args, **kwargs):
)) ))
# This produces multi line log output, unfortunately. Not sure how to fix. # This produces multi line log output, unfortunately. Not sure how to fix.
# For now, prioritizing human readability over machine readability. # For now, prioritizing human readability over machine readability.
logger.debug(proc.stdout.decode()) logger.debug(proc.stdout.decode())
def get_plugin_manager():
"""
Return plugin manager instance
"""
# Set up plugin infrastructure
pm = pluggy.PluginManager('tljh')
pm.add_hookspecs(hooks)
pm.load_setuptools_entrypoints('tljh')
return pm