mirror of
https://github.com/jupyterhub/the-littlest-jupyterhub.git
synced 2025-12-18 21:54:05 +08:00
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
"""
|
|
Miscellaneous functions useful in at least two places unrelated to each other
|
|
"""
|
|
import logging
|
|
import subprocess
|
|
|
|
# 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):
|
|
"""
|
|
Run given cmd with smart output behavior.
|
|
|
|
If command succeeds, print output to debug logging.
|
|
If it fails, print output to info logging.
|
|
|
|
In TLJH, this sends successful output to the installer log,
|
|
and failed output directly to the user's screen
|
|
"""
|
|
logger = logging.getLogger('tljh')
|
|
proc = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, *args, **kwargs)
|
|
printable_command = ' '.join(cmd)
|
|
if proc.returncode != 0:
|
|
# Our process failed! Show output to the user
|
|
logger.error('Ran {command} with exit code {code}'.format(
|
|
command=printable_command, code=proc.returncode
|
|
))
|
|
logger.error(proc.stdout.decode())
|
|
raise subprocess.CalledProcessError(cmd=cmd, returncode=proc.returncode)
|
|
else:
|
|
# This goes into installer.log
|
|
logger.debug('Ran {command} with exit code {code}'.format(
|
|
command=printable_command, code=proc.returncode
|
|
))
|
|
# This produces multi line log output, unfortunately. Not sure how to fix.
|
|
# For now, prioritizing human readability over machine readability.
|
|
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
|