1
# Copyright (C) 2005, 2006, 2007, 2009, 2010 Canonical Ltd
1
# Copyright (C) 2005, 2006 Canonical Ltd
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20
def shellcomplete(context=None, outfile=None):
20
def shellcomplete(context=None, outfile = None):
21
21
if outfile is None:
22
22
outfile = sys.stdout
23
23
if context is None:
24
shellcomplete_commands(outfile=outfile)
24
shellcomplete_commands(outfile = outfile)
26
shellcomplete_on_command(context, outfile=outfile)
26
shellcomplete_on_command(context, outfile = outfile)
29
29
def shellcomplete_on_command(cmdname, outfile=None):
33
33
outfile = sys.stdout
35
35
from inspect import getdoc
36
from . import commands
37
37
cmdobj = commands.get_cmd_object(cmdname)
39
39
doc = getdoc(cmdobj)
41
raise NotImplementedError(
42
"sorry, no detailed shellcomplete yet for %r" % cmdname)
41
raise NotImplementedError("sorry, no detailed shellcomplete yet for %r" % cmdname)
44
43
shellcomplete_on_options(cmdobj.options().values(), outfile=outfile)
45
44
for aname in cmdobj.takes_args:
49
48
def shellcomplete_on_options(options, outfile=None):
50
49
for opt in options:
51
short_name = opt.short_name()
53
51
outfile.write('"(--%s -%s)"{--%s,-%s}\n'
54
% (opt.name, short_name, opt.name, short_name))
52
% (opt.name, opt.short_name(), opt.name, opt.short_name()))
56
54
outfile.write('--%s\n' % opt.name)
59
def shellcomplete_commands(outfile=None):
57
def shellcomplete_commands(outfile = None):
60
58
"""List all commands"""
61
from . import commands
62
61
from inspect import getdoc
64
commands.install_bzr_command_hooks()
66
63
if outfile is None:
67
64
outfile = sys.stdout
70
for cmdname in commands.all_command_names():
71
cmd = commands.get_cmd_object(cmdname)
72
cmds.append((cmdname, cmd))
73
for alias in cmd.aliases:
74
cmds.append((alias, cmd))
67
for cmdname, cmdclass in commands.get_all_cmds():
68
cmds.append((cmdname, cmdclass))
69
for alias in cmdclass.aliases:
70
cmds.append((alias, cmdclass))
76
for cmdname, cmd in cmds:
72
for cmdname, cmdclass in cmds:
75
doc = getdoc(cmdclass)
81
77
outfile.write(cmdname + '\n')