[stgit] / stgit / main.py

"""Basic quilt-like functionality
"""

__copyright__ = """
Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License version 2 as
published by the Free Software Foundation.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""

import sys, os
from optparse import OptionParser, make_option

from stgit.utils import *
from stgit import stack, git
from stgit.version import version
from stgit.config import config
from stgit.commands.common import *

# The commands
import stgit.commands.add
import stgit.commands.applied
import stgit.commands.delete
import stgit.commands.diff
import stgit.commands.clean
import stgit.commands.clone
import stgit.commands.export
import stgit.commands.files
import stgit.commands.imprt
import stgit.commands.init
import stgit.commands.mail
import stgit.commands.new
import stgit.commands.pop
import stgit.commands.pull
import stgit.commands.push
import stgit.commands.refresh
import stgit.commands.rename
import stgit.commands.resolved
import stgit.commands.rm
import stgit.commands.series
import stgit.commands.status
import stgit.commands.top
import stgit.commands.unapplied


#
# The commands map
#
commands = {
    'add':      stgit.commands.add,
    'applied':  stgit.commands.applied,
    'delete':   stgit.commands.delete,
    'diff':     stgit.commands.diff,
    'clean':    stgit.commands.clean,
    'clone':    stgit.commands.clone,
    'export':   stgit.commands.export,
    'files':    stgit.commands.files,
    'import':   stgit.commands.imprt,
    'init':     stgit.commands.init,
    'mail':     stgit.commands.mail,
    'new':      stgit.commands.new,
    'pop':      stgit.commands.pop,
    'pull':     stgit.commands.pull,
    'push':     stgit.commands.push,
    'refresh':  stgit.commands.refresh,
    'rename':   stgit.commands.rename,
    'resolved': stgit.commands.resolved,
    'rm':       stgit.commands.rm,
    'series':   stgit.commands.series,
    'status':   stgit.commands.status,
    'top':      stgit.commands.top,
    'unapplied':stgit.commands.unapplied,
    }

def print_help():
    print 'usage: %s <command> [options]' % os.path.basename(sys.argv[0])
    print
    print 'commands:'
    print '  help        print this message'

    cmds = commands.keys()
    cmds.sort()
    for cmd in cmds:
        print '  ' + cmd + ' ' * (12 - len(cmd)) + commands[cmd].help

#
# The main function (command dispatcher)
#
def main():
    """The main function
    """
    prog = os.path.basename(sys.argv[0])

    if len(sys.argv) < 2:
        print >> sys.stderr, 'Unknown command'
        print >> sys.stderr, \
              '  Try "%s help" for a list of supported commands' % prog
        sys.exit(1)

    cmd = sys.argv[1]

    if cmd in ['-h', '--help', 'help']:
        print_help()
        sys.exit(0)
    if cmd in ['-v', '--version']:
        print '%s %s' % (prog, version)
        sys.exit(0)
    if not cmd in commands:
        print >> sys.stderr, 'Unknown command: %s' % cmd
        print >> sys.stderr, '  Try "%s help" for a list of supported commands' \
              % prog
        sys.exit(1)

    # re-build the command line arguments
    sys.argv[0] += ' %s' % cmd
    del(sys.argv[1])

    command = commands[cmd]
    parser = OptionParser(usage = command.usage,
                          option_list = command.options)
    options, args = parser.parse_args()
    try:
        # 'clone' doesn't expect an already initialised GIT tree
        if cmd == 'clone':
            stgit.commands.common.crt_series = stack.Series('master')
        else:
            stgit.commands.common.crt_series = stack.Series()
        # the line below is a simple way to avoid an exception when
        # stgit is run outside an initialised tree
        setattr(command, 'crt_series', stgit.commands.common.crt_series)

        command.func(parser, options, args)
    except (IOError, CmdException, stack.StackException, git.GitException), \
               err:
        print >> sys.stderr, '%s %s: %s' % (prog, cmd, err)
        sys.exit(2)

    sys.exit(0)
Commit	Line	Data
41a6d859 CM	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
873a27fd	24	from stgit.utils import *
41a6d859 CM	25	from stgit import stack, git
	26	from stgit.version import version
	27	from stgit.config import config
fcee87cf CM	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
de4c9d27	35	import stgit.commands.clean
1008fbce	36	import stgit.commands.clone
fcee87cf CM	37	import stgit.commands.export
fcee87cf CM	38	import stgit.commands.files
0d2cd1e4	39	import stgit.commands.imprt
fcee87cf	40	import stgit.commands.init
b4bddc06	41	import stgit.commands.mail
fcee87cf CM	42	import stgit.commands.new
fcee87cf CM	43	import stgit.commands.pop
f338c3c0	44	import stgit.commands.pull
fcee87cf CM	45	import stgit.commands.push
fcee87cf CM	46	import stgit.commands.refresh
e55b53e0	47	import stgit.commands.rename
fcee87cf CM	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
41a6d859 CM	54
41a6d859 CM	55
41a6d859 CM	56	#
	57	# The commands map
	58	#
	59	commands = {
fcee87cf CM	60	'add': stgit.commands.add,
	61	'applied': stgit.commands.applied,
	62	'delete': stgit.commands.delete,
	63	'diff': stgit.commands.diff,
de4c9d27	64	'clean': stgit.commands.clean,
1008fbce	65	'clone': stgit.commands.clone,
fcee87cf CM	66	'export': stgit.commands.export,
fcee87cf CM	67	'files': stgit.commands.files,
0d2cd1e4	68	'import': stgit.commands.imprt,
fcee87cf	69	'init': stgit.commands.init,
b4bddc06	70	'mail': stgit.commands.mail,
fcee87cf CM	71	'new': stgit.commands.new,
fcee87cf CM	72	'pop': stgit.commands.pop,
f338c3c0	73	'pull': stgit.commands.pull,
fcee87cf CM	74	'push': stgit.commands.push,
fcee87cf CM	75	'refresh': stgit.commands.refresh,
e55b53e0	76	'rename': stgit.commands.rename,
fcee87cf CM	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,
41a6d859 CM	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	"""
41a6d859 CM	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,
fcee87cf	130	option_list = command.options)
41a6d859 CM	131	options, args = parser.parse_args()
41a6d859 CM	132	try:
1008fbce CM	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
9ac09909	139	# stgit is run outside an initialised tree
9ac09909 CM	140	setattr(command, 'crt_series', stgit.commands.common.crt_series)
9ac09909 CM	141
41a6d859	142	command.func(parser, options, args)
fcee87cf	143	except (IOError, CmdException, stack.StackException, git.GitException), \
41a6d859 CM	144	err:
	145	print >> sys.stderr, '%s %s: %s' % (prog, cmd, err)
	146	sys.exit(2)
	147
	148	sys.exit(0)