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.clean
  36 import stgit.commands.export
  37 import stgit.commands.files
  38 import stgit.commands.init
  39 import stgit.commands.mail
  40 import stgit.commands.new
  41 import stgit.commands.pop
  42 import stgit.commands.pull
  43 import stgit.commands.push
  44 import stgit.commands.refresh
  45 import stgit.commands.rename
  46 import stgit.commands.resolved
  47 import stgit.commands.rm
  48 import stgit.commands.series
  49 import stgit.commands.status
  50 import stgit.commands.top
  51 import stgit.commands.unapplied
  52
  53
  54 #
  55 # The commands map
  56 #
  57 commands = {
  58     'add':      stgit.commands.add,
  59     'applied':  stgit.commands.applied,
  60     'delete':   stgit.commands.delete,
  61     'diff':     stgit.commands.diff,
  62     'clean':    stgit.commands.clean,
  63     'export':   stgit.commands.export,
  64     'files':    stgit.commands.files,
  65     'init':     stgit.commands.init,
  66     'mail':     stgit.commands.mail,
  67     'new':      stgit.commands.new,
  68     'pop':      stgit.commands.pop,
  69     'pull':     stgit.commands.pull,
  70     'push':     stgit.commands.push,
  71     'refresh':  stgit.commands.refresh,
  72     'rename':   stgit.commands.rename,
  73     'resolved': stgit.commands.resolved,
  74     'rm':       stgit.commands.rm,
  75     'series':   stgit.commands.series,
  76     'status':   stgit.commands.status,
  77     'top':      stgit.commands.top,
  78     'unapplied':stgit.commands.unapplied,
  79     }
  80
  81 def print_help():
  82     print 'usage: %s <command> [options]' % os.path.basename(sys.argv[0])
  83     print
  84     print 'commands:'
  85     print '  help        print this message'
  86
  87     cmds = commands.keys()
  88     cmds.sort()
  89     for cmd in cmds:
  90         print '  ' + cmd + ' ' * (12 - len(cmd)) + commands[cmd].help
  91
  92 #
  93 # The main function (command dispatcher)
  94 #
  95 def main():
  96     """The main function
  97     """
  98     prog = os.path.basename(sys.argv[0])
  99
 100     if len(sys.argv) < 2:
 101         print >> sys.stderr, 'Unknown command'
 102         print >> sys.stderr, \
 103               '  Try "%s help" for a list of supported commands' % prog
 104         sys.exit(1)
 105
 106     cmd = sys.argv[1]
 107
 108     if cmd in ['-h', '--help', 'help']:
 109         print_help()
 110         sys.exit(0)
 111     if cmd in ['-v', '--version']:
 112         print '%s %s' % (prog, version)
 113         sys.exit(0)
 114     if not cmd in commands:
 115         print >> sys.stderr, 'Unknown command: %s' % cmd
 116         print >> sys.stderr, '  Try "%s help" for a list of supported commands' \
 117               % prog
 118         sys.exit(1)
 119
 120     # re-build the command line arguments
 121     sys.argv[0] += ' %s' % cmd
 122     del(sys.argv[1])
 123
 124     command = commands[cmd]
 125     parser = OptionParser(usage = command.usage,
 126                           option_list = command.options)
 127     options, args = parser.parse_args()
 128     try:
 129         # the lines below are a simple way to avoid an exception when
 130         # stgit is run outside an initialised tree
 131         stgit.commands.common.crt_series = stack.Series()
 132         setattr(command, 'crt_series', stgit.commands.common.crt_series)
 133
 134         command.func(parser, options, args)
 135     except (IOError, CmdException, stack.StackException, git.GitException), \
 136                err:
 137         print >> sys.stderr, '%s %s: %s' % (prog, cmd, err)
 138         sys.exit(2)
 139
 140     sys.exit(0)