1 """Basic quilt-like functionality
5 Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>
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.
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.
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
21 import sys
, os
, traceback
24 from stgit
.out
import *
25 from stgit
import argparse
, run
, utils
31 """Commands class. It performs on-demand module loading
33 def canonical_cmd(self
, key
):
34 """Return the canonical name for a possibly-shortenned
37 candidates
= [cmd
for cmd
in self
.keys() if cmd
.startswith(key
)]
40 out
.error('Unknown command: %s' % key
,
41 'Try "%s help" for a list of supported commands' % prog
)
42 sys
.exit(utils
.STGIT_GENERAL_ERROR
)
43 elif len(candidates
) > 1:
44 out
.error('Ambiguous command: %s' % key
,
45 'Candidates are: %s' %
', '.join(candidates
))
46 sys
.exit(utils
.STGIT_GENERAL_ERROR
)
50 def __getitem__(self
, key
):
51 cmd_mod
= self
.get(key
) or self
.get(self
.canonical_cmd(key
))
52 return stgit
.commands
.get_command(cmd_mod
)
54 cmd_list
= stgit
.commands
.get_commands()
55 commands
= Commands((cmd
, mod
) for cmd
, (mod
, kind
, help)
56 in cmd_list
.iteritems())
59 print 'usage: %s <command> [options]' % os
.path
.basename(sys
.argv
[0])
61 print 'Generic commands:'
62 print ' help print the detailed command usage'
63 print ' version display version information'
64 print ' copyright display copyright information'
66 stgit
.commands
.pretty_command_list(cmd_list
, sys
.stdout
)
69 # The main function (command dispatcher)
76 prog
= os
.path
.basename(sys
.argv
[0])
79 print >> sys
.stderr
, 'usage: %s <command>' % prog
80 print >> sys
.stderr
, \
81 ' Try "%s --help" for a list of supported commands' % prog
82 sys
.exit(utils
.STGIT_GENERAL_ERROR
)
86 if cmd
in ['-h', '--help']:
87 if len(sys
.argv
) >= 3:
88 cmd
= commands
.canonical_cmd(sys
.argv
[2])
89 sys
.argv
[2] = '--help'
92 sys
.exit(utils
.STGIT_SUCCESS
)
94 if len(sys
.argv
) == 3 and not sys
.argv
[2] in ['-h', '--help']:
95 cmd
= commands
.canonical_cmd(sys
.argv
[2])
96 if not cmd
in commands
:
97 out
.error('%s help: "%s" command unknown' %
(prog
, cmd
))
98 sys
.exit(utils
.STGIT_GENERAL_ERROR
)
100 sys
.argv
[0] += ' %s' % cmd
101 command
= commands
[cmd
]
102 parser
= argparse
.make_option_parser(command
)
103 from pydoc
import pager
104 pager(parser
.format_help())
107 sys
.exit(utils
.STGIT_SUCCESS
)
108 if cmd
in ['-v', '--version', 'version']:
109 from stgit
.version
import version
110 print 'Stacked GIT %s' % version
111 os
.system('git --version')
112 print 'Python version %s' % sys
.version
113 sys
.exit(utils
.STGIT_SUCCESS
)
114 if cmd
in ['copyright']:
116 sys
.exit(utils
.STGIT_SUCCESS
)
118 # re-build the command line arguments
119 cmd
= commands
.canonical_cmd(cmd
)
120 sys
.argv
[0] += ' %s' % cmd
123 command
= commands
[cmd
]
124 parser
= argparse
.make_option_parser(command
)
125 options
, args
= parser
.parse_args()
126 directory
= command
.directory
128 # These modules are only used from this point onwards and do not
129 # need to be imported earlier
130 from stgit
.exception
import StgException
131 from stgit
.config
import config_setup
132 from ConfigParser
import ParsingError
, NoSectionError
133 from stgit
.stack
import Series
136 debug_level
= int(os
.environ
.get('STGIT_DEBUG_LEVEL', 0))
138 out
.error('Invalid STGIT_DEBUG_LEVEL environment variable')
139 sys
.exit(utils
.STGIT_GENERAL_ERROR
)
145 # Some commands don't (always) need an initialized series.
146 if directory
.needs_current_series
:
147 if hasattr(options
, 'branch') and options
.branch
:
148 command
.crt_series
= Series(options
.branch
)
150 command
.crt_series
= Series()
152 ret
= command
.func(parser
, options
, args
)
153 except (StgException
, IOError, ParsingError
, NoSectionError
), err
:
154 directory
.write_log(cmd
)
155 out
.error(str(err
), title
= '%s %s' %
(prog
, cmd
))
157 traceback
.print_exc()
158 sys
.exit(utils
.STGIT_COMMAND_ERROR
)
160 # Triggered by the option parser when it finds bad commandline
162 sys
.exit(utils
.STGIT_COMMAND_ERROR
)
163 except KeyboardInterrupt:
164 sys
.exit(utils
.STGIT_GENERAL_ERROR
)
166 out
.error('Unhandled exception:')
167 traceback
.print_exc()
168 sys
.exit(utils
.STGIT_BUG_ERROR
)
170 directory
.write_log(cmd
)
171 sys
.exit(ret
or utils
.STGIT_SUCCESS
)