home · contact · privacy
Refactor parser code.
[plomrogue2] / plomrogue / tasks.py
1 from plomrogue.errors import PlayError, GameError
2 from plomrogue.misc import quote
3
4
5
6 class Task:
7     argtypes = ''
8     todo = 1
9
10     def __init__(self, thing, args=()):
11         self.thing = thing
12         self.args = args
13
14     def _get_move_target(self):
15         if self.args[0] == 'HERE':
16             return self.thing.position
17         return self.thing.game.map_geometry.move_yxyx(self.thing.position,
18                                                       self.args[0])
19
20     def check(self):
21         pass
22
23
24
25 class Task_WAIT(Task):
26
27     def do(self):
28         pass
29
30
31
32 class Task_MOVE(Task):
33     argtypes = 'string:direction'
34
35     def check(self):
36         test_yxyx = self._get_move_target()
37         move_blockers = self.thing.game.get_movement_blockers()
38         if test_yxyx in [t.position for t in self.thing.game.things
39                          if t.blocks_movement]:
40             raise PlayError('blocked by other thing')
41         elif self.thing.game.maps[test_yxyx[0]][test_yxyx[1]] in move_blockers:
42             raise PlayError('blocked by impassable tile')
43
44     def do(self):
45         if self.thing.type_ == 'Player' and not self.thing.standing:
46             self.thing.standing = True
47             self.thing.send_msg('CHAT "You get up."')
48         self.thing.game.record_change(self.thing.position, 'other')
49         if self.thing.blocks_light:
50             self.thing.game.record_change(self.thing.position, 'fov')
51         self.thing.position = self._get_move_target()
52         for t in [t for t in self.thing.game.things
53                   if t.type_ == 'Player' and not t == self.thing
54                   and t.position == self.thing.position]:
55             self.thing.send_msg('CHAT %s' %
56                                 quote('You get awkwardly close to %s.' % t.name))
57             for c_id in self.thing.game.sessions:
58                 if self.thing.game.sessions[c_id]['thing_id'] == t.id_:
59                     t.send_msg('CHAT %s' %
60                                quote('%s gets awkwardly close to you.' %
61                                      self.thing.name))
62                     break
63         self.thing.game.record_change(self.thing.position, 'other')
64         self.thing.try_to_sit()
65         self.thing.invalidate('fov')
66         if self.thing.blocks_light:
67             self.thing.game.record_change(self.thing.position, 'fov')
68         if self.thing.carrying:
69             self.thing.carrying.position = self.thing.position
70             if self.thing.carrying.type_ == 'Crate':
71                 for t in self.thing.carrying.content:
72                     t.position = self.thing.position
73
74
75
76 class Task_WRITE(Task):
77     argtypes = 'string:char string'
78
79     def check(self):
80         if not self.thing.game.can_do_tile_with_pw(*self.thing.position,
81                                                    self.args[1]):
82             raise GameError('wrong password for tile')
83
84     def do(self):
85         big_yx = self.thing.position[0]
86         little_yx = self.thing.position[1]
87         self.thing.game.maps[big_yx][little_yx] = self.args[0]
88         self.thing.game.record_change((big_yx, little_yx), 'fov')
89
90
91
92 class Task_FLATTEN_SURROUNDINGS(Task):
93     argtypes = 'string'
94
95     def check(self):
96         pass
97
98     def do(self):
99         for yxyx in [self.thing.position] + \
100                 list(self.thing.game.map_geometry.get_neighbors_yxyx(
101                     self.thing.position).values()):
102             if not self.thing.game.can_do_tile_with_pw(*yxyx, self.args[0]):
103                 continue
104             self.thing.game.maps[yxyx[0]][yxyx[1]] = self.thing.game.get_flatland()
105             self.thing.game.record_change(yxyx, 'fov')
106
107
108
109 class Task_PICK_UP(Task):
110     argtypes = 'int:pos'
111
112     def check(self):
113         if self.thing.carrying:
114             raise PlayError('already carrying something')
115         to_pick_up = self.thing.game.get_thing(self.args[0])
116         neighbors = self.thing.game.map_geometry.\
117             get_neighbors_yxyx(self.thing.position).values()
118         reach = [self.thing.position] + list(neighbors)
119         if to_pick_up is None:
120             raise PlayError('no such thing exists')
121         elif to_pick_up == self.thing:
122             raise PlayError('cannot pick up oneself')
123         elif to_pick_up.type_ == 'Player':
124             raise PlayError('cannot pick up player')
125         elif to_pick_up.carried:
126             raise PlayError('thing already carried by a player')
127         elif to_pick_up.position not in reach:
128             raise PlayError('thing not in reach')
129         elif not to_pick_up.portable:
130             raise PlayError('thing not portable')
131
132     def do(self):
133         to_pick_up = self.thing.game.get_thing(self.args[0])
134         to_pick_up.position = self.thing.position[:]
135         self.thing.carrying = to_pick_up
136         to_pick_up.carried = True
137         for t in [t for t in self.thing.game.things
138                   if t.type_ == 'Crate' and to_pick_up in t.content]:
139             t.remove_from_crate(to_pick_up)
140             self.thing.send_msg('CHAT "You take the item out of the crate."')
141             break
142         self.thing.game.record_change(self.thing.position, 'other')
143
144
145
146 class Task_DROP(Task):
147     argtypes = 'string:direction+here'
148
149     def check(self):
150         if not self.thing.carrying:
151             raise PlayError('nothing to drop')
152         target_position = self._get_move_target()
153         targets = [t for t in self.thing.game.things
154                    if t.position == target_position
155                    and not t == self.thing.carrying]
156         for target in targets:
157             if target.type_ == 'CookieSpawner' and\
158                not self.thing.carrying.cookable:
159                 raise PlayError('cannot cook items of this type')
160             elif target.type_ == 'BottleDeposit':
161                 if not self.thing.carrying.type_ == 'Bottle':
162                     raise PlayError('cannot only put bottle into bottle deposit')
163                 if self.thing.carrying.full:
164                     raise PlayError('cannot drop full '
165                                     'bottle into bottle deposit')
166             elif target.type_ == 'Crate' and\
167                  self.thing.carrying.type_ == 'Crate':
168                 raise PlayError('cannot put crate into crate')
169
170     def do(self):
171         target_position = self._get_move_target()
172         dropped = self.thing.uncarry()
173         dropped.position = target_position
174         if dropped.type_ == 'Crate':
175             for item in dropped.content:
176                 item.position = target_position
177         targets = [t for t in self.thing.game.things
178                    if t.position == dropped.position and not t == dropped]
179         for target in targets:
180             if target.type_ == 'CookieSpawner':
181                 target.accept(dropped)
182                 self.thing.game.remove_thing(dropped)
183                 break
184             elif target.type_ == 'BottleDeposit':
185                 target.accept()
186                 self.thing.game.remove_thing(dropped)
187                 break
188             elif target.type_ == 'Crate':
189                 target.accept(dropped)
190                 self.thing.send_msg('CHAT "You put the item into the crate."')
191                 break
192             elif target.type_ == 'HatRemixer':
193                 t.accept(dropped)
194                 break
195         self.thing.game.record_change(self.thing.position, 'other')
196
197
198
199 class Task_DOOR(Task):
200
201     def check(self):
202         action_radius = list(self.thing.game.map_geometry.
203                              get_neighbors_yxyx(self.thing.position).values())
204         reachable_doors = [t for t in self.thing.game.things if
205                            t.type_ == 'Door' and t.position in action_radius]
206         if len(reachable_doors) == 0:
207             raise PlayError('not standing next to a door to open/close')
208         if len([d for d in reachable_doors if not d.portable]) == 0:
209             raise PlayError('can only open/close doors that are installed')
210         for door in reachable_doors:
211             if not door.blocks_movement:
212                 return
213             if not door.locked:
214                 return
215             if self.thing.carrying and self.thing.carrying.type_ == 'DoorKey'\
216                and self.thing.carrying.door == door:
217                 return
218         raise PlayError('cannot open locked door without its key')
219
220     def do(self):
221         action_radius = list(self.thing.game.map_geometry.
222                              get_neighbors_yxyx(self.thing.position).values())
223         for t in [t for t in self.thing.game.things if
224                   t.type_ == 'Door' and t.position in action_radius]:
225             if t.blocks_movement:
226                 t.open()
227             else:
228                 t.close()
229                 if self.thing.carrying and\
230                    self.thing.carrying.type_ == 'DoorKey' and\
231                    self.thing.carrying.door == t:
232                     self.thing.send_msg('CHAT "You lock the door."')
233                     t.lock()
234             self.thing.game.record_change(t.position, 'other')
235             self.thing.game.record_change(t.position, 'fov')
236
237
238
239 class Task_INTOXICATE(Task):
240
241     def check(self):
242         if self.thing.carrying is None:
243             raise PlayError('carrying nothing to consume')
244         if not self.thing.carrying.consumable:
245             raise PlayError('cannot consume this kind of thing')
246         if self.thing.carrying.type_ == 'Bottle' and\
247            not self.thing.carrying.full:
248             raise PlayError('bottle is empty')
249
250     def do(self):
251         if self.thing.carrying.type_ == 'Bottle':
252             self.thing.carrying.full = False
253             self.thing.carrying.empty()
254             self.thing.send_msg('CHAT "You are drunk now."')
255             self.thing.need_for_toilet += 1
256             self.thing.drunk += 100000
257             self.thing.invalidate('fov')
258             self.thing.game.record_change(self.thing.position, 'other')
259         elif self.thing.carrying.type_ == 'Psychedelic':
260             self.thing.tripping += 100000
261             self.thing.send_msg('CHAT "You start tripping."')
262             self.thing.send_msg('RANDOM_COLORS')
263             eaten = self.thing.uncarry()
264             self.thing.game.remove_thing(eaten)
265         elif self.thing.carrying.type_ == 'Cookie':
266             self.thing.send_msg('CHAT ' + quote('You eat a cookie that grants the ability to draw the following character: "%s"' % self.thing.carrying.thing_char))
267             self.thing.add_cookie_char(self.thing.carrying.thing_char)
268             eaten = self.thing.uncarry()
269             self.thing.game.remove_thing(eaten)
270         elif self.thing.carrying.type_ == 'Stimulant':
271             self.thing.send_msg('CHAT "You feel a flash of energy."')
272             self.thing.energy += 50
273             eaten = self.thing.uncarry()
274             self.thing.game.remove_thing(eaten)
275
276
277
278 class Task_COMMAND(Task):
279     argtypes = 'string'
280
281     def check(self):
282         if self.thing.carrying is None:
283             raise PlayError('nothing to command')
284         if not self.thing.carrying.commandable:
285             raise PlayError('cannot command this item type')
286
287     def do(self):
288         reply_lines = self.thing.carrying.interpret(self.args[0])
289         for line in reply_lines:
290             self.thing.send_msg('REPLY ' + quote(line))
291
292
293
294 class Task_INSTALL(Task):
295     argtypes = 'string'
296
297     def _get_uninstallables(self):
298         return [t for t in self.thing.game.things
299                 if t != self.thing
300                 and hasattr(t, 'installable') and t.installable
301                 and (not t.portable)
302                 and t.position == self.thing.position]
303
304     def check(self):
305         if not self.thing.game.can_do_tile_with_pw(*self.thing.position,
306                                                    self.args[0]):
307             raise GameError('wrong password for tile')
308         if self.thing.carrying:
309             if not hasattr(self.thing.carrying, 'installable')\
310                or not self.thing.carrying.installable:
311                 raise PlayError('carried thing not installable')
312         elif len(self._get_uninstallables()) == 0:
313             raise PlayError('nothing to uninstall here')
314
315     def do(self):
316         if self.thing.carrying:
317             t = self.thing.uncarry()
318             t.install()
319             self.thing.send_msg('CHAT "You install the thing you carry."')
320         else:
321             self._get_uninstallables()[0].uninstall()
322             self.thing.send_msg('CHAT "You uninstall the thing here."')
323         self.thing.game.record_change(self.thing.position, 'other')
324
325
326
327 class Task_WEAR(Task):
328
329     def check(self):
330         if self.thing.name in self.thing.game.hats:
331             return
332         if not self.thing.carrying:
333             raise PlayError('carrying nothing to wear')
334         if self.thing.name in self.thing.game.hats:
335             raise PlayError('already wearing a hat')
336         if self.thing.carrying.type_ not in {'Hat', 'Bottle'}:
337             raise PlayError('can not wear the kind of thing you hold')
338
339     def do(self):
340         if self.thing.name in self.thing.game.hats:
341             t = self.thing.game.add_thing('Hat', self.thing.position)
342             t.design = self.thing.game.hats[self.thing.name]
343             del self.thing.game.hats[self.thing.name]
344             self.thing.send_msg('CHAT "You drop your hat."')
345             for remixer in [t for t in self.thing.game.things
346                             if t.type_ == 'HatRemixer'
347                             and t.position == self.thing.position]:
348                 remixer.accept(t)
349                 break
350         else:
351             if self.thing.carrying.type_ == 'Bottle':
352                 self.thing.send_msg('CHAT "Your attempt to wear a bottle on '
353                                     'your head fails."')
354                 self.thing.carrying.sound('BOTTLE', 'SMASH')
355             elif self.thing.carrying.type_ == 'Hat':
356                 self.thing.game.hats[self.thing.name] =\
357                     self.thing.carrying.design
358                 self.thing.send_msg('CHAT "You put on a hat."')
359             dropped = self.thing.uncarry()
360             self.thing.game.remove_thing(dropped)
361         self.thing.game.record_change(self.thing.position, 'other')
362
363
364
365 class Task_SPIN(Task):
366
367     def check(self):
368         if not self.thing.carrying:
369             raise PlayError('holding nothing to spin')
370         if not hasattr(self.thing.carrying, 'spinnable'):
371             raise PlayError('held object not spinnable')
372
373     def do(self):
374         self.thing.carrying.spin()
375         self.thing.send_msg('CHAT "You spin this object."')
376
377
378
379 class Task_DANCE(Task):
380
381     def do(self):
382         self.thing.send_msg('CHAT "You dance."')
383         self.thing.dancing += 10
384         self.thing.game.changed = True