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