Mercurial > python-cmd2
annotate example/example.py @ 266:584f970ca66a
permit bare prefix
author | Catherine Devlin <catherine.devlin@gmail.com> |
---|---|
date | Wed, 08 Apr 2009 17:02:13 -0400 |
parents | 5147fa4166b0 |
children | 1cd23003e8d5 |
rev | line source |
---|---|
106 | 1 '''A sample application for cmd2.''' |
2 | |
3 from cmd2 import Cmd, make_option, options, Cmd2TestCase | |
4 import unittest, optparse, sys | |
5 | |
6 class CmdLineApp(Cmd): | |
7 multilineCommands = ['orate'] | |
8 Cmd.shortcuts.update({'&': 'speak'}) | |
9 maxrepeats = 3 | |
10 Cmd.settable.append('maxrepeats') | |
11 | |
12 @options([make_option('-p', '--piglatin', action="store_true", help="atinLay"), | |
13 make_option('-s', '--shout', action="store_true", help="N00B EMULATION MODE"), | |
14 make_option('-r', '--repeat', type="int", help="output [n] times") | |
15 ]) | |
16 def do_speak(self, arg, opts=None): | |
17 """Repeats what you tell me to.""" | |
18 arg = ''.join(arg) | |
19 if opts.piglatin: | |
20 arg = '%s%say' % (arg[1:], arg[0]) | |
21 if opts.shout: | |
22 arg = arg.upper() | |
23 repetitions = opts.repeat or 1 | |
24 for i in range(min(repetitions, self.maxrepeats)): | |
25 self.stdout.write(arg) | |
26 self.stdout.write('\n') | |
27 # self.stdout.write is better than "print", because Cmd can be | |
28 # initialized with a non-standard output destination | |
29 | |
30 do_say = do_speak # now "say" is a synonym for "speak" | |
31 do_orate = do_speak # another synonym, but this one takes multi-line input | |
32 | |
33 class TestMyAppCase(Cmd2TestCase): | |
34 CmdApp = CmdLineApp | |
35 parser = optparse.OptionParser() | |
259
5147fa4166b0
multiple transcript tests working - sinister BASH trick defeated
Catherine Devlin <catherine.devlin@gmail.com>
parents:
258
diff
changeset
|
36 parser.add_option('-t', '--test', dest='test', action="store_true", |
5147fa4166b0
multiple transcript tests working - sinister BASH trick defeated
Catherine Devlin <catherine.devlin@gmail.com>
parents:
258
diff
changeset
|
37 help='Test against transcript(s) in FILE (wildcards OK)') |
106 | 38 (callopts, callargs) = parser.parse_args() |
257
d62bb3dd58a0
multiple test files still not quite working, yet works when run through debugger, aaargh
Catherine Devlin <catherine.devlin@gmail.com>
parents:
254
diff
changeset
|
39 if callopts.test: |
259
5147fa4166b0
multiple transcript tests working - sinister BASH trick defeated
Catherine Devlin <catherine.devlin@gmail.com>
parents:
258
diff
changeset
|
40 CmdLineApp.testfiles = callargs |
257
d62bb3dd58a0
multiple test files still not quite working, yet works when run through debugger, aaargh
Catherine Devlin <catherine.devlin@gmail.com>
parents:
254
diff
changeset
|
41 sys.argv = [sys.argv[0]] # the --test argument upsets unittest.main() |
106 | 42 unittest.main() |
43 else: | |
257
d62bb3dd58a0
multiple test files still not quite working, yet works when run through debugger, aaargh
Catherine Devlin <catherine.devlin@gmail.com>
parents:
254
diff
changeset
|
44 CmdLineApp().cmdloop() |