|
351
|
1 from cmd2 import Cmd, options, make_option
|
|
|
2 # prompts and defaults
|
|
|
3
|
|
|
4 class Pirate(Cmd):
|
|
|
5 gold = 3
|
|
|
6 prompt = 'arrr> '
|
|
|
7 def default(self, line):
|
|
|
8 print('What mean ye by "{0}"?'.format(line))
|
|
|
9 def do_loot(self, arg):
|
|
|
10 'Seize booty from a passing ship.'
|
|
|
11 self.gold += 1
|
|
|
12 def do_drink(self, arg):
|
|
|
13 '''Drown your sorrrows in rrrum.
|
|
|
14
|
|
|
15 drink [n] - drink [n] barrel[s] o' rum.'''
|
|
|
16 try:
|
|
|
17 self.gold -= int(arg)
|
|
|
18 except:
|
|
|
19 if arg:
|
|
|
20 print('''What's "{0}"? I'll take rrrum.'''.format(arg))
|
|
|
21 self.gold -= 1
|
|
|
22 def precmd(self, line):
|
|
|
23 self.initial_gold = self.gold
|
|
|
24 return line
|
|
|
25 def postcmd(self, stop, line):
|
|
|
26 if self.gold != self.initial_gold:
|
|
374
|
27 print('Now we gots {0} doubloons'
|
|
|
28 .format(self.gold))
|
|
351
|
29 if self.gold < 0:
|
|
374
|
30 print("Off to debtorrr's prison.")
|
|
|
31 stop = True
|
|
351
|
32 return stop
|
|
|
33 def do_quit(self, arg):
|
|
|
34 print("Quiterrr!")
|
|
|
35 return True
|
|
|
36 default_to_shell = True
|
|
|
37 multilineCommands = ['sing']
|
|
|
38 terminators = Cmd.terminators + ['...']
|
|
356
|
39 songcolor = 'blue'
|
|
|
40 settable = Cmd.settable + 'songcolor Color to ``sing`` in (red/blue/green/cyan/magenta, bold, underline)'
|
|
|
41 Cmd.shortcuts.update({'~': 'sing'})
|
|
351
|
42 def do_sing(self, arg):
|
|
356
|
43 print(self.colorize(arg, self.songcolor))
|
|
372
|
44 @options([make_option('--ho', type='int', default=2,
|
|
|
45 help="How often to chant 'ho'"),
|
|
|
46 make_option('-c', '--commas',
|
|
|
47 action="store_true",
|
|
|
48 help="Intersperse commas")])
|
|
351
|
49 def do_yo(self, arg, opts):
|
|
|
50 chant = ['yo'] + ['ho'] * opts.ho
|
|
372
|
51 separator = ', ' if opts.commas else ' '
|
|
352
|
52 chant = separator.join(chant)
|
|
372
|
53 print('{0} and a bottle of {1}'
|
|
|
54 .format(chant, arg))
|
|
351
|
55
|
|
|
56 pirate = Pirate()
|
|
372
|
57 pirate.cmdloop()
|
