zhangyiss/the-littlest-jupyterhub
1
0
Fork 0
mirror of https://github.com/jupyterhub/the-littlest-jupyterhub.git synced 2025-12-18 21:54:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup bootstrap + add docstrings

Browse Source
This commit is contained in:
yuvipanda
2018-07-02 15:55:53 -07:00
parent 715860707b
commit f3cfd36d52
1 changed files with 21 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
bootstrap/bootstrap.py
View File

@@ -47,28 +47,28 @@ def check_miniconda_version(prefix, version):
@contextlib.contextmanager @contextlib.contextmanager
def download_miniconda_installer(version): def download_miniconda_installer(version, md5sum):
md5sums = { """
'4.5.4': "a946ea1d0c4a642ddf0c3a26a18bb16d" Context manager to download miniconda installer of given version.
}
if version not in md5sums:
raise ValueError(
'minicondaversion {} not supported. Supported version:'.format(
version, ' '.join(md5sums.keys())
))
This should be used as a contextmanager. It downloads miniconda installer
of given version, verifies the md5sum & provides path to it to the `with`
block to run.
"""
with tempfile.NamedTemporaryFile() as f: with tempfile.NamedTemporaryFile() as f:
installer_url = "https://repo.continuum.io/miniconda/Miniconda3-{}-Linux-x86_64.sh".format(version) installer_url = "https://repo.continuum.io/miniconda/Miniconda3-{}-Linux-x86_64.sh".format(version)
urllib.request.urlretrieve(installer_url, f.name) urllib.request.urlretrieve(installer_url, f.name)
if md5_file(f.name) != md5sums[version]: if md5_file(f.name) != md5sum:
raise Exception('md5 hash mismatch! Downloaded file corrupted') raise Exception('md5 hash mismatch! Downloaded file corrupted')
yield f.name yield f.name
def install_miniconda(installer_path, prefix): def install_miniconda(installer_path, prefix):
"""
Install miniconda with installer at installer_path under prefix
"""
subprocess.check_output([ subprocess.check_output([
'/bin/bash', '/bin/bash',
installer_path, installer_path,
@@ -78,6 +78,14 @@ def install_miniconda(installer_path, prefix):
def pip_install(prefix, packages, editable=False): def pip_install(prefix, packages, editable=False):
"""
Install pip packages in the conda environment under prefix.
Packages are upgraded if possible.
Set editable=True to add '--editable' to the pip install commandline.
Very useful when doing active development
"""
flags = '--no-cache-dir --upgrade' flags = '--no-cache-dir --upgrade'
if editable: if editable:
flags += '--editable' flags += '--editable'
@@ -91,13 +99,12 @@ def pip_install(prefix, packages, editable=False):
def main(): def main():
install_prefix = os.environ.get('TLJH_INSTALL_PREFIX', '/opt/tljh') install_prefix = os.environ.get('TLJH_INSTALL_PREFIX', '/opt/tljh')
hub_prefix = os.path.join(install_prefix, 'hub') hub_prefix = os.path.join(install_prefix, 'hub')
miniconda_version = '4.5.4'
print('Checking if TLJH is already installed...') print('Checking if TLJH is already installed...')
if not check_miniconda_version(hub_prefix, miniconda_version): if not check_miniconda_version(hub_prefix, miniconda_version):
initial_setup = True initial_setup = True
print('Downloading & setting up hub environment...') print('Downloading & setting up hub environment...')
with download_miniconda_installer(miniconda_version) as installer_path: with download_miniconda_installer('4.5.4', "a946ea1d0c4a642ddf0c3a26a18bb16d") as installer_path:
install_miniconda(installer_path, hub_prefix) install_miniconda(installer_path, hub_prefix)
print('Hub environment set up!') print('Hub environment set up!')
else: else:
@@ -121,5 +128,6 @@ def main():
'tljh.installer' 'tljh.installer'
) )
if __name__ == '__main__': if __name__ == '__main__':
main() main()