summaryrefslogtreecommitdiff
path: root/docs/pycon2010/pirate8.py
blob: 58b56208365d17c95680f093ae7e591c3317bae2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
# coding=utf-8
from cmd2 import Cmd, options, make_option


# prompts and defaults

class Pirate(Cmd):
    gold = 3
    initial_gold = gold
    prompt = 'arrr> '

    def default(self, line):
        print('What mean ye by "{0}"?'.format(line))

    def do_loot(self, arg):
        'Seize booty from a passing ship.'
        self.gold += 1

    def do_drink(self, arg):
        '''Drown your sorrrows in rrrum.

        drink [n] - drink [n] barrel[s] o' rum.'''
        try:
            self.gold -= int(arg)
        except:
            if arg:
                print('''What's "{0}"?  I'll take rrrum.'''.format(arg))
            self.gold -= 1

    def precmd(self, line):
        self.initial_gold = self.gold
        return line

    def postcmd(self, stop, line):
        if self.gold != self.initial_gold:
            print('Now we gots {0} doubloons'
                  .format(self.gold))
        if self.gold < 0:
            print("Off to debtorrr's prison.")
            stop = True
        return stop

    def do_quit(self, arg):
        print("Quiterrr!")
        return True

    default_to_shell = True
    multilineCommands = ['sing']
    terminators = Cmd.terminators + ['...']
    songcolor = 'blue'
    settable = Cmd.settable + 'songcolor Color to ``sing`` in (red/blue/green/cyan/magenta, bold, underline)'
    Cmd.shortcuts.update({'~': 'sing'})

    def do_sing(self, arg):
        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):
        chant = ['yo'] + ['ho'] * opts.ho
        separator = ', ' if opts.commas else ' '
        chant = separator.join(chant)
        print('{0} and a bottle of {1}'
              .format(chant, arg))


pirate = Pirate()
pirate.cmdloop()