diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/spack/spack/__init__.py | 18 | ||||
-rw-r--r-- | lib/spack/spack/cmd/compiler.py | 13 | ||||
-rw-r--r-- | lib/spack/spack/cmd/mirror.py | 168 | ||||
-rw-r--r-- | lib/spack/spack/config.py | 13 | ||||
-rw-r--r-- | lib/spack/spack/package.py | 24 | ||||
-rw-r--r-- | lib/spack/spack/stage.py | 46 | ||||
-rw-r--r-- | lib/spack/spack/util/compression.py | 9 |
7 files changed, 226 insertions, 65 deletions
diff --git a/lib/spack/spack/__init__.py b/lib/spack/spack/__init__.py index c4a2de0b2a..904c6262c8 100644 --- a/lib/spack/spack/__init__.py +++ b/lib/spack/spack/__init__.py @@ -138,24 +138,6 @@ do_checksum = True sys_type = None # -# Places to download tarballs from. -# -# TODO: move to configuration. -# -# Examples: -# -# For a local directory: -# mirrors = ['file:///Users/gamblin2/spack-mirror'] -# -# For a website: -# mirrors = ['http://spackports.org/spack-mirror/'] -# -# For no mirrors: -# mirrors = [] -# -mirrors = [] - -# # Extra imports that should be generally usable from package.py files. # from llnl.util.filesystem import working_dir diff --git a/lib/spack/spack/cmd/compiler.py b/lib/spack/spack/cmd/compiler.py index 0586484c32..a4cd2df7e2 100644 --- a/lib/spack/spack/cmd/compiler.py +++ b/lib/spack/spack/cmd/compiler.py @@ -60,8 +60,17 @@ def compiler_add(args): if not paths: paths = get_path('PATH') - compilers = spack.compilers.find_compilers(*args.add_paths) - spack.compilers.add_compilers_to_config('user', *compilers) + compilers = [c for c in spack.compilers.find_compilers(*args.add_paths) + if c.spec not in spack.compilers.all_compilers()] + + if compilers: + spack.compilers.add_compilers_to_config('user', *compilers) + n = len(compilers) + tty.msg("Added %d new compiler%s to %s" % ( + n, 's' if n > 1 else '', spack.config.get_filename('user'))) + colify(reversed(sorted(c.spec for c in compilers)), indent=4) + else: + tty.msg("Found no new compilers") def compiler_remove(args): diff --git a/lib/spack/spack/cmd/mirror.py b/lib/spack/spack/cmd/mirror.py index 94e40557ff..813c58b84d 100644 --- a/lib/spack/spack/cmd/mirror.py +++ b/lib/spack/spack/cmd/mirror.py @@ -25,69 +25,197 @@ import os import shutil import argparse +from datetime import datetime +from contextlib import closing import llnl.util.tty as tty +from llnl.util.tty.colify import colify from llnl.util.filesystem import mkdirp, join_path import spack import spack.cmd +import spack.config +from spack.spec import Spec +from spack.error import SpackError from spack.stage import Stage +from spack.util.compression import extension -description = "Create a directory full of package tarballs that can be used as a spack mirror." +description = "Manage spack mirrors." def setup_parser(subparser): subparser.add_argument( - 'directory', help="Directory in which to create mirror.") - subparser.add_argument( - 'packages', nargs=argparse.REMAINDER, help="names of packages to put in mirror") + '-n', '--no-checksum', action='store_true', dest='no_checksum', + help="Do not check fetched packages against checksum") + sp = subparser.add_subparsers( + metavar='SUBCOMMAND', dest='mirror_command') -def mirror(parser, args): - if not args.packages: - args.packages = [p for p in spack.db.all_package_names()] + create_parser = sp.add_parser('create', help=mirror_create.__doc__) + create_parser.add_argument('-d', '--directory', default=None, + help="Directory in which to create mirror.") + create_parser.add_argument( + 'specs', nargs=argparse.REMAINDER, help="Specs of packages to put in mirror") + create_parser.add_argument( + '-f', '--file', help="File with specs of packages to put in mirror.") + + add_parser = sp.add_parser('add', help=mirror_add.__doc__) + add_parser.add_argument('name', help="Mnemonic name for mirror.") + add_parser.add_argument( + 'url', help="URL of mirror directory created by 'spack mirror create'.") + + remove_parser = sp.add_parser('remove', help=mirror_remove.__doc__) + remove_parser.add_argument('name') + + list_parser = sp.add_parser('list', help=mirror_list.__doc__) + + +def mirror_add(args): + """Add a mirror to Spack.""" + config = spack.config.get_config('user') + config.set_value('mirror', args.name, 'url', args.url) + config.write() + + +def mirror_remove(args): + """Remove a mirror by name.""" + config = spack.config.get_config('user') + name = args.name + + if not config.has_named_section('mirror', name): + tty.die("No such mirror: %s" % name) + config.remove_named_section('mirror', name) + config.write() + + +def mirror_list(args): + """Print out available mirrors to the console.""" + config = spack.config.get_config() + sec_names = config.get_section_names('mirror') + if not sec_names: + tty.msg("No mirrors configured.") + return + + max_len = max(len(s) for s in sec_names) + fmt = "%%-%ds%%s" % (max_len + 4) + + for name in sec_names: + val = config.get_value('mirror', name, 'url') + print fmt % (name, val) + + +def mirror_create(args): + """Create a directory to be used as a spack mirror, and fill it with + package archives.""" + # try to parse specs from the command line first. + args.specs = spack.cmd.parse_specs(args.specs) + + # If there is a file, parse each line as a spec and add it to the list. + if args.file: + with closing(open(args.file, "r")) as stream: + for i, string in enumerate(stream): + try: + s = Spec(string) + s.package + args.specs.append(s) + except SpackError, e: + tty.die("Parse error in %s, line %d:" % (args.file, i+1), + ">>> " + string, str(e)) + + if not args.specs: + args.specs = spack.db.all_package_names() + + # Default name for directory is spack-mirror-<DATESTAMP> + if not args.directory: + timestamp = datetime.now().strftime("%Y-%m-%d") + args.directory = 'spack-mirror-' + timestamp + + # Make sure nothing is in the way. if os.path.isfile(args.directory): tty.error("%s already exists and is a file." % args.directory) + # Create a directory if none exists if not os.path.isdir(args.directory): mkdirp(args.directory) + tty.msg("Created new mirror in %s" % args.directory) + else: + tty.msg("Adding to existing mirror in %s" % args.directory) - # save working directory + # Things to keep track of while parsing specs. working_dir = os.getcwd() + num_mirrored = 0 + num_error = 0 # Iterate through packages and download all the safe tarballs for each of them - for pkg_name in args.packages: - pkg = spack.db.get(pkg_name) + for spec in args.specs: + pkg = spec.package # Skip any package that has no checksummed versions. if not pkg.versions: - tty.msg("No safe (checksummed) versions for package %s. Skipping." - % pkg_name) + tty.msg("No safe (checksummed) versions for package %s." + % pkg.name) continue # create a subdir for the current package. - pkg_path = join_path(args.directory, pkg_name) + pkg_path = join_path(args.directory, pkg.name) mkdirp(pkg_path) # Download all the tarballs using Stages, then move them into place for version in pkg.versions: + # Skip versions that don't match the spec + vspec = Spec('%s@%s' % (pkg.name, version)) + if not vspec.satisfies(spec): + continue + + mirror_path = "%s/%s-%s.%s" % ( + pkg.name, pkg.name, version, extension(pkg.url)) + + os.chdir(working_dir) + mirror_file = join_path(args.directory, mirror_path) + if os.path.exists(mirror_file): + tty.msg("Already fetched %s." % mirror_file) + num_mirrored += 1 + continue + + # Get the URL for the version and set up a stage to download it. url = pkg.url_for_version(version) stage = Stage(url) try: + # fetch changes directory into the stage stage.fetch() - basename = os.path.basename(stage.archive_file) - final_dst = join_path(pkg_path, basename) + if not args.no_checksum and version in pkg.versions: + digest = pkg.versions[version] + stage.check(digest) + tty.msg("Checksum passed for %s@%s" % (pkg.name, version)) + + # change back and move the new archive into place. os.chdir(working_dir) - shutil.move(stage.archive_file, final_dst) - tty.msg("Added %s to mirror" % final_dst) + shutil.move(stage.archive_file, mirror_file) + tty.msg("Added %s to mirror" % mirror_file) + num_mirrored += 1 except Exception, e: - tty.warn("Error while fetching %s. Skipping." % url, e.message) + tty.warn("Error while fetching %s." % url, e.message) + num_error += 1 finally: stage.destroy() - # Success! - tty.msg("Created Spack mirror in %s" % args.directory) + # If nothing happened, try to say why. + if not num_mirrored: + if num_error: + tty.error("No packages added to mirror.", + "All packages failed to fetch.") + else: + tty.error("No packages added to mirror. No versions matched specs:") + colify(args.specs, indent=4) + + +def mirror(parser, args): + action = { 'create' : mirror_create, + 'add' : mirror_add, + 'remove' : mirror_remove, + 'list' : mirror_list } + action[args.mirror_command](args) diff --git a/lib/spack/spack/config.py b/lib/spack/spack/config.py index 00ff4313a2..5494adc324 100644 --- a/lib/spack/spack/config.py +++ b/lib/spack/spack/config.py @@ -225,8 +225,7 @@ class SpackConfigParser(cp.RawConfigParser): OPTCRE_NV = re.compile(r'\s*' + cp.RawConfigParser.OPTCRE_NV.pattern) def __init__(self, file_or_files): - cp.RawConfigParser.__init__( - self, dict_type=OrderedDict, allow_no_value=True) + cp.RawConfigParser.__init__(self, dict_type=OrderedDict) if isinstance(file_or_files, basestring): self.read([file_or_files]) @@ -286,6 +285,16 @@ class SpackConfigParser(cp.RawConfigParser): return self.has_option(sn, option) + def has_named_section(self, section, name): + sn = _make_section_name(section, name) + return self.has_section(sn) + + + def remove_named_section(self, section, name): + sn = _make_section_name(section, name) + self.remove_section(sn) + + def get_section_names(self, sectype): """Get all named sections with the specified type. A named section looks like this: diff --git a/lib/spack/spack/package.py b/lib/spack/spack/package.py index ef8b639adf..da599037dd 100644 --- a/lib/spack/spack/package.py +++ b/lib/spack/spack/package.py @@ -50,11 +50,10 @@ import spack.spec import spack.error import spack.build_environment as build_env import spack.url as url -import spack.util.crypto as crypto from spack.version import * from spack.stage import Stage from spack.util.web import get_pages -from spack.util.compression import allowed_archive +from spack.util.compression import allowed_archive, extension """Allowed URL schemes for spack packages.""" _ALLOWED_URL_SCHEMES = ["http", "https", "ftp", "file"] @@ -399,7 +398,9 @@ class Package(object): raise ValueError("Can only get a stage for a concrete package.") if self._stage is None: - mirror_path = "%s/%s" % (self.name, os.path.basename(self.url)) + # TODO: move this logic into a mirror module. + mirror_path = "%s/%s" % (self.name, "%s-%s.%s" % ( + self.name, self.version, extension(self.url))) self._stage = Stage( self.url, mirror_path=mirror_path, name=self.spec.short_spec) return self._stage @@ -537,7 +538,7 @@ class Package(object): raise ValueError("Can only fetch concrete packages.") if spack.do_checksum and not self.version in self.versions: - raise ChecksumError( + raise FetchError( "Cannot fetch %s safely; there is no checksum on file for version %s." % (self.name, self.version), "Add a checksum to the package file, or use --no-checksum to " @@ -547,13 +548,8 @@ class Package(object): if spack.do_checksum and self.version in self.versions: digest = self.versions[self.version] - checker = crypto.Checker(digest) - if checker.check(self.stage.archive_file): - tty.msg("Checksum passed for %s" % self.name) - else: - raise ChecksumError( - "%s checksum failed for %s." % (checker.hash_name, self.name), - "Expected %s but got %s." % (digest, checker.sum)) + self.stage.check(digest) + tty.msg("Checksum passed for %s@%s" % (self.name, self.version)) def do_stage(self): @@ -866,12 +862,6 @@ class FetchError(spack.error.SpackError): super(FetchError, self).__init__(message, long_msg) -class ChecksumError(FetchError): - """Raised when archive fails to checksum.""" - def __init__(self, message, long_msg): - super(ChecksumError, self).__init__(message, long_msg) - - class InstallError(spack.error.SpackError): """Raised when something goes wrong during install or uninstall.""" def __init__(self, message, long_msg=None): diff --git a/lib/spack/spack/stage.py b/lib/spack/spack/stage.py index 8574df71e9..839555d630 100644 --- a/lib/spack/spack/stage.py +++ b/lib/spack/spack/stage.py @@ -31,9 +31,12 @@ import llnl.util.tty as tty from llnl.util.filesystem import * import spack -import spack.error as serr +import spack.config +import spack.error +import spack.util.crypto as crypto from spack.util.compression import decompressor_for + STAGE_PREFIX = 'spack-stage-' @@ -185,9 +188,13 @@ class Stage(object): @property def archive_file(self): """Path to the source archive within this stage directory.""" - path = os.path.join(self.path, os.path.basename(self.url)) - if os.path.exists(path): - return path + paths = [os.path.join(self.path, os.path.basename(self.url))] + if self.mirror_path: + paths.append(os.path.join(self.path, os.path.basename(self.mirror_path))) + + for path in paths: + if os.path.exists(path): + return path return None @@ -247,6 +254,7 @@ class Stage(object): "'spack clean --dist' to remove the bad archive, then fix", "your internet gateway issue and install again.") + def fetch(self): """Downloads the file at URL to the stage. Returns true if it was downloaded, false if it already existed.""" @@ -257,7 +265,7 @@ class Stage(object): else: urls = [self.url] if self.mirror_path: - urls = ["%s/%s" % (m, self.mirror_path) for m in spack.mirrors] + urls + urls = ["%s/%s" % (m, self.mirror_path) for m in _get_mirrors()] + urls for url in urls: tty.msg("Trying to fetch from %s" % url) @@ -271,6 +279,15 @@ class Stage(object): return self.archive_file + def check(self, digest): + """Check the downloaded archive against a checksum digest""" + checker = crypto.Checker(digest) + if not checker.check(self.archive_file): + raise ChecksumError( + "%s checksum failed for %s." % (checker.hash_name, self.archive_file), + "Expected %s but got %s." % (digest, checker.sum)) + + def expand_archive(self): """Changes to the stage directory and attempt to expand the downloaded archive. Fail if the stage is not set up or if the archive is not yet @@ -320,6 +337,17 @@ class Stage(object): os.chdir(os.path.dirname(self.path)) +def _get_mirrors(): + """Get mirrors from spack configuration.""" + config = spack.config.get_config() + + mirrors = [] + sec_names = config.get_section_names('mirror') + for name in sec_names: + mirrors.append(config.get_value('mirror', name, 'url')) + return mirrors + + def ensure_access(file=spack.stage_path): """Ensure we can access a directory and die with an error if we can't.""" if not can_access(file): @@ -366,9 +394,15 @@ def find_tmp_root(): return None -class FailedDownloadError(serr.SpackError): +class FailedDownloadError(spack.error.SpackError): """Raised wen a download fails.""" def __init__(self, url, msg=""): super(FailedDownloadError, self).__init__( "Failed to fetch file from URL: %s" % url, msg) self.url = url + + +class ChecksumError(spack.error.SpackError): + """Raised when archive fails to checksum.""" + def __init__(self, message, long_msg): + super(ChecksumError, self).__init__(message, long_msg) diff --git a/lib/spack/spack/util/compression.py b/lib/spack/spack/util/compression.py index 427a9cf020..7ce8e8c65b 100644 --- a/lib/spack/spack/util/compression.py +++ b/lib/spack/spack/util/compression.py @@ -56,3 +56,12 @@ def stem(path): if re.search(suffix, path): return re.sub(suffix, "", path) return path + + +def extension(path): + """Get the archive extension for a path.""" + for type in ALLOWED_ARCHIVE_TYPES: + suffix = r'\.%s$' % type + if re.search(suffix, path): + return type + return None |