"""
To append instructions in modulefile.
"""
import os
from pathlib import Path
from typing import Tuple
from . import get_module_filename, get_parser
from venvmod.modulefile import add_command
[docs]def append_command(arguments: Tuple[str, str, str],
description: str,
help_arguments: str,
command: str):
if arguments is None:
options = get_parser(description=description,
help_arguments=help_arguments,
with_appli=True)
arguments = " ".join(options.arguments)
appli = options.appli
virtual_env = options.virtual_env
else:
virtual_env, appli, arguments = arguments
add_command(filename=get_module_filename(virtual_env_name=virtual_env,
appli_name=appli),
line=f"{command} {arguments}")
[docs]def module_use(arguments: Tuple[str, str, str] = None):
append_command(arguments,
description="Add dir(s) to MODULEPATH variable.",
help_arguments="path1 path2 ...",
command="module use")
[docs]def module_load(arguments: Tuple[str, str, str] = None):
append_command(arguments,
description="Load modulefile(s).",
help_arguments="module1 module2 ...",
command="module load")
[docs]def source_sh(arguments: Tuple[str, str, str] = None):
append_command(arguments,
description="Script(s) to source.",
help_arguments="SHELL script1 script2 ...",
command="source-sh")
[docs]def prepend_path(arguments: Tuple[str, str, str] = None):
append_command(arguments,
description="Prepend value to environment variable.",
help_arguments="ENV_VAR paths/to/add/No1 paths/to/add/No2 ...",
command="prepend-path")
[docs]def append_path(arguments: Tuple[str, str, str] = None):
append_command(arguments,
description="Append value to environment variable.",
help_arguments="ENV_VAR paths/to/add/No1 paths/to/add/No2 ...",
command="append-path")
[docs]def setenv(arguments: Tuple[str, str, str] = None):
append_command(arguments,
description="Define environment variable.",
help_arguments="ENV_VAR value",
command="setenv")
[docs]def remove_path(arguments: Tuple[str, str, str] = None):
append_command(arguments,
description="Remove value from environment variable.",
help_arguments="ENV_VAR value",
command="remove-path")
[docs]def set_aliases(arguments: Tuple[str, str, str] = None):
append_command(arguments,
description="Define aliases.",
help_arguments="alias command",
command="set-aliases")
[docs]def read_env(arguments: Tuple[str, str] = None):
if arguments is None:
options = get_parser(description="Read environment variable to extend modulefile.",
with_appli=True)
appli = options.appli if options.appli else options.virtual_env
appli = str(Path(appli).stem) #virtual_env may be a path
virtual_env = options.virtual_env
else:
virtual_env, appli = arguments
path_to_prepend = ["LD_LIBRARY_PATH", "PYTHONPATH", "PATH"]
for envvar, value in os.environ.items():
if not envvar.lower().startswith(appli.lower()):
continue
for path in path_to_prepend:
if envvar.endswith(path):
prepend_path(arguments=(virtual_env, appli, path + " " + value.replace(":", " ")))
break
if envvar.endswith("MODULE_USE"):
module_use(arguments=(virtual_env, appli, value))
if envvar.endswith("MODULEFILES"):
module_load(arguments=(virtual_env, appli, value))
if envvar.endswith("SOURCEFILES"):
source_sh(arguments=(virtual_env, appli, value))
if envvar.endswith("EXPORTS"):
for var in value.split():
setenv(arguments=(virtual_env, appli, var.replace("=", " ")))
if envvar.endswith("ALIASES"):
for var in value.split():
set_aliases(arguments=(virtual_env, appli, var.replace("=", " ")))
if envvar.endswith("REMOVE_PATHS"):
for var in value.split():
remove_path(arguments=(virtual_env, appli, var.replace("=", " ")))