mirror of
https://github.com/jupyterhub/the-littlest-jupyterhub.git
synced 2025-12-18 21:54:05 +08:00
4d42f24e48
verify behavior for: - current version (no change) - old, supported version (upgrade, but not too far) - too old, re-run installer - directory exists, no conda
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
"""
|
|
Miscellaneous functions useful in at least two places unrelated to each other
|
|
"""
|
|
import logging
|
|
import re
|
|
import subprocess
|
|
|
|
# Copied into bootstrap/bootstrap.py. Make sure these two copies are exactly the same!
|
|
import pluggy
|
|
|
|
from tljh import hooks
|
|
|
|
|
|
# This function is needed also by the bootstrap script that starts this
|
|
# installer script. Make sure its replica at bootstrap/bootstrap.py stays in
|
|
# sync with this version!
|
|
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
|
|
|
|
|
|
def parse_version(version_string):
|
|
"""Parse version string to tuple
|
|
|
|
Finds all numbers and returns a tuple of ints
|
|
_very_ loose version parsing, like the old distutils.version.LooseVersion
|
|
"""
|
|
# return a tuple of all the numbers in the version string
|
|
# always succeeds, even if passed nonsense
|
|
return tuple(int(part) for part in re.findall(r"\d+", version_string))
|