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