diff -r 000000000000 -r 9117c6561b0b mercurial/fancyopts.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mercurial/fancyopts.py Tue May 03 13:16:10 2005 -0800 @@ -0,0 +1,51 @@ +import sys, os, getopt + +def fancyopts(args, options, state, syntax=''): + long=[] + short='' + map={} + dt={} + + def help(state, opt, arg, options=options, syntax=syntax): + print "Usage: ", syntax + + for s, l, d, c in options: + opt=' ' + if s: opt = opt + '-' + s + ' ' + if l: opt = opt + '--' + l + ' ' + if d: opt = opt + '(' + str(d) + ')' + print opt + if c: print ' %s' % c + sys.exit(0) + + if len(args) == 0: + help(state, None, args) + + options=[('h', 'help', help, 'Show usage info')] + options + + for s, l, d, c in options: + map['-'+s] = map['--'+l]=l + state[l] = d + dt[l] = type(d) + if not d is None and not type(d) is type(help): s, l=s+':', l+'=' + if s: short = short + s + if l: long.append(l) + + if os.environ.has_key("HG_OPTS"): + args = os.environ["HG_OPTS"].split() + args + + try: + opts, args = getopt.getopt(args, short, long) + except getopt.GetoptError: + help(state, None, args) + sys.exit(-1) + + for opt, arg in opts: + if dt[map[opt]] is type(help): state[map[opt]](state,map[opt],arg) + elif dt[map[opt]] is type(1): state[map[opt]] = int(arg) + elif dt[map[opt]] is type(''): state[map[opt]] = arg + elif dt[map[opt]] is type([]): state[map[opt]].append(arg) + elif dt[map[opt]] is type(None): state[map[opt]] = 1 + + return args +