Files
entropy/server/eit/commands/graph.py

106 lines
3.0 KiB
Python

# -*- coding: utf-8 -*-
"""
@author: Fabio Erculiani <lxnay@sabayon.org>
@contact: lxnay@sabayon.org
@copyright: Fabio Erculiani
@license: GPL-2
B{Entropy Infrastructure Toolkit}.
"""
import sys
import argparse
import functools
from entropy.i18n import _
from eit.commands.descriptor import EitCommandDescriptor
from eit.commands.command import EitCommand
from eit.utils import graph_packages
class EitGraph(EitCommand):
"""
Main Eit graph command.
"""
NAME = "graph"
ALIASES = []
ALLOW_UNPRIVILEGED = True
def __init__(self, args):
EitCommand.__init__(self, args)
self._packages = []
self._complete = False
self._quiet = False
self._repository_id = None
def _get_parser(self):
descriptor = EitCommandDescriptor.obtain_descriptor(
EitGraph.NAME)
parser = argparse.ArgumentParser(
description=descriptor.get_description(),
formatter_class=argparse.RawDescriptionHelpFormatter,
prog="%s %s" % (sys.argv[0], EitGraph.NAME))
parser.add_argument("packages", nargs='+', metavar="<package>",
help=_("package name"))
parser.add_argument("--in", metavar="<repository>",
help=_("search packages in given repository"),
dest="inrepo", default=None)
parser.add_argument("--complete", action="store_true",
default=self._complete,
help=_('show system packages, build deps, circular deps'))
parser.add_argument("--quiet", "-q", action="store_true",
default=self._quiet,
help=_('quiet output, for scripting purposes'))
return parser
INTRODUCTION = """\
Show direct dependency graph (printed as tree, actually) for given
package dependencies.
For a reverse dependency graph, please see *eit revgraph*.
"""
SEE_ALSO = "eit-revgraph(1)"
def man(self):
"""
Overridden from EitCommand.
"""
return self._man()
def parse(self):
parser = self._get_parser()
try:
nsargs = parser.parse_args(self._args)
except IOError:
return functools.partial(self.print_help, parser), []
self._quiet = nsargs.quiet
self._packages += nsargs.packages
self._complete = nsargs.complete
self._repository_id = nsargs.inrepo
return self._call_unlocked, [self._graph, self._repository_id]
def _graph(self, entropy_server):
"""
Actual Eit graph code.
"""
if self._repository_id is None:
repository_ids = entropy_server.repositories()
else:
repository_ids = [self._repository_id]
return graph_packages(
self._packages, entropy_server,
complete = self._complete,
repository_ids = repository_ids, quiet = self._quiet)
EitCommandDescriptor.register(
EitCommandDescriptor(
EitGraph,
EitGraph.NAME,
_('show dependency graph for packages'))
)