Mercurial > python-cmd2
annotate example/example.py @ 319:c58cd7e48db7
begin to switch settable to TextLineList
author | catherine@dellzilla |
---|---|
date | Thu, 11 Feb 2010 13:07:05 -0500 |
parents | 9d91406ca3a7 |
children | 4172feeddf76 |
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 | |
291 | 10 Cmd.settable.append('maxrepeats Max number of `--repeat`s allowed') |
106 | 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: | |
301 | 20 arg = '%s%say' % (arg[1:].rstrip(), arg[0]) |
106 | 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 | |
310 | 28 # initialized with a non-standard output destination |
106 | 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 | |
287
1cd23003e8d5
refactoring, but something went wrong with comments
catherine@bothari
parents:
259
diff
changeset
|
33 c = CmdLineApp() |
1cd23003e8d5
refactoring, but something went wrong with comments
catherine@bothari
parents:
259
diff
changeset
|
34 |
106 | 35 class TestMyAppCase(Cmd2TestCase): |
36 CmdApp = CmdLineApp | |
37 parser = optparse.OptionParser() | |
259
5147fa4166b0
multiple transcript tests working - sinister BASH trick defeated
Catherine Devlin <catherine.devlin@gmail.com>
parents:
258
diff
changeset
|
38 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
|
39 help='Test against transcript(s) in FILE (wildcards OK)') |
106 | 40 (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
|
41 if callopts.test: |
259
5147fa4166b0
multiple transcript tests working - sinister BASH trick defeated
Catherine Devlin <catherine.devlin@gmail.com>
parents:
258
diff
changeset
|
42 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
|
43 sys.argv = [sys.argv[0]] # the --test argument upsets unittest.main() |
106 | 44 unittest.main() |
45 else: | |
290 | 46 CmdLineApp().cmdloop() |
287
1cd23003e8d5
refactoring, but something went wrong with comments
catherine@bothari
parents:
259
diff
changeset
|
47 |