home · contact · privacy
Initial commit.
[plomrogue2] / plomrogue / tasks.py
1 from plomrogue.errors import PlayError
2 from plomrogue.mapping import YX
3
4
5
6 class Task:
7     argtypes = ''
8     todo = 3
9
10     def __init__(self, thing, args=()):
11         self.thing = thing
12         self.args = args
13
14     def check(self):
15         pass
16
17
18
19 class Task_WAIT(Task):
20     todo = 1
21
22     def do(self):
23         return 'success'
24
25
26
27 class Task_MOVE(Task):
28     todo = 1
29     argtypes = 'string:direction'
30
31     def get_move_target(self):
32         return self.thing.game.map_geometry.move(self.thing.position,
33                                                  self.args[0])
34
35     def check(self):
36         test_pos = self.get_move_target()
37         if test_pos is None:
38             raise PlayError('would move out of map')
39         elif test_pos in [t.position for t in self.thing.game.things]:
40             raise PlayError('would collide with other things')
41         elif self.thing.game.map[test_pos] != '.':
42             raise PlayError('would move into illegal territory')
43
44     def do(self):
45         self.thing.position = self.get_move_target()
46
47
48
49 class Task_WRITE(Task):
50     todo = 1
51     argtypes = 'string:char'
52
53     def check(self):
54         pass
55
56     def do(self):
57         self.thing.game.map[self.thing.position] = self.args[0]
58
59
60
61 class Task_FLATTEN_SURROUNDINGS(Task):
62     todo = 10
63
64     def check(self):
65         pass
66
67     def do(self):
68         self.thing.game.map[self.thing.position] = '.'
69         for yx in self.thing.game.map_geometry.get_neighbors(self.thing.position).values():
70             if yx is not None:
71                 self.thing.game.map[yx] = '.'