76 lines
1.9 KiB
Python
76 lines
1.9 KiB
Python
|
"""Micromamba environments module."""
|
||
|
|
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from operator import itemgetter
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
MICROMAMBA_ROOT = Path('/micromamba') # default micromamba root location
|
||
|
|
|
||
|
|
GLOBAL_USER = 'operator'
|
||
|
|
GLOBAL_EXCLUDED = 'glicid-jupyterhub'
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class MicromambaEnv:
|
||
|
|
"""Generic micromamba environment."""
|
||
|
|
|
||
|
|
scope: str
|
||
|
|
name: str
|
||
|
|
path: str
|
||
|
|
|
||
|
|
|
||
|
|
def _envs(folder, excluded=None) -> list:
|
||
|
|
"""List micromamba environments."""
|
||
|
|
# Convert excluded as as list
|
||
|
|
if excluded is None:
|
||
|
|
excluded = []
|
||
|
|
elif isinstance(excluded, str):
|
||
|
|
excluded = [excluded]
|
||
|
|
|
||
|
|
# Get micromamba envs as pathlib.Path
|
||
|
|
envs = MICROMAMBA_ROOT / folder / 'envs'
|
||
|
|
|
||
|
|
if not envs.exists():
|
||
|
|
return []
|
||
|
|
|
||
|
|
return sorted(
|
||
|
|
[
|
||
|
|
(env.name, str(env))
|
||
|
|
for env in envs.iterdir()
|
||
|
|
if env.is_dir() and env.name not in excluded
|
||
|
|
],
|
||
|
|
key=itemgetter(0),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def _envs_user(username: str) -> list:
|
||
|
|
"""Micromamba environment(s) available to the user."""
|
||
|
|
return [MicromambaEnv('USER', name, path) for name, path in _envs(username)]
|
||
|
|
|
||
|
|
|
||
|
|
def _envs_team(username: str) -> list:
|
||
|
|
"""Micromamba environment(s) available to the user's team.
|
||
|
|
|
||
|
|
Warning
|
||
|
|
-------
|
||
|
|
At the moment micromamba team environments is not available on GLiCID.
|
||
|
|
|
||
|
|
"""
|
||
|
|
teams = [] # FIXME: pull user teams list from groups
|
||
|
|
return [
|
||
|
|
MicromambaEnv('TEAM', name, path) for team in teams for name, path in _envs(f'teams/{team}')
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def _envs_global() -> list:
|
||
|
|
"""Micromamba environment(s) available globally."""
|
||
|
|
return [
|
||
|
|
MicromambaEnv('GLOBAL', name, path)
|
||
|
|
for name, path in _envs(GLOBAL_USER, excluded=GLOBAL_EXCLUDED)
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def get_envs(username: str) -> list:
|
||
|
|
"""List of all the micromamba environment available to a user."""
|
||
|
|
return _envs_user(username) + _envs_team(username) + _envs_global()
|