plomlompom.com Git - plomrogue2/blob - plomrogue/tasks.py

   1 from plomrogue.errors import PlayError, GameError
   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 string'
  52
  53     def check(self):
  54         tile_class = self.thing.game.map_control[self.thing.position]
  55         if tile_class in self.thing.game.map_control_passwords:
  56             tile_pw = self.thing.game.map_control_passwords[tile_class]
  57             if self.args[1] != tile_pw:
  58                 raise GameError('wrong password for tile')
  59
  60     def do(self):
  61         self.thing.game.map[self.thing.position] = self.args[0]
  62
  63
  64
  65 class Task_FLATTEN_SURROUNDINGS(Task):
  66     todo = 10
  67
  68     def check(self):
  69         pass
  70
  71     def do(self):
  72         self.thing.game.map[self.thing.position] = '.'
  73         for yx in self.thing.game.map_geometry.get_neighbors(self.thing.position).values():
  74             if yx is not None:
  75                 self.thing.game.map[yx] = '.'