Mercurial > python-cmd2
annotate example/example.py @ 325:4172feeddf76
want to incorporate run() for tests - not yet working
| author | catherine@dellzilla |
|---|---|
| date | Thu, 11 Feb 2010 17:03:45 -0500 |
| parents | 9d91406ca3a7 |
| children | 237a89d5a4a9 |
| rev | line source |
|---|---|
| 106 | 1 '''A sample application for cmd2.''' |
| 2 | |
|
325
4172feeddf76
want to incorporate run() for tests - not yet working
catherine@dellzilla
parents:
310
diff
changeset
|
3 from cmd2 import Cmd, make_option, options, Cmd2TestCase, run |
| 106 | 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() |
|
325
4172feeddf76
want to incorporate run() for tests - not yet working
catherine@dellzilla
parents:
310
diff
changeset
|
34 #run(c) |
|
287
1cd23003e8d5
refactoring, but something went wrong with comments
catherine@bothari
parents:
259
diff
changeset
|
35 |
| 106 | 36 class TestMyAppCase(Cmd2TestCase): |
| 37 CmdApp = CmdLineApp | |
| 38 parser = optparse.OptionParser() | |
|
259
5147fa4166b0
multiple transcript tests working - sinister BASH trick defeated
Catherine Devlin <catherine.devlin@gmail.com>
parents:
258
diff
changeset
|
39 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
|
40 help='Test against transcript(s) in FILE (wildcards OK)') |
| 106 | 41 (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
|
42 if callopts.test: |
|
259
5147fa4166b0
multiple transcript tests working - sinister BASH trick defeated
Catherine Devlin <catherine.devlin@gmail.com>
parents:
258
diff
changeset
|
43 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
|
44 sys.argv = [sys.argv[0]] # the --test argument upsets unittest.main() |
| 106 | 45 unittest.main() |
| 46 else: | |
| 290 | 47 CmdLineApp().cmdloop() |