diff options
Diffstat (limited to 'pym/gentoolkit/equery/hasuse.py')
| -rw-r--r-- | pym/gentoolkit/equery/hasuse.py | 272 |
1 files changed, 140 insertions, 132 deletions
diff --git a/pym/gentoolkit/equery/hasuse.py b/pym/gentoolkit/equery/hasuse.py index 07207bc..13fbb24 100644 --- a/pym/gentoolkit/equery/hasuse.py +++ b/pym/gentoolkit/equery/hasuse.py @@ -4,7 +4,7 @@ """List all installed packages that have a given USE flag""" -__docformat__ = 'epytext' +__docformat__ = "epytext" # ======= # Imports @@ -24,153 +24,161 @@ from gentoolkit.query import Query # ======= QUERY_OPTS = { - "in_installed": True, - "in_porttree": False, - "in_overlay": False, - "include_masked": True, - "show_progress": False, - "package_format": None + "in_installed": True, + "in_porttree": False, + "in_overlay": False, + "include_masked": True, + "show_progress": False, + "package_format": None, } # ========= # Functions # ========= + def print_help(with_description=True): - """Print description, usage and a detailed help message. - - @type with_description: bool - @param with_description: if true, print module's __doc__ string - """ - - if with_description: - print(__doc__.strip()) - print() - print(mod_usage(mod_name="hasuse", arg="USE-flag")) - print() - print(pp.command("options")) - print(format_options(( - (" -h, --help", "display this help message"), - (" -I, --exclude-installed", - "exclude installed packages from search path"), - (" -o, --overlay-tree", "include overlays in search path"), - (" -p, --portage-tree", "include entire portage tree in search path"), - (" -F, --format=TMPL", "specify a custom output format"), - (" TMPL", - "a format template using (see man page):") - ))) - print(" " * 24, ', '.join(pp.emph(x) for x in FORMAT_TMPL_VARS)) + """Print description, usage and a detailed help message. + + @type with_description: bool + @param with_description: if true, print module's __doc__ string + """ + + if with_description: + print(__doc__.strip()) + print() + print(mod_usage(mod_name="hasuse", arg="USE-flag")) + print() + print(pp.command("options")) + print( + format_options( + ( + (" -h, --help", "display this help message"), + ( + " -I, --exclude-installed", + "exclude installed packages from search path", + ), + (" -o, --overlay-tree", "include overlays in search path"), + (" -p, --portage-tree", "include entire portage tree in search path"), + (" -F, --format=TMPL", "specify a custom output format"), + (" TMPL", "a format template using (see man page):"), + ) + ) + ) + print(" " * 24, ", ".join(pp.emph(x) for x in FORMAT_TMPL_VARS)) def display_useflags(query, pkg): - """Display USE flag information for a given package.""" - - try: - useflags = [x.lstrip("+-") for x in pkg.environment("IUSE").split()] - except errors.GentoolkitFatalError: - # aux_get KeyError or other unexpected result - return False - - if query not in useflags: - return False - - if CONFIG['verbose']: - pkgstr = PackageFormatter( - pkg, - do_format=True, - custom_format=QUERY_OPTS["package_format"] - ) - else: - pkgstr = PackageFormatter( - pkg, - do_format=False, - custom_format=QUERY_OPTS["package_format"] - ) - - if (QUERY_OPTS["in_installed"] and - not QUERY_OPTS["in_porttree"] and - not QUERY_OPTS["in_overlay"]): - if not 'I' in pkgstr.location: - return False - if (QUERY_OPTS["in_porttree"] and - not QUERY_OPTS["in_overlay"]): - if not 'P' in pkgstr.location: - return False - if (QUERY_OPTS["in_overlay"] and - not QUERY_OPTS["in_porttree"]): - if not 'O' in pkgstr.location: - return False - pp.uprint(pkgstr) - - return True - + """Display USE flag information for a given package.""" + + try: + useflags = [x.lstrip("+-") for x in pkg.environment("IUSE").split()] + except errors.GentoolkitFatalError: + # aux_get KeyError or other unexpected result + return False + + if query not in useflags: + return False + + if CONFIG["verbose"]: + pkgstr = PackageFormatter( + pkg, do_format=True, custom_format=QUERY_OPTS["package_format"] + ) + else: + pkgstr = PackageFormatter( + pkg, do_format=False, custom_format=QUERY_OPTS["package_format"] + ) + + if ( + QUERY_OPTS["in_installed"] + and not QUERY_OPTS["in_porttree"] + and not QUERY_OPTS["in_overlay"] + ): + if not "I" in pkgstr.location: + return False + if QUERY_OPTS["in_porttree"] and not QUERY_OPTS["in_overlay"]: + if not "P" in pkgstr.location: + return False + if QUERY_OPTS["in_overlay"] and not QUERY_OPTS["in_porttree"]: + if not "O" in pkgstr.location: + return False + pp.uprint(pkgstr) + + return True def parse_module_options(module_opts): - """Parse module options and update QUERY_OPTS""" - - # Parse module options - opts = (x[0] for x in module_opts) - posargs = (x[1] for x in module_opts) - for opt, posarg in zip(opts, posargs): - if opt in ('-h', '--help'): - print_help() - sys.exit(0) - elif opt in ('-I', '--exclue-installed'): - QUERY_OPTS['in_installed'] = False - elif opt in ('-p', '--portage-tree'): - QUERY_OPTS['in_porttree'] = True - elif opt in ('-o', '--overlay-tree'): - QUERY_OPTS['in_overlay'] = True - elif opt in ('-F', '--format'): - QUERY_OPTS["package_format"] = posarg + """Parse module options and update QUERY_OPTS""" + + # Parse module options + opts = (x[0] for x in module_opts) + posargs = (x[1] for x in module_opts) + for opt, posarg in zip(opts, posargs): + if opt in ("-h", "--help"): + print_help() + sys.exit(0) + elif opt in ("-I", "--exclue-installed"): + QUERY_OPTS["in_installed"] = False + elif opt in ("-p", "--portage-tree"): + QUERY_OPTS["in_porttree"] = True + elif opt in ("-o", "--overlay-tree"): + QUERY_OPTS["in_overlay"] = True + elif opt in ("-F", "--format"): + QUERY_OPTS["package_format"] = posarg def main(input_args): - """Parse input and run the program""" - - short_opts = "hiIpoF:" # -i was option for default action - # --installed is no longer needed, kept for compatibility (djanderson '09) - long_opts = ('help', 'installed', 'exclude-installed', 'portage-tree', - 'overlay-tree', 'format=') - - try: - module_opts, queries = gnu_getopt(input_args, short_opts, long_opts) - except GetoptError as err: - sys.stderr.write(pp.error("Module %s" % err)) - print() - print_help(with_description=False) - sys.exit(2) - - parse_module_options(module_opts) - - if not queries: - print_help() - sys.exit(2) - - matches = Query("*").smart_find(**QUERY_OPTS) - matches.sort() - - # - # Output - # - - first_run = True - got_match = False - for query in queries: - if not first_run: - print() - - if CONFIG['verbose']: - pp.uprint(" * Searching for USE flag %s ... " % pp.emph(query)) - - for pkg in matches: - if display_useflags(query, pkg): - got_match = True - - first_run = False + """Parse input and run the program""" + + short_opts = "hiIpoF:" # -i was option for default action + # --installed is no longer needed, kept for compatibility (djanderson '09) + long_opts = ( + "help", + "installed", + "exclude-installed", + "portage-tree", + "overlay-tree", + "format=", + ) + + try: + module_opts, queries = gnu_getopt(input_args, short_opts, long_opts) + except GetoptError as err: + sys.stderr.write(pp.error("Module %s" % err)) + print() + print_help(with_description=False) + sys.exit(2) + + parse_module_options(module_opts) + + if not queries: + print_help() + sys.exit(2) + + matches = Query("*").smart_find(**QUERY_OPTS) + matches.sort() + + # + # Output + # + + first_run = True + got_match = False + for query in queries: + if not first_run: + print() + + if CONFIG["verbose"]: + pp.uprint(" * Searching for USE flag %s ... " % pp.emph(query)) + + for pkg in matches: + if display_useflags(query, pkg): + got_match = True + + first_run = False + + if not got_match: + sys.exit(1) - if not got_match: - sys.exit(1) # vim: set ts=4 sw=4 tw=79: |