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
130
131
132
133
134
135
136
137
138
139
|
#!/usr/bin/python
# vim: set fileencoding=utf-8 :
#
# (C) 2013 Guido Günther <agx@sigxcpu.org>
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""Supercommand for all gbp commands"""
import glob
import os
import re
import sys
# Command is this module and common/ is shared code
# so we don't allow these to be imported:
invalid_modules = [ 'common', 'supercommand' ]
def sanitize(cmd):
"""
'-' is not allowed in module names
so turn it into an underscore.
"""
return cmd.replace('-', '_')
def usage():
print """
Usage:
gbp <command> [<args>]
The most commonly used commands are:
buildpackage - build a Debian package
import-orig - import a new upstream tarball
import-dsc - import a single Debian source package
import-dscs - import multiple Debian source packages
Use '--list-cmds' to list all available commands.
"""
def version(prog):
try:
from gbp.version import gbp_version
except ImportError:
gbp_version = '[Unknown version]'
print("%s %s" % (os.path.basename(prog), gbp_version))
def import_command(cmd):
"""
Import the module that implements the given command
"""
modulename = sanitize(cmd)
if (not re.match(r'[a-z][a-z0-9_]', modulename) or
modulename in invalid_modules):
raise ImportError('Illegal module name %s' % modulename)
return __import__('gbp.scripts.%s' % modulename, fromlist='main', level=0)
def pymod_to_cmd(mod):
"""
>>> pymod_to_cmd('/x/y/z/a_cmd.py')
'a-cmd'
"""
return os.path.basename(mod.rsplit('.', 1)[0]).replace('_','-')
def get_available_commands(path):
cmds = []
for f in glob.glob(os.path.join(path, '*.py')):
if os.path.basename(f) in ['__init__.py', 'supercommand.py']:
continue
cmds.append((pymod_to_cmd(f), f))
return cmds
def list_available_commands():
mod = __import__('gbp.scripts', fromlist='main', level=0)
path = os.path.dirname(mod.__file__)
maxlen = 0
print("Available commands in %s\n" % path)
cmds = sorted(get_available_commands(path))
for cmd in cmds:
if len(cmd[0]) > maxlen:
maxlen = len(cmd[0])
for cmd in cmds:
mod = import_command(cmd[0])
doc = mod.__doc__
print(" %s - %s" % (cmd[0].rjust(maxlen), doc))
print('')
def supercommand(argv=None):
argv = argv or sys.argv
if len(argv) < 2:
usage()
return 1
prg, cmd = argv[0:2]
args = argv[1:]
if cmd in ['--help', '-h', 'help' ]:
usage()
return 0
elif cmd in [ '--version', 'version' ]:
version(argv[0])
return 0
elif cmd in [ '--list-cmds', 'list-cmds' ]:
list_available_commands()
return 0
try:
module = import_command(cmd)
except ImportError as e:
print >>sys.stderr, "'%s' is not a valid command." % cmd
usage()
if '--verbose' in args:
print >>sys.stderr, e
return 2
return module.main(args)
if __name__ == '__main__':
sys.exit(supercommand())
# vim:et:ts=4:sw=4:et:sts=4:ai:set list listchars=tab\:»·,trail\:·:
|
