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