325 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			325 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
#!/usr/bin/env python
 | 
						|
 | 
						|
"""Copyright 2011 Gentoo Foundation
 | 
						|
Distributed under the terms of the GNU General Public License v2
 | 
						|
"""
 | 
						|
 | 
						|
from __future__ import print_function
 | 
						|
import os
 | 
						|
 | 
						|
# Meta
 | 
						|
 | 
						|
__author__ = "Corentin Chary (iksaif)"
 | 
						|
__email__ = "corentin.chary@gmail.com"
 | 
						|
__version__ = "git"
 | 
						|
__productname__ = "euscan"
 | 
						|
__description__ = "A tool to detect new upstream releases."
 | 
						|
 | 
						|
 | 
						|
# Imports
 | 
						|
 | 
						|
import sys
 | 
						|
import getopt
 | 
						|
from errno import EINTR, EINVAL
 | 
						|
from httplib import HTTPConnection
 | 
						|
 | 
						|
from portage.output import white, yellow, turquoise, green
 | 
						|
from portage.exception import AmbiguousPackageName
 | 
						|
 | 
						|
from gentoolkit import pprinter as pp
 | 
						|
from gentoolkit.eclean.search import port_settings
 | 
						|
from gentoolkit.errors import GentoolkitException
 | 
						|
 | 
						|
from euscan import CONFIG, output
 | 
						|
 | 
						|
 | 
						|
# Globals
 | 
						|
isatty = os.environ.get('TERM') != 'dumb' and sys.stdout.isatty()
 | 
						|
isatty_stderr = os.environ.get('TERM') != 'dumb' and sys.stderr.isatty()
 | 
						|
 | 
						|
 | 
						|
def exit_helper(status):
 | 
						|
    if CONFIG["format"]:
 | 
						|
        print(output.get_formatted_output())
 | 
						|
    sys.exit(status)
 | 
						|
 | 
						|
 | 
						|
def setup_signals():
 | 
						|
    """This block ensures that ^C interrupts are handled quietly."""
 | 
						|
    import signal
 | 
						|
 | 
						|
    def exithandler(signum, frame):
 | 
						|
        signal.signal(signal.SIGINT, signal.SIG_IGN)
 | 
						|
        signal.signal(signal.SIGTERM, signal.SIG_IGN)
 | 
						|
        print()
 | 
						|
        exit_helper(EINTR)
 | 
						|
 | 
						|
    signal.signal(signal.SIGINT, exithandler)
 | 
						|
    signal.signal(signal.SIGTERM, exithandler)
 | 
						|
    signal.signal(signal.SIGPIPE, signal.SIG_DFL)
 | 
						|
 | 
						|
 | 
						|
def print_version():
 | 
						|
    """Output the version info."""
 | 
						|
    print("%s (%s) - %s" \
 | 
						|
            % (__productname__, __version__, __description__))
 | 
						|
    print()
 | 
						|
    print("Author: %s <%s>" % (__author__, __email__))
 | 
						|
    print("Copyright 2011 Gentoo Foundation")
 | 
						|
    print("Distributed under the terms of the GNU General Public License v2")
 | 
						|
 | 
						|
 | 
						|
def print_usage(_error=None, help=None):
 | 
						|
    """Print help message. May also print partial help to stderr if an
 | 
						|
    error from {'options'} is specified."""
 | 
						|
 | 
						|
    out = sys.stdout
 | 
						|
    if _error:
 | 
						|
        out = sys.stderr
 | 
						|
 | 
						|
    if not _error in ('global-options', 'packages',):
 | 
						|
        _error = None
 | 
						|
 | 
						|
    if not _error and not help:
 | 
						|
        help = 'all'
 | 
						|
 | 
						|
    if _error in ('global-options',):
 | 
						|
        output.eerror("Wrong option on command line.\n")
 | 
						|
 | 
						|
    if _error in ('packages',):
 | 
						|
        output.eerror("You need to specify exactly one package.\n")
 | 
						|
 | 
						|
    print(white("Usage:"), file=out)
 | 
						|
    if _error in ('global-options', 'packages',) or help == 'all':
 | 
						|
        print(" " + turquoise(__productname__),
 | 
						|
               yellow("[options]"),
 | 
						|
               green("<package> [<package> [...]]"), file=out)
 | 
						|
    if _error in ('global-options',) or help == 'all':
 | 
						|
        print(" " + turquoise(__productname__),
 | 
						|
               yellow("[--help, --version]"), file=out)
 | 
						|
 | 
						|
    print(file=out)
 | 
						|
    if _error in ('global-options',) or help:
 | 
						|
        print("Available ", yellow("options") + ":", file=out)
 | 
						|
        print(yellow(" -C, --nocolor") +
 | 
						|
            "                      - turn off colors on output", file=out)
 | 
						|
        print(yellow(" -q, --quiet") +
 | 
						|
            "                        - be as quiet as possible", file=out)
 | 
						|
        print(yellow(" -h, --help") +
 | 
						|
            "                         - display the help screen", file=out)
 | 
						|
        print(yellow(" -V, --version") +
 | 
						|
            "                      - display version info", file=out)
 | 
						|
        print(file=out)
 | 
						|
        print(yellow(" -1, --oneshot") +
 | 
						|
            "                      - stop as soon as a new version is found",
 | 
						|
            file=out)
 | 
						|
        print(yellow(" -b, --brute-force=<level>") +
 | 
						|
            "          - define the brute force " + yellow("<level>") +
 | 
						|
            " (default: 2)\n" +
 | 
						|
            " " * 38 + "bigger levels will generate more versions numbers\n" +
 | 
						|
            " " * 38 + "0 means disabled", file=out)
 | 
						|
        print(yellow(" -f, --format=<format>") +
 | 
						|
            "              - define the output " + yellow("<format>") +
 | 
						|
            " (available: json)", file=out)
 | 
						|
        print(yellow(" -p, --progress") +
 | 
						|
            "                     - display a progress bar", file=out)
 | 
						|
        print(yellow(" -i, --ignore-pre-release") +
 | 
						|
            " " * 11 + "- Ignore non-stable versions", file=out)
 | 
						|
        print(yellow(" -I, --ignore-pre-release-if-stable") +
 | 
						|
            " - Ignore non-stable versions only if current\n" +
 | 
						|
            " " * 38 + "version is stable", file=out)
 | 
						|
        print(yellow("     --mirror") +
 | 
						|
            "                       - use mirror:// URIs", file=out)
 | 
						|
        print(yellow("     --ebuild-uri") +
 | 
						|
            "                   - use ebuild variables in URIs", file=out)
 | 
						|
        print(file=out)
 | 
						|
 | 
						|
    if _error in ('packages',) or help:
 | 
						|
        print(green(" package") +
 | 
						|
            " " * 28 + "- the packages (or ebuilds) you want to scan",
 | 
						|
            file=out)
 | 
						|
        print(file=out)
 | 
						|
 | 
						|
        #print( "More detailed instruction can be found in",
 | 
						|
        #turquoise("`man %s`" % __productname__), file=out)
 | 
						|
 | 
						|
 | 
						|
class ParseArgsException(Exception):
 | 
						|
    """For parseArgs() -> main() communications."""
 | 
						|
    def __init__(self, value):
 | 
						|
        self.value = value
 | 
						|
 | 
						|
    def __str__(self):
 | 
						|
        return repr(self.value)
 | 
						|
 | 
						|
 | 
						|
def parse_args():
 | 
						|
    """Parse the command line arguments. Raise exceptions on
 | 
						|
    errors. Returns packages and affects the CONFIG dict.
 | 
						|
    """
 | 
						|
 | 
						|
    def option_switch(opts):
 | 
						|
        """local function for interpreting command line options
 | 
						|
        and setting options accordingly"""
 | 
						|
        return_code = True
 | 
						|
        for o, a in opts:
 | 
						|
            if o in ("-h", "--help"):
 | 
						|
                raise ParseArgsException('help')
 | 
						|
            elif o in ("-V", "--version"):
 | 
						|
                raise ParseArgsException('version')
 | 
						|
            elif o in ("-C", "--nocolor"):
 | 
						|
                CONFIG['nocolor'] = True
 | 
						|
                pp.output.nocolor()
 | 
						|
            elif o in ("-q", "--quiet"):
 | 
						|
                CONFIG['quiet'] = True
 | 
						|
                CONFIG['verbose'] = 0
 | 
						|
            elif o in ("-1", "--oneshot"):
 | 
						|
                CONFIG['oneshot'] = True
 | 
						|
            elif o in ("-b", "--brute-force"):
 | 
						|
                CONFIG['brute-force'] = int(a)
 | 
						|
            elif o in ("-v", "--verbose") and not CONFIG['quiet']:
 | 
						|
                CONFIG['verbose'] += 1
 | 
						|
            elif o in ("-f", "--format"):
 | 
						|
                CONFIG['format'] = a
 | 
						|
                CONFIG['nocolor'] = True
 | 
						|
                pp.output.nocolor()
 | 
						|
            elif o in ("-p", "--progress"):
 | 
						|
                CONFIG['progress'] = isatty_stderr
 | 
						|
            elif o in ("--mirror"):
 | 
						|
                CONFIG['mirror'] = True
 | 
						|
            elif o in ("-i", "--ignore-pre-release"):
 | 
						|
                CONFIG['ignore-pre-release'] = True
 | 
						|
            elif o in ("-I", "--ignore-pre-release-if-stable"):
 | 
						|
                CONFIG['ignore-pre-release-if-stable'] = True
 | 
						|
            elif o in ("--ebuild-uri"):
 | 
						|
                CONFIG['ebuild-uri'] = True
 | 
						|
            else:
 | 
						|
                return_code = False
 | 
						|
 | 
						|
        return return_code
 | 
						|
 | 
						|
    # here are the different allowed command line options (getopt args)
 | 
						|
    getopt_options = {'short': {}, 'long': {}}
 | 
						|
    getopt_options['short']['global'] = "hVCqv1bf:piI"
 | 
						|
    getopt_options['long']['global'] = [
 | 
						|
        "help", "version", "nocolor", "quiet", "verbose", "oneshot",
 | 
						|
        "brute-force=", "format=", "progress", "mirror", "ignore-pre-release",
 | 
						|
        "ignore-pre-release-if-stable", "ebuild-uri"
 | 
						|
    ]
 | 
						|
 | 
						|
    short_opts = getopt_options['short']['global']
 | 
						|
    long_opts = getopt_options['long']['global']
 | 
						|
    opts_mode = 'global'
 | 
						|
 | 
						|
    # apply getopts to command line, show partial help on failure
 | 
						|
    try:
 | 
						|
        opts, args = getopt.getopt(sys.argv[1:], short_opts, long_opts)
 | 
						|
    except:
 | 
						|
        raise ParseArgsException(opts_mode + '-options')
 | 
						|
 | 
						|
    # set options accordingly
 | 
						|
    option_switch(opts)
 | 
						|
 | 
						|
    if len(args) < 1:
 | 
						|
        raise ParseArgsException('packages')
 | 
						|
 | 
						|
    return args
 | 
						|
 | 
						|
 | 
						|
def main():
 | 
						|
    """Parse command line and execute all actions."""
 | 
						|
 | 
						|
    CONFIG['nocolor'] = (
 | 
						|
        port_settings["NOCOLOR"] in ('yes', 'true') or not isatty
 | 
						|
    )
 | 
						|
    if CONFIG['nocolor']:
 | 
						|
        pp.output.nocolor()
 | 
						|
 | 
						|
    # parse command line options and actions
 | 
						|
    try:
 | 
						|
        queries = parse_args()
 | 
						|
    except ParseArgsException as e:
 | 
						|
        if e.value == 'help':
 | 
						|
            print_usage(help='all')
 | 
						|
            exit_helper(0)
 | 
						|
 | 
						|
        elif e.value[:5] == 'help-':
 | 
						|
            print_usage(help=e.value[5:])
 | 
						|
            exit_helper(0)
 | 
						|
 | 
						|
        elif e.value == 'version':
 | 
						|
            print_version()
 | 
						|
            exit_helper(0)
 | 
						|
 | 
						|
        else:
 | 
						|
            print_usage(e.value)
 | 
						|
            exit_helper(EINVAL)
 | 
						|
 | 
						|
    # Importing stuff here for performance reasons
 | 
						|
    from euscan.scan import scan_upstream
 | 
						|
    from euscan.out import progress_bar
 | 
						|
 | 
						|
    if CONFIG['verbose'] > 2:
 | 
						|
        HTTPConnection.debuglevel = 1
 | 
						|
 | 
						|
    if not CONFIG["format"] and not CONFIG['quiet']:
 | 
						|
        CONFIG["progress"] = False
 | 
						|
 | 
						|
    on_progress = None
 | 
						|
    if CONFIG['progress']:
 | 
						|
        on_progress_gen = progress_bar()
 | 
						|
        on_progress = on_progress_gen.next()
 | 
						|
        on_progress(maxval=len(queries) * 100, increment=0, label="Working...")
 | 
						|
 | 
						|
    for query in queries:
 | 
						|
        if CONFIG["progress"]:
 | 
						|
            on_progress(increment=10, label=query)
 | 
						|
 | 
						|
        ret = []
 | 
						|
 | 
						|
        output.set_query(query)
 | 
						|
 | 
						|
        try:
 | 
						|
            ret = scan_upstream(query, on_progress)
 | 
						|
        except AmbiguousPackageName as e:
 | 
						|
            pkgs = e.args[0]
 | 
						|
            output.eerror("\n".join(pkgs))
 | 
						|
 | 
						|
            from os.path import basename  # To get the short name
 | 
						|
 | 
						|
            output.error(
 | 
						|
                "The short ebuild name '%s' is ambiguous. Please specify" %
 | 
						|
                    basename(pkgs[0]),
 | 
						|
                "one of the above fully-qualified ebuild names instead."
 | 
						|
            )
 | 
						|
            exit_helper(1)
 | 
						|
 | 
						|
        except GentoolkitException as err:
 | 
						|
            output.eerror('%s: %s' % (query, str(err)))
 | 
						|
            exit_helper(1)
 | 
						|
 | 
						|
        except Exception as err:
 | 
						|
            output.eerror('%s: %s' % (query, str(err)))
 | 
						|
            exit_helper(1)
 | 
						|
 | 
						|
        if not ret and not CONFIG['quiet']:
 | 
						|
            output.ewarn(
 | 
						|
                "Didn't find any new version, check package's homepage " +
 | 
						|
                "for more informations"
 | 
						|
            )
 | 
						|
 | 
						|
        if not (CONFIG['format'] or CONFIG['quiet']) and len(queries) > 1:
 | 
						|
            print("")
 | 
						|
 | 
						|
    if CONFIG['progress']:
 | 
						|
        on_progress_gen.next()
 | 
						|
        print("\n", file=sys.stderr)
 | 
						|
 | 
						|
    output.set_query(None)
 | 
						|
 | 
						|
 | 
						|
if __name__ == "__main__":
 | 
						|
    setup_signals()
 | 
						|
    main()
 | 
						|
    exit_helper(0)
 |