Source code for validphys.scripts.vp_pdfrename

#!/usr/bin/env python
    vp-pdfrename - command line tool to rename LHAPDFs

    To obtain the PDF from an fit, simply run
    vp-pdfrename <path-to-fit> <PDF name>. Optional flags allow for the
    resulting pdf to be placed in the LHAPDF directory, as well as modifying
    various fields of the info file. In addition, it is possible to compress
    the resulting PDF also using tar archiving.

import argparse
import logging
import os
import pathlib
import shutil
import sys
import tarfile
import tempfile

import lhapdf

from reportengine import colors
from reportengine.compat import yaml

from validphys.renametools import rename_pdf

# Taking command line arguments
[docs]def process_args(): parser = argparse.ArgumentParser(description="Script to rename LHAPDFs") parser.add_argument("Source", help="Path to source PDF") parser.add_argument("Destination", help="Name of the desired PDF set output") parser.add_argument( "--author", action="append", help='''The author to be added to the PDF .info file. Apply this argument multiple times for multiple authors, quotation marks can be used for an author name containing several words, e.g "The NNPDF collaboration"''', ) parser.add_argument( "--description", help="""The set description to be added to the PDF .info file. Quotations should be used for this field.""", ) parser.add_argument( "--data-version", type=int, help="The data version to be added to the PDF .info file.", ) parser.add_argument( "--index", help="The set index to be added to the PDF .info file." ) parser.add_argument( "--reference", help="The reference to be added to the PDF .info file, usually an arXiv reference.", ) parser.add_argument( "-l", "--lhapdf_path", action="store_true", help="Place the output LHAPDF in the LHAPDF directory.", ) parser.add_argument( "-c", "--compress", action="store_true", help="Compress the resulting PDF." ) args = parser.parse_args() return args
[docs]def fixup_ref(pdf_path: pathlib.Path, field_dict): """Function to handle alterations of the info file. The argparser namespace is read in as a dictionary which is then used to write to the resulting output file. If the user did not provide a field then we revert to the pre existing field. """ pdf_name = infopath = pdf_path / f"{pdf_name}.info" with open(infopath) as f: y = yaml.YAML() res = y.load(f) # If a field entry is not provided, then we revert to the existing # field in pre-existing info file. if field_dict["author"]: # Note: bool(None) is False res["Authors"] = field_dict["author"] if field_dict["description"]: res["SetDesc"] = field_dict["description"] if field_dict["data_version"]: res["DataVersion"] = field_dict["data_version"] if field_dict["index"]: res["SetIndex"] = field_dict["index"] if field_dict["reference"]: res["Reference"] = field_dict["reference"] with open(infopath, "w") as f: y.default_flow_style = True y.dump(res, f)
[docs]def compress(lhapdf_path: pathlib.Path): """ Function to compress the resulting PDF. Dereferences are handled in order to account for possible symbolic linking of grids. """ output = + ".tar.gz" with, "w:gz", dereference=True) as tar: tar.add(str(lhapdf_path), arcname=os.path.basename(str(lhapdf_path)))
[docs]def main(): args = process_args() # We need to use the os method to avoid resolving the symlink # when we use pathlib.Path.resolve() source_path = pathlib.Path(os.path.abspath(args.Source)) pdf_name = args.Destination log = logging.getLogger() log.setLevel(logging.DEBUG) log.addHandler(colors.ColorHandler()) if args.lhapdf_path: dest_path = pathlib.Path(lhapdf.paths()[-1]) / pdf_name else: dest_path = source_path.with_name(pdf_name) if dest_path.exists(): log.error(f"Destination path {dest_path.absolute()} already exists.") sys.exit(1) if not source_path.is_dir(): log.error( f"Could not find fit. Path '{source_path.absolute()}' is not a directory." ) sys.exit(1) with tempfile.TemporaryDirectory(dir=dest_path.parent) as tmp: tmp = pathlib.Path(tmp) copied_fit = tmp / shutil.copytree(source_path, copied_fit) fixup_ref(copied_fit, vars(args)) rename_pdf(copied_fit,, pdf_name) lhapdf_path = copied_fit.with_name(pdf_name) lhapdf_path.rename(dest_path)"PDF generated and placed in {dest_path.parent}") if args.compress: from validphys.renametools import Spinner"Compressing output") with Spinner(): compress(dest_path)