diff options
Diffstat (limited to 'examples/pirate.py')
| -rwxr-xr-x | examples/pirate.py | 38 |
1 files changed, 20 insertions, 18 deletions
diff --git a/examples/pirate.py b/examples/pirate.py index 32d7769e..cfe545d6 100755 --- a/examples/pirate.py +++ b/examples/pirate.py @@ -6,7 +6,8 @@ presented as part of her PyCon 2010 talk. It demonstrates many features of cmd2. """ -from cmd2 import Cmd, options, make_option +import argparse +from cmd2 import Cmd, with_argparser class Pirate(Cmd): @@ -24,13 +25,13 @@ class Pirate(Cmd): """Initialize the base class as well as this one""" Cmd.__init__(self) # prompts and defaults - self.gold = 3 + self.gold = 0 self.initial_gold = self.gold self.prompt = 'arrr> ' def default(self, line): """This handles unknown commands.""" - print('What mean ye by "{0}"?'.format(line)) + self.poutput('What mean ye by "{0}"?'.format(line)) def precmd(self, line): """Runs just before a command line is parsed, but after the prompt is presented.""" @@ -40,10 +41,10 @@ class Pirate(Cmd): def postcmd(self, stop, line): """Runs right before a command is about to return.""" if self.gold != self.initial_gold: - print('Now we gots {0} doubloons' + self.poutput('Now we gots {0} doubloons' .format(self.gold)) if self.gold < 0: - print("Off to debtorrr's prison.") + self.poutput("Off to debtorrr's prison.") stop = True return stop @@ -60,29 +61,30 @@ class Pirate(Cmd): self.gold -= int(arg) except ValueError: if arg: - print('''What's "{0}"? I'll take rrrum.'''.format(arg)) + self.poutput('''What's "{0}"? I'll take rrrum.'''.format(arg)) self.gold -= 1 def do_quit(self, arg): """Quit the application gracefully.""" - print("Quiterrr!") + self.poutput("Quiterrr!") return True def do_sing(self, arg): """Sing a colorful song.""" - print(self.colorize(arg, self.songcolor)) - - @options([make_option('--ho', type='int', default=2, - help="How often to chant 'ho'"), - make_option('-c', '--commas', - action="store_true", - help="Intersperse commas")]) - def do_yo(self, arg, opts): + self.poutput(self.colorize(arg, self.songcolor)) + + yo_parser = argparse.ArgumentParser() + yo_parser.add_argument('--ho', type=int, default=2, help="How often to chant 'ho'") + yo_parser.add_argument('-c', '--commas', action='store_true', help='Intersperse commas') + yo_parser.add_argument('beverage', help='beverage to drink with the chant') + + @with_argparser(yo_parser) + def do_yo(self, args): """Compose a yo-ho-ho type chant with flexible options.""" - chant = ['yo'] + ['ho'] * opts.ho - separator = ', ' if opts.commas else ' ' + chant = ['yo'] + ['ho'] * args.ho + separator = ', ' if args.commas else ' ' chant = separator.join(chant) - print('{0} and a bottle of {1}'.format(chant, arg)) + self.poutput('{0} and a bottle of {1}'.format(chant, args.beverage)) if __name__ == '__main__': |