home · contact · privacy
edfc3d10efddbd1fc4a9d95031b3fd8ea4101aca
[plomrogue2] / plomrogue / tasks.py
1 from plomrogue.errors import PlayError, GameError
2
3
4
5 class Task:
6     argtypes = ''
7     todo = 1
8
9     def __init__(self, thing, args=()):
10         self.thing = thing
11         self.args = args
12
13     def check(self):
14         pass
15
16
17
18 class Task_WAIT(Task):
19
20     def do(self):
21         pass
22
23
24
25 class Task_MOVE(Task):
26     argtypes = 'string:direction'
27
28     def get_move_target(self):
29         return self.thing.game.map_geometry.move_yxyx(self.thing.position,
30                                                       self.args[0])
31
32     def check(self):
33         test_yxyx = self.get_move_target()
34         if test_yxyx in [t.position for t in self.thing.game.things
35                          if t.blocking]:
36             raise PlayError('blocked by other thing')
37         elif self.thing.game.maps[test_yxyx[0]][test_yxyx[1]] != '.':
38             raise PlayError('blocked by impassable tile')
39
40     def do(self):
41         self.thing.position = self.get_move_target()
42         if self.thing.carrying:
43             self.thing.carrying.position = self.thing.position
44         self.thing.game.changed_fovs = True
45
46
47
48 class Task_WRITE(Task):
49     argtypes = 'string:char string'
50
51     def check(self):
52         if not self.thing.game.can_do_tile_with_pw(*self.thing.position,
53                                                    self.args[1]):
54             raise GameError('wrong password for tile')
55
56     def do(self):
57         big_yx = self.thing.position[0]
58         little_yx = self.thing.position[1]
59         self.thing.game.maps[big_yx][little_yx] = self.args[0]
60         self.thing.game.changed_fovs = True
61
62
63
64 class Task_FLATTEN_SURROUNDINGS(Task):
65     argtypes = 'string'
66
67     def check(self):
68         pass
69
70     def do(self):
71         for yxyx in [self.thing.position] + \
72                 list(self.thing.game.map_geometry.get_neighbors_yxyx(
73                     self.thing.position).values()):
74             if not self.thing.game.can_do_tile_with_pw(*yxyx, self.args[0]):
75                 continue
76             self.thing.game.maps[yxyx[0]][yxyx[1]] = '.'
77         self.thing.game.changed_fovs = True
78
79
80
81 class Task_PICK_UP(Task):
82     argtypes = 'int:pos'
83
84     def check(self):
85         if self.thing.carrying:
86             raise PlayError('already carrying something')
87         to_pick_up = self.thing.game.get_thing(self.args[0])
88         neighbors = self.thing.game.map_geometry.\
89             get_neighbors_yxyx(self.thing.position).values()
90         reach = [self.thing.position] + list(neighbors)
91         if to_pick_up is None:
92             raise PlayError('no such thing exists')
93         elif to_pick_up == self.thing:
94             raise PlayError('cannot pick up oneself')
95         elif to_pick_up.type_ == 'Player':
96             raise PlayError('cannot pick up player')
97         elif to_pick_up.position not in reach:
98             raise PlayError('thing not in reach')
99         elif not to_pick_up.portable:
100             raise PlayError('thing not portable')
101
102     def do(self):
103         to_pick_up = self.thing.game.get_thing(self.args[0])
104         to_pick_up.position = self.thing.position[:]
105         self.thing.carrying = to_pick_up
106         #self.thing.game.changed_fovs = True
107
108
109
110 class Task_DROP(Task):
111
112     def check(self):
113         if not self.thing.carrying:
114             raise PlayError('nothing to drop')
115         if self.thing.carrying.type_ == 'Bottle' and self.thing.carrying.full:
116             for t in [t for t in self.thing.game.things
117                       if t.type_ == 'BottleDeposit'
118                       and t.position == self.thing.position]:
119                 raise PlayError('cannot drop full bottle into bottle deposit')
120
121     def do(self):
122         if self.thing.carrying.type_ == 'Bottle' and not self.thing.carrying.full:
123             for t in [t for t in self.thing.game.things
124                       if t.type_ == 'BottleDeposit'
125                       and t.position == self.thing.position]:
126                 t.accept()
127                 self.thing.game.things.remove(self.thing.carrying)
128                 break
129         elif self.thing.carrying.type_ == 'Hat':
130             for t in [t for t in self.thing.game.things
131                       if t.type_ == 'HatRemixer'
132                       and t.position == self.thing.position]:
133                 t.accept(self.thing.carrying)
134                 break
135         self.thing.carrying = None
136         #self.thing.game.changed_fovs = True
137
138
139
140 class Task_DOOR(Task):
141
142     def do(self):
143         self.thing.carrying = None
144         action_radius = list(self.thing.game.map_geometry.
145                              get_neighbors_yxyx(self.thing.position).values())
146         for t in [t for t in self.thing.game.things if
147                   t.type_ == 'Door' and t.position in action_radius]:
148             if t.blocking:
149                 t.open()
150             else:
151                 t.close()
152         self.thing.game.changed_fovs = True
153
154
155
156 class Task_INTOXICATE(Task):
157
158     def check(self):
159         if self.thing.carrying is None:
160             raise PlayError('carrying nothing to drink from')
161         if self.thing.carrying.type_ != 'Bottle':
162             raise PlayError('cannot drink from non-bottle')
163         if not self.thing.carrying.full:
164             raise PlayError('bottle is empty')
165
166     def do(self):
167         self.thing.carrying.full = False
168         self.thing.carrying.empty()
169         self.thing.send_msg('RANDOM_COLORS')
170         self.thing.send_msg('CHAT "You are drunk now."')
171         self.thing.drunk = 10000
172         self.thing.game.changed_fovs = True
173
174
175
176 class Task_COMMAND(Task):
177     argtypes = 'string'
178
179     def check(self):
180         if self.thing.carrying is None:
181             raise PlayError('nothing to command')
182         if not self.thing.carrying.commandable:
183             raise PlayError('cannot command this item type')
184
185     def do(self):
186         from plomrogue.misc import quote
187         reply_lines = self.thing.carrying.interpret(self.args[0])
188         for line in reply_lines:
189             self.thing.send_msg('REPLY ' + quote(line))
190
191
192
193 class Task_INSTALL(Task):
194
195     def _get_uninstallables(self):
196         return [t for t in self.thing.game.things
197                 if t != self.thing
198                 and hasattr(t, 'installable') and t.installable
199                 and (not t.portable)
200                 and t.position == self.thing.position]
201
202     def check(self):
203         if self.thing.carrying:
204             if not hasattr(self.thing.carrying, 'installable')\
205                or not self.thing.carrying.installable:
206                 raise PlayError('carried thing not installable')
207         elif len(self._get_uninstallables()) == 0:
208             raise PlayError('nothing to uninstall here')
209
210     def do(self):
211         if self.thing.carrying:
212             self.thing.carrying.install()
213             self.thing.carrying = None
214             self.thing.send_msg('CHAT "You install the thing you carry."')
215         else:
216             self._get_uninstallables()[0].uninstall()
217             self.thing.send_msg('CHAT "You uninstall the thing here."')
218
219
220
221 class Task_WEAR(Task):
222
223     def check(self):
224         if self.thing.name in self.thing.game.hats:
225             return
226         if not self.thing.carrying:
227             raise PlayError('carrying nothing to wear')
228         if self.thing.name in self.thing.game.hats:
229             raise PlayError('already wearing a hat')
230         if self.thing.carrying.type_ != 'Hat':
231             raise PlayError('can only wear a hat')
232
233     def do(self):
234         if self.thing.name in self.thing.game.hats:
235             t = self.thing.game.thing_types['Hat'](self.thing.game,
236                                                    position=self.thing.position)
237             self.thing.game.things += [t]
238             t.design = self.thing.game.hats[self.thing.name]
239             del self.thing.game.hats[self.thing.name]
240             self.thing.send_msg('CHAT "You drop your hat."')
241             for remixer in [t for t in self.thing.game.things
242                             if t.type_ == 'HatRemixer'
243                             and t.position == self.thing.position]:
244                 remixer.accept(t)
245                 break
246         else:
247             from plomrogue.misc import quote
248             self.thing.game.hats[self.thing.name] = self.thing.carrying.design
249             self.thing.game.things.remove(self.thing.carrying)
250             self.thing.carrying = None
251             self.thing.send_msg('CHAT "You put on a hat."')