mpcdf_info.py

#!/usr/bin/python2
from __future__ import print_function
from functools import reduce

import mpcdf_common

import os
import osc
import osc.conf
import osc.core
import osc.cmdln


def do_mpcdf_info(self, subcmd, opts, *args):
    """${cmd_name}: Basic information about an MPCDF OBS project

    Usage:
        osc ${cmd_name} [PROJECT]

    ${cmd_option_list}

    """

    if len(args) == 0:
        if osc.core.is_project_dir(os.curdir) or osc.core.is_package_dir(os.curdir):
            project = osc.core.store_read_project(os.curdir)
        else:
            raise osc.oscerr.WrongArgs('Specify PROJECT or run command in an osc checkout directory')

    elif len(args) == 1:
        project, = args
    else:
        raise osc.oscerr.WrongArgs("Too many arguments")

    print("Project `{0}`\n".format(project))

    api_url = self.get_api_url()

    def print_attribute(description, attribute):
        if mpcdf_common.has_attribute(api_url, project, None, attribute):
            values = list(sorted(mpcdf_common.get_attribute_values(api_url, project, None, attribute)))
            print(" {0}:\n   ".format(description) + "\n   ".join(values))
            print()

    print_attribute("Compilers", "MPCDF:compiler_modules")
    print_attribute("MPI libraries", "MPCDF:mpi_modules")
    print_attribute("CUDA versions", "MPCDF:cuda_modules")

    unmanaged = []

    packages = osc.core.meta_get_packagelist(api_url, project)

    pkg_name_width = reduce(max, (len(p) for p in packages))
    pkg_name_width = max(pkg_name_width, len("Package"))

    pkg_name_fmt = "{{:{0}}}".format(pkg_name_width)

    print(" " + pkg_name_fmt.format("Package"), "Enabled repositories")
    print(" " + "-" * (pkg_name_width), "-" * len("Enabled repositories"))

    for package in packages:
        if mpcdf_common.has_attribute(api_url, project, package, "MPCDF:enable_repositories"):
            enabled_repos = mpcdf_common.get_attribute_values(api_url, project, package, "MPCDF:enable_repositories")
        else:
            unmanaged.append(package)
            continue

        def subset(description, attribute):
            if mpcdf_common.has_attribute(api_url, project, package, attribute):
                return " ({0}: {1})".format(description, ", ".join(mpcdf_common.get_attribute_values(api_url, project, package, attribute)))
            return ""

        print(" " + pkg_name_fmt.format(package), ", ".join(enabled_repos),
              subset("compiler subset", "MPCDF:compiler_modules") +
              subset("MPI subset", "MPCDF:mpi_modules") +
              subset("CUDA subet", "MPCDF:cuda_modules"))
    print()

    if unmanaged:
        print(" Unmanaged packages")
        print(" " + "-" * (1 + pkg_name_width))
        for package in unmanaged:
            print(" " + package)
        print()