pym/gentoolkit/eshowkw/__init__.py

   1 # Copyright 2010 Gentoo Foundation
   2 # Distributed under the terms of the GNU General Public License v2
   3
   4 __package__ = 'gentoolkit.eshowkw'
   5 __version__ = 'svn'
   6 __author__ = "Tomáš Chvátal <scarabeus@gentoo.org>"
   7
   8 import portage
   9
  10 import sys, os, fnmatch
  11 import argparse
  12 from portage import output as porto
  13 from portage import settings as ports
  14 from portage import config as portc
  15 from portage import portdbapi as portdbapi
  16 from portage import db as portdb
  17
  18 from .keywords_header import keywords_header
  19 from .keywords_content import keywords_content
  20 from .display_pretty import string_rotator
  21 from .display_pretty import display
  22
  23 ignore_slots = False
  24 bold = False
  25 order = 'bottom'
  26 topper = 'versionlist'
  27
  28 def process_display(package, keywords, dbapi):
  29         portdata = keywords_content(package, keywords.keywords, dbapi, ignore_slots, order, bold, topper)
  30         if topper == 'archlist':
  31                 header = string_rotator().rotateContent(keywords.content, keywords.length, bold)
  32                 extra = string_rotator().rotateContent(keywords.extra, keywords.length, bold, False)
  33                 # -1 : space is taken in account and appended by us
  34                 filler = ''.ljust(portdata.slot_length-1)
  35                 header = ['%s%s%s' % (x, filler, y) for x, y in zip(header, extra)]
  36                 content = portdata.content
  37                 header_length = portdata.version_length
  38                 content_length = keywords.length
  39         else:
  40                 header = string_rotator().rotateContent(portdata.content, portdata.content_length, order, bold)
  41                 content = keywords.content
  42                 sep = [''.ljust(keywords.length) for x in range(portdata.slot_length-1)]
  43                 content.extend(sep)
  44                 content.extend(keywords.extra)
  45                 header_length = keywords.length
  46                 content_length = portdata.version_length
  47         display(content, header, header_length, content_length, portdata.cp, topper)
  48
  49 def process_args(argv):
  50         """Option parsing via argc"""
  51         parser = argparse.ArgumentParser(prog=__package__,
  52                 formatter_class=argparse.ArgumentDefaultsHelpFormatter,
  53                 description='Display keywords for specified package or for package that is in pwd.')
  54
  55         parser.add_argument('-v', '--version', action='version', version=__version__, help='show package version and exit')
  56
  57         parser.add_argument('package', nargs='*', default=None, help='Packages to check.')
  58
  59         parser.add_argument('-a', '--arch', nargs='+', default=[], help='Display only specified arch(s)')
  60
  61         parser.add_argument('-A', '--align', nargs='?', default='bottom', choices=['top', 'bottom'],
  62                 help='Specify alignment for descriptions.')
  63         parser.add_argument('-T', '--top-position', nargs='?', default='archlist', choices=['archlist', 'versionlist'],
  64                 help='Specify which fields we want to have in top listing.')
  65
  66         parser.add_argument('-B', '--bold', action='store_true', default=False,
  67                 help='Print out each other column in bold for easier visual separation.')
  68         parser.add_argument('-C', '--color', action='store_true', default=False,
  69                 help='Force colored output')
  70         parser.add_argument('-O', '--overlays', action='store_true', default=False,
  71                 help='Search also overlays')
  72         parser.add_argument('-P', '--prefix', action='store_true', default=False,
  73                 help='Display prefix keywords in output.')
  74         parser.add_argument('-S', '--ignore-slot', action='store_true', default=False,
  75                 help='Treat slots as irelevant during detection of redundant pacakges.')
  76
  77         return parser.parse_args(args=argv[1:])
  78
  79 def main(argv):
  80         global ignore_slots, bold, order, topper
  81
  82         #opts parsing
  83         opts = process_args(argv)
  84         ignore_slots = opts.ignore_slot
  85         use_overlays = opts.overlays
  86         # user can do both --arch=a,b,c or --arch a b c
  87         if len(opts.arch) > 1:
  88                 opts.arch = ','.join(opts.arch)
  89         highlight_arch = ''.join(opts.arch).split(',')
  90         bold = opts.bold
  91         order = opts.align
  92         topper = opts.top_position
  93         prefix = opts.prefix
  94         color = opts.color
  95         package = opts.package
  96         # disable colors when redirected and they are not forced on
  97         if not color and not sys.stdout.isatty():
  98                 # disable colors
  99                 porto.nocolor()
 100         keywords = keywords_header(prefix, highlight_arch, order)
 101         if len(package) > 0:
 102                 dbapi = portdb[ports['ROOT']]['porttree'].dbapi
 103                 if not use_overlays:
 104                         dbapi.porttrees = [dbapi.porttree_root]
 105                 map(lambda x: process_display(x, keywords, dbapi), package)
 106         else:
 107                 currdir = os.getcwd()
 108                 package=os.path.basename(currdir)
 109                 # check if there are actualy some ebuilds
 110                 ebuilds = ['%s' % x for x in os.listdir(currdir)
 111                         if fnmatch.fnmatch(x, '*.ebuild')]
 112                 if len(ebuilds) <= 0:
 113                         msg_err = 'No ebuilds at "%s"' % currdir
 114                         raise SystemExit(msg_err)
 115                 ourtree = os.path.abspath('../../')
 116
 117                 mysettings = portc(env={'PORTDIR_OVERLAY': os.path.abspath('../../')})
 118                 dbapi = portdbapi(mysettings=mysettings)
 119                 # specify that we want just our nice tree we are in cwd
 120                 dbapi.porttrees = [ourtree]
 121                 process_display(package, keywords, dbapi)
 122         return 0
 123
 124 if __name__ == '__main__':
 125         sys.exit(main(sys.argv))