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