aboutsummaryrefslogtreecommitdiff
blob: 82625b61e674ef71c12458d3b1ae2b09ba824f64 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#!/usr/bin/python
#
# Copyright 2010 Brian Dolbec <brian.dolbec@gmail.com>
# Copyright(c) 2010, Gentoo Foundation
# Copyright 2003-2004 Karl Trygve Kalleberg
# Licensed under the GNU General Public License, v2
#
# $Header: $

"""Gentoo's installed packages analysis and repair tool"""


# Move to Imports section after Python 2.6 is stable


__docformat__ = 'epytext'
# version is dynamically set by distutils sdist
__version__ = "svn"
__productname__ = "analyse"
__authors__ = (
	'Brian Dolbec, <brian.dolbec@gmail.com>'

)

# make an exportable copy of the info for help output
MODULE_INFO = {
	"__docformat__": __docformat__,
	"__doc__": __doc__,
	"__version__": __version__,
	"__productname__": __productname__,
	"__authors__": __authors__

}

import errno
import sys
import time
from getopt import getopt, GetoptError

import portage
from portage import os

import gentoolkit as gen
from gentoolkit import errors
from gentoolkit import pprinter as pp
from gentoolkit.base import (initialize_configuration, split_arguments,
	parse_global_options, print_help)
from gentoolkit.formatters import format_options


NAME_MAP = {
	'a': 'analyse',
	'r': 'rebuild'
}

FORMATTED_OPTIONS = (
		("    (a)nalyse",
		"analyses the installed PKG database USE flag or keyword useage"),
		("    (r)ebuild",
		"analyses the Installed PKG database and generates files suitable"),
		("  ",
		"to replace corrupted or missing /etc/portage/package.* files")
	)

def expand_module_name(module_name):
	"""Returns one of the values of NAME_MAP or raises KeyError"""

	if module_name == 'list':
		# list is a Python builtin type, so we must rename our module
		return 'list_'
	elif module_name in NAME_MAP.values():
		return module_name
	else:
		return NAME_MAP[module_name]


def main():
	"""Parse input and run the program."""

	short_opts = "hqCNV"
	long_opts = (
		'help', 'quiet', 'nocolor', 'no-color', 'no-pipe', 'version', 'debug'
	)

	initialize_configuration()

	try:
		global_opts, args = getopt(sys.argv[1:], short_opts, long_opts)
	except GetoptError as err:
		sys.stderr.write(" \n")
		sys.stderr.write(pp.error("Global %s\n" % err))
		print_help(MODULE_INFO, FORMATTED_OPTIONS, with_description=False)
		sys.exit(2)

	# Parse global options
	need_help = parse_global_options(global_opts, args, MODULE_INFO, FORMATTED_OPTIONS)

	if gen.CONFIG['quiet']:
		gen.CONFIG['verbose'] = False

	try:
		module_name, module_args = split_arguments(args)
	except IndexError:
		print_help(MODULE_INFO,  FORMATTED_OPTIONS)
		sys.exit(2)

	if need_help:
		module_args.append('--help')

	try:
		expanded_module_name = expand_module_name(module_name)
	except KeyError:
		sys.stderr.write(pp.error("Unknown module '%s'" % module_name))
		print_help(MODULE_INFO, FORMATTED_OPTIONS, with_description=False)
		sys.exit(2)

	try:
		loaded_module = __import__(
			expanded_module_name, globals(), locals(), [], -1
		)
		loaded_module.main(module_args)
	except portage.exception.AmbiguousPackageName as err:
		raise errors.GentoolkitAmbiguousPackage(err)
	except IOError as err:
		if err.errno != errno.EPIPE:
			raise

if __name__ == '__main__':
	main()