aboutsummaryrefslogtreecommitdiff
blob: 6a402157b7c013a0eef97c0f5f68dcbb5889144e (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
#!/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

"""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__ = "git"
__productname__ = "enalyze"
__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
from getopt import getopt, GetoptError

import portage

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)


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

FORMATTED_OPTIONS = (
		("    (a)nalyze",
		"analyzes the installed PKG database USE flag or keyword useage"),
		("    (r)ebuild",
		"analyzes 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.args[0])
	except IOError as err:
		if err.errno != errno.EPIPE:
			raise

if __name__ == '__main__':
	main()