43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
# Copyright 2013-2024 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 os
|
|
import sys
|
|
from os.path import dirname as dn
|
|
|
|
|
|
def get_spack_sys_paths(spack_prefix):
|
|
"""Given a spack prefix, return all the paths Spack needs to function."""
|
|
spack_libs = os.path.join(spack_prefix, "lib", "spack")
|
|
external_libs = os.path.join(spack_libs, "external")
|
|
vendored_libs = os.path.join(external_libs, "_vendoring")
|
|
|
|
# spack externals take precedence, then vendored packages, then spack itself
|
|
return [external_libs, vendored_libs, spack_libs]
|
|
|
|
|
|
def main(argv=None):
|
|
# Find spack's location and its prefix.
|
|
this_file = os.path.realpath(os.path.expanduser(__file__))
|
|
spack_prefix = dn(dn(dn(dn(this_file))))
|
|
|
|
# Add all the sys paths that allow spack libs to be imported
|
|
sys.path[:0] = get_spack_sys_paths(spack_prefix)
|
|
|
|
# Here we delete ruamel.yaml in case it has been already imported from site
|
|
# (see #9206 for a broader description of the issue).
|
|
#
|
|
# Briefly: ruamel.yaml produces a .pth file when installed with pip that
|
|
# makes the site installed package the preferred one, even though sys.path
|
|
# is modified to point to another version of ruamel.yaml.
|
|
if "ruamel.yaml" in sys.modules:
|
|
del sys.modules["ruamel.yaml"]
|
|
|
|
if "ruamel" in sys.modules:
|
|
del sys.modules["ruamel"]
|
|
|
|
import spack.main # noqa: E402
|
|
|
|
sys.exit(spack.main.main(argv))
|