chiark - git - mdw - stgit/blob - stgit/main.py

   1 """Basic quilt-like functionality
   2 """
   3
   4 __copyright__ = """
   5 Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>
   6
   7 This program is free software; you can redistribute it and/or modify
   8 it under the terms of the GNU General Public License version 2 as
   9 published by the Free Software Foundation.
  10
  11 This program is distributed in the hope that it will be useful,
  12 but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 GNU General Public License for more details.
  15
  16 You should have received a copy of the GNU General Public License
  17 along with this program; if not, write to the Free Software
  18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  19 """
  20
  21 import sys, os
  22 from optparse import OptionParser, make_option
  23
  24 from stgit.utils import *
  25 from stgit import stack, git
  26 from stgit.version import version
  27 from stgit.config import config
  28 from stgit.commands.common import *
  29
  30 # The commands
  31 import stgit.commands.add
  32 import stgit.commands.applied
  33 import stgit.commands.delete
  34 import stgit.commands.diff
  35 import stgit.commands.export
  36 import stgit.commands.files
  37 import stgit.commands.init
  38 import stgit.commands.new
  39 import stgit.commands.pop
  40 import stgit.commands.push
  41 import stgit.commands.refresh
  42 import stgit.commands.resolved
  43 import stgit.commands.rm
  44 import stgit.commands.series
  45 import stgit.commands.status
  46 import stgit.commands.top
  47 import stgit.commands.unapplied
  48
  49
  50 #
  51 # The commands map
  52 #
  53 commands = {
  54     'add':      stgit.commands.add,
  55     'applied':  stgit.commands.applied,
  56     'delete':   stgit.commands.delete,
  57     'diff':     stgit.commands.diff,
  58     'export':   stgit.commands.export,
  59     'files':    stgit.commands.files,
  60     'init':     stgit.commands.init,
  61     'new':      stgit.commands.new,
  62     'pop':      stgit.commands.pop,
  63     'push':     stgit.commands.push,
  64     'refresh':  stgit.commands.refresh,
  65     'resolved': stgit.commands.resolved,
  66     'rm':       stgit.commands.rm,
  67     'series':   stgit.commands.series,
  68     'status':   stgit.commands.status,
  69     'top':      stgit.commands.top,
  70     'unapplied':stgit.commands.unapplied,
  71     }
  72
  73 def print_help():
  74     print 'usage: %s <command> [options]' % os.path.basename(sys.argv[0])
  75     print
  76     print 'commands:'
  77     print '  help        print this message'
  78
  79     cmds = commands.keys()
  80     cmds.sort()
  81     for cmd in cmds:
  82         print '  ' + cmd + ' ' * (12 - len(cmd)) + commands[cmd].help
  83
  84 #
  85 # The main function (command dispatcher)
  86 #
  87 def main():
  88     """The main function
  89     """
  90     prog = os.path.basename(sys.argv[0])
  91
  92     if len(sys.argv) < 2:
  93         print >> sys.stderr, 'Unknown command'
  94         print >> sys.stderr, \
  95               '  Try "%s help" for a list of supported commands' % prog
  96         sys.exit(1)
  97
  98     cmd = sys.argv[1]
  99
 100     if cmd in ['-h', '--help', 'help']:
 101         print_help()
 102         sys.exit(0)
 103     if cmd in ['-v', '--version']:
 104         print '%s %s' % (prog, version)
 105         sys.exit(0)
 106     if not cmd in commands:
 107         print >> sys.stderr, 'Unknown command: %s' % cmd
 108         print >> sys.stderr, '  Try "%s help" for a list of supported commands' \
 109               % prog
 110         sys.exit(1)
 111
 112     # re-build the command line arguments
 113     sys.argv[0] += ' %s' % cmd
 114     del(sys.argv[1])
 115
 116     command = commands[cmd]
 117     parser = OptionParser(usage = command.usage,
 118                           option_list = command.options)
 119     options, args = parser.parse_args()
 120     try:
 121         command.func(parser, options, args)
 122     except (IOError, CmdException, stack.StackException, git.GitException), \
 123                err:
 124         print >> sys.stderr, '%s %s: %s' % (prog, cmd, err)
 125         sys.exit(2)
 126
 127     sys.exit(0)