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