Source code for validphys.scripts.vp_get

#!/usr/bin/env python3
NNPDF resource downloader. The basic syntax is

vp-get <resource_type> <resource_name>


vp-get --list

to see a list of resource types.
If the resource is already installed, a string with its name will be
printed to stdout. If not, it will be searched in the remote repositories
and installed if found.
import sys
import argparse
import logging

from reportengine.baseexceptions import ErrorWithAlternatives
from reportengine import colors
from validphys.loader import FallbackLoader as Loader, LoadFailedError

log = logging.getLogger()

[docs]class ListAction(argparse.Action): def __init__(self, *args, loader ,**kwargs): self.loader = loader super().__init__(*args, **kwargs) def __call__(self, parser, *args, **kwargs): prefix = 'download_' prefix_len = len(prefix) tps = [f'\n - {it[prefix_len:]}' for it in dir(self.loader) if it.startswith(prefix)] print(f"Available resource types:{''.join(tps)}") parser.exit()
[docs]def main(): p = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter) try: l = Loader() except Exception as e: log.error(f"Could not set up environment: {e}") sys.exit(1) p.add_argument('resource_type', help="Type of the resource to be obtained. " "See --list for a list of resource types.") p.add_argument('resource_name', help="Identifier of the resource.") p.add_argument('--list', action=ListAction, loader=l, nargs=0, help="List available resources and exit.") args = p.parse_args() tp = args.resource_type name = args.resource_name try: f = getattr(l, f'check_{tp}') except AttributeError as e: sys.exit(f"No such resource {tp}") try: res = f(name) except LoadFailedError as e: print(ErrorWithAlternatives(f"Could not find resource ({tp}): '{name}'.", name)) sys.exit("Failed to download resource.") print(repr(res))