from optparse import OptionParser
import stgit.commands
-from stgit.stack import Series, StackException
-from stgit.git import GitException
-from stgit.commands.common import CmdException
-from stgit.gitmergeonefile import GitMergeException
+from stgit.out import *
+from stgit import utils
#
# The commands map
class Commands(dict):
"""Commands class. It performs on-demand module loading
"""
+ def canonical_cmd(self, key):
+ """Return the canonical name for a possibly-shortenned
+ command name.
+ """
+ candidates = [cmd for cmd in self.keys() if cmd.startswith(key)]
+
+ if not candidates:
+ out.error('Unknown command: %s' % key,
+ 'Try "%s help" for a list of supported commands' % prog)
+ sys.exit(utils.STGIT_GENERAL_ERROR)
+ elif len(candidates) > 1:
+ out.error('Ambiguous command: %s' % key,
+ 'Candidates are: %s' % ', '.join(candidates))
+ sys.exit(utils.STGIT_GENERAL_ERROR)
+
+ return candidates[0]
+
def __getitem__(self, key):
- cmd_mod = self.get(key)
+ """Return the command python module name based.
+ """
+ global prog
+
+ cmd_mod = self.get(key) or self.get(self.canonical_cmd(key))
+
__import__('stgit.commands.' + cmd_mod)
return getattr(stgit.commands, cmd_mod)
commands = Commands({
'add': 'add',
'applied': 'applied',
- 'assimilate': 'assimilate',
'branch': 'branch',
'delete': 'delete',
'diff': 'diff',
'clean': 'clean',
'clone': 'clone',
'commit': 'commit',
+ 'cp': 'copy',
+ 'edit': 'edit',
'export': 'export',
'files': 'files',
'float': 'float',
'fold': 'fold',
'goto': 'goto',
+ 'hide': 'hide',
'id': 'id',
'import': 'imprt',
'init': 'init',
'pop': 'pop',
'pull': 'pull',
'push': 'push',
+ 'rebase': 'rebase',
'refresh': 'refresh',
'rename': 'rename',
+ 'repair': 'repair',
'resolved': 'resolved',
'rm': 'rm',
'series': 'series',
'show': 'show',
+ 'sink': 'sink',
'status': 'status',
+ 'sync': 'sync',
'top': 'top',
'unapplied': 'unapplied',
- 'uncommit': 'uncommit'
+ 'uncommit': 'uncommit',
+ 'unhide': 'unhide'
})
# classification: repository, stack, patch, working copy
repocommands = (
- 'branch',
'clone',
'id',
- 'pull'
)
stackcommands = (
'applied',
- 'assimilate',
+ 'branch',
'clean',
'commit',
'float',
'goto',
+ 'hide',
'init',
+ 'patches',
'pop',
+ 'pull',
'push',
+ 'rebase',
+ 'repair',
'series',
+ 'sink',
'top',
'unapplied',
- 'uncommit'
+ 'uncommit',
+ 'unhide',
)
patchcommands = (
'delete',
+ 'edit',
'export',
'files',
'fold',
'pick',
'refresh',
'rename',
- 'show'
+ 'show',
+ 'sync',
)
wccommands = (
'add',
+ 'cp',
'diff',
- 'patches',
'resolved',
'rm',
- 'status'
+ 'status',
)
def _print_helpstring(cmd):
def main():
"""The main function
"""
+ global prog
+
prog = os.path.basename(sys.argv[0])
if len(sys.argv) < 2:
print >> sys.stderr, 'usage: %s <command>' % prog
print >> sys.stderr, \
' Try "%s --help" for a list of supported commands' % prog
- sys.exit(1)
+ sys.exit(utils.STGIT_GENERAL_ERROR)
cmd = sys.argv[1]
if cmd in ['-h', '--help']:
- if len(sys.argv) >= 3 and sys.argv[2] in commands:
- cmd = sys.argv[2]
+ if len(sys.argv) >= 3:
+ cmd = commands.canonical_cmd(sys.argv[2])
sys.argv[2] = '--help'
else:
print_help()
- sys.exit(0)
+ sys.exit(utils.STGIT_SUCCESS)
if cmd == 'help':
if len(sys.argv) == 3 and not sys.argv[2] in ['-h', '--help']:
- cmd = sys.argv[2]
+ cmd = commands.canonical_cmd(sys.argv[2])
if not cmd in commands:
- print >> sys.stderr, '%s help: "%s" command unknown' \
- % (prog, cmd)
- sys.exit(1)
+ out.error('%s help: "%s" command unknown' % (prog, cmd))
+ sys.exit(utils.STGIT_GENERAL_ERROR)
sys.argv[0] += ' %s' % cmd
command = commands[cmd]
parser = OptionParser(usage = command.usage,
option_list = command.options)
- parser.print_help()
+ from pydoc import pager
+ pager(parser.format_help())
else:
print_help()
- sys.exit(0)
+ sys.exit(utils.STGIT_SUCCESS)
if cmd in ['-v', '--version', 'version']:
from stgit.version import version
print 'Stacked GIT %s' % version
os.system('git --version')
print 'Python version %s' % sys.version
- sys.exit(0)
+ sys.exit(utils.STGIT_SUCCESS)
if cmd in ['copyright']:
print __copyright__
- sys.exit(0)
- if not cmd in commands:
- print >> sys.stderr, 'Unknown command: %s' % cmd
- print >> sys.stderr, ' Try "%s help" for a list of supported ' \
- 'commands' % prog
- sys.exit(1)
+ sys.exit(utils.STGIT_SUCCESS)
# re-build the command line arguments
+ cmd = commands.canonical_cmd(cmd)
sys.argv[0] += ' %s' % cmd
del(sys.argv[1])
usage = command.usage.split('\n')[0].strip()
parser = OptionParser(usage = usage, option_list = command.options)
options, args = parser.parse_args()
+ directory = command.directory
+
+ # These modules are only used from this point onwards and do not
+ # need to be imported earlier
+ from stgit.exception import StgException
+ from stgit.config import config_setup
+ from ConfigParser import ParsingError, NoSectionError
+ from stgit.stack import Series
+
try:
- # 'clone' doesn't expect an already initialised GIT tree. A Series
- # object will be created after the GIT tree is cloned
- if cmd != 'clone':
+ debug_level = int(os.environ.get('STGIT_DEBUG_LEVEL', 0))
+ except ValueError:
+ out.error('Invalid STGIT_DEBUG_LEVEL environment variable')
+ sys.exit(utils.STGIT_GENERAL_ERROR)
+
+ try:
+ directory.setup()
+ config_setup()
+
+ # Some commands don't (always) need an initialized series.
+ if directory.needs_current_series:
if hasattr(options, 'branch') and options.branch:
command.crt_series = Series(options.branch)
else:
command.crt_series = Series()
- stgit.commands.common.crt_series = command.crt_series
command.func(parser, options, args)
- except (IOError, CmdException, StackException, GitException,
- GitMergeException), err:
- print >> sys.stderr, '%s %s: %s' % (prog, cmd, err)
- sys.exit(2)
+ except (StgException, IOError, ParsingError, NoSectionError), err:
+ out.error(str(err), title = '%s %s' % (prog, cmd))
+ if debug_level > 0:
+ raise
+ else:
+ sys.exit(utils.STGIT_COMMAND_ERROR)
except KeyboardInterrupt:
- sys.exit(1)
+ sys.exit(utils.STGIT_GENERAL_ERROR)
- sys.exit(0)
+ sys.exit(utils.STGIT_SUCCESS)
~mdw / stgit / blobdiff