603569e321
* Style: black 23, skip magic trailing commas * isort should use same line length as black * Fix unused import * Update version of black used in CI * Update new packages * Update new packages
119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
# Copyright 2013-2023 Lawrence Livermore National Security, LLC and other
|
|
# Spack Project Developers. See the top-level COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
import sys
|
|
|
|
import spack.cmd
|
|
import spack.cmd.common.arguments as arguments
|
|
import spack.cmd.find
|
|
import spack.environment as ev
|
|
import spack.store
|
|
import spack.user_environment as uenv
|
|
import spack.util.environment
|
|
|
|
description = "add package to the user environment"
|
|
section = "user environment"
|
|
level = "short"
|
|
|
|
|
|
def setup_parser(subparser):
|
|
"""Parser is only constructed so that this prints a nice help
|
|
message with -h."""
|
|
arguments.add_common_arguments(subparser, ["constraint"])
|
|
|
|
shells = subparser.add_mutually_exclusive_group()
|
|
shells.add_argument(
|
|
"--sh",
|
|
action="store_const",
|
|
dest="shell",
|
|
const="sh",
|
|
help="print sh commands to load the package",
|
|
)
|
|
shells.add_argument(
|
|
"--csh",
|
|
action="store_const",
|
|
dest="shell",
|
|
const="csh",
|
|
help="print csh commands to load the package",
|
|
)
|
|
shells.add_argument(
|
|
"--fish",
|
|
action="store_const",
|
|
dest="shell",
|
|
const="fish",
|
|
help="print fish commands to load the package",
|
|
)
|
|
shells.add_argument(
|
|
"--bat",
|
|
action="store_const",
|
|
dest="shell",
|
|
const="bat",
|
|
help="print bat commands to load the package",
|
|
)
|
|
|
|
subparser.add_argument(
|
|
"--first",
|
|
action="store_true",
|
|
default=False,
|
|
dest="load_first",
|
|
help="load the first match if multiple packages match the spec",
|
|
)
|
|
|
|
subparser.add_argument(
|
|
"--only",
|
|
default="package,dependencies",
|
|
dest="things_to_load",
|
|
choices=["package", "dependencies"],
|
|
help="""select whether to load the package and its dependencies
|
|
the default is to load the package and all dependencies
|
|
alternatively one can decide to load only the package or only
|
|
the dependencies""",
|
|
)
|
|
|
|
subparser.add_argument(
|
|
"--list",
|
|
action="store_true",
|
|
default=False,
|
|
help="show loaded packages: same as `spack find --loaded`",
|
|
)
|
|
|
|
|
|
def load(parser, args):
|
|
env = ev.active_environment()
|
|
|
|
if args.list:
|
|
results = spack.cmd.filter_loaded_specs(args.specs())
|
|
if sys.stdout.isatty():
|
|
spack.cmd.print_how_many_pkgs(results, "loaded")
|
|
spack.cmd.display_specs(results)
|
|
return
|
|
|
|
specs = [
|
|
spack.cmd.disambiguate_spec(spec, env, first=args.load_first)
|
|
for spec in spack.cmd.parse_specs(args.constraint)
|
|
]
|
|
|
|
if not args.shell:
|
|
specs_str = " ".join(args.constraint) or "SPECS"
|
|
spack.cmd.common.shell_init_instructions(
|
|
"spack load", " eval `spack load {sh_arg} %s`" % specs_str
|
|
)
|
|
return 1
|
|
|
|
with spack.store.db.read_transaction():
|
|
if "dependencies" in args.things_to_load:
|
|
include_roots = "package" in args.things_to_load
|
|
specs = [
|
|
dep for spec in specs for dep in spec.traverse(root=include_roots, order="post")
|
|
]
|
|
|
|
env_mod = spack.util.environment.EnvironmentModifications()
|
|
for spec in specs:
|
|
env_mod.extend(uenv.environment_modifications_for_spec(spec))
|
|
env_mod.prepend_path(uenv.spack_loaded_hashes_var, spec.dag_hash())
|
|
cmds = env_mod.shell_modifications(args.shell)
|
|
|
|
sys.stdout.write(cmds)
|