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