[Solo] implement the "solo config" command.

client/solo/commands/config.py

@@ -0,0 +1,186 @@
+# -*- coding: utf-8 -*-
+"""
+
+    @author: Fabio Erculiani <lxnay@sabayon.org>
+    @contact: lxnay@sabayon.org
+    @copyright: Fabio Erculiani
+    @license: GPL-2
+
+    B{Entropy Command Line Client}.
+
+"""
+import os
+import sys
+import argparse
+
+from entropy.i18n import _
+from entropy.output import darkred, darkgreen, blue, brown
+
+from solo.utils import enlightenatom
+from solo.commands.descriptor import SoloCommandDescriptor
+from solo.commands._manage import SoloManage
+
+class SoloConfig(SoloManage):
+    """
+    Main Solo Config command.
+    """
+
+    NAME = "config"
+    ALIASES = []
+    ALLOW_UNPRIVILEGED = False
+
+    INTRODUCTION = """\
+Configure installed packages (calling pkg_config() hook).
+"""
+    SEE_ALSO = ""
+
+    def __init__(self, args):
+        SoloManage.__init__(self, args)
+        self._commands = {}
+
+    def _get_parser(self):
+        """
+        Overridden from SoloCommand.
+        """
+        _commands = {}
+
+        descriptor = SoloCommandDescriptor.obtain_descriptor(
+            SoloConfig.NAME)
+        parser = argparse.ArgumentParser(
+            description=descriptor.get_description(),
+            formatter_class=argparse.RawDescriptionHelpFormatter,
+            prog="%s %s" % (sys.argv[0], SoloConfig.NAME))
+        parser.set_defaults(func=self._config)
+
+        parser.add_argument(
+            "packages", nargs='+',
+            metavar="<package>", help=_("package name"))
+
+        mg_group = parser.add_mutually_exclusive_group()
+        mg_group.add_argument(
+            "--ask", action="store_true",
+            default=False,
+            help=_("ask before making any changes"))
+        _commands["--ask"] = {}
+        mg_group.add_argument(
+            "--pretend", action="store_true",
+            default=False,
+            help=_("show what would be done"))
+        _commands["--pretend"] = {}
+
+        parser.add_argument(
+            "--verbose", action="store_true",
+            default=False,
+            help=_("verbose output"))
+
+        self._commands = _commands
+        return parser
+
+    def bashcomp(self, last_arg):
+        """
+        Overridden from SoloCommand.
+        """
+        self._get_parser() # this will generate self._commands
+        return self._hierarchical_bashcomp(last_arg, [], self._config)
+
+    def _config(self, entropy_client):
+        """
+        Solo config command.
+        """
+        exit_st, _show_cfgupd = self._config_action(entropy_client)
+        if _show_cfgupd:
+            self._show_config_files_update(entropy_client)
+        return exit_st
+
+    def _config_action(self, entropy_client):
+        """
+        Solo Config command action.
+        """
+        ask = self._nsargs.ask
+        pretend = self._nsargs.pretend
+        verbose = self._nsargs.verbose
+
+        packages = entropy_client.packages_expand(
+            self._nsargs.packages)
+        inst_repo = entropy_client.installed_repository()
+        package_ids = self._scan_installed_packages(
+            entropy_client, inst_repo, packages)
+
+        if not packages:
+            entropy_client.output(
+                "%s." % (
+                    darkred(_("No packages found")),),
+                level="error", importance=1)
+            return 1, False
+
+        for count, package_id in enumerate(package_ids):
+
+            count += 1 # start from zero
+            atom = inst_repo.retrieveAtom(package_id)
+            installed_from = inst_repo.getInstalledPackageRepository(
+                package_id)
+            if installed_from is None:
+                installed_from = _("Not available")
+
+            mytxt = "%s | %s: %s" % (
+                enlightenatom(atom),
+                brown(_("installed from")),
+                darkred(installed_from),
+                )
+            entropy_client.output(
+                mytxt,
+                count=(count, len(package_ids)),
+                header=darkgreen("   # "))
+
+        if verbose or ask or pretend:
+            entropy_client.output(
+                "%s: %s" % (
+                    blue(_("Packages involved")),
+                    len(package_ids),),
+                header=darkred(" @@ "))
+
+        if ask:
+            exit_st = entropy_client.ask_question(
+                question = "     %s" % (
+                    _("Would you like to continue ?"),))
+            if exit_st == _("No"):
+                return 1, False
+
+        if pretend:
+            return 0, False
+
+        for count, package_id in enumerate(package_ids):
+
+            count += 1 # start from zero
+            atom = inst_repo.retrieveAtom(package_id)
+            pkg = None
+
+            try:
+                pkg = entropy_client.Package()
+                pkg.prepare((package_id,), "config")
+
+                xterm_header = "equo (%s) :: %d of %d ::" % (
+                    _("configure"), count, len(package_ids))
+
+                entropy_client.output(
+                    darkgreen(atom),
+                    count=(count, len(package_ids)),
+                    header=darkred(" ::: ") + ">>> ")
+
+                exit_st = pkg.run(xterm_header=xterm_header)
+                if exit_st not in (0, 3):
+                    return 1, True
+
+            finally:
+                if pkg is not None:
+                    pkg.kill()
+
+        return 0, True
+
+
+SoloCommandDescriptor.register(
+    SoloCommandDescriptor(
+        SoloConfig,
+        SoloConfig.NAME,
+        _("configure installed packages"))
+    )