def after_move(self):
"""After moving onto fire, keep moving in the same direction if possible."""
if (self.get_cell_type() == 'Fire'
and self.level.sprite_can_enter(self.get_pos_in_dir(self.move_dir))):
self.to_move = 1
Sprite.after_move(self)
def after_move(self):
"""Pick up items in this cell, and start adjacent Rovers following."""
for item in self.pushing:
item.after_move()
self.pushing.clear()
for item in self.level.sprites.item.on(self.pos):
if item not in self.carrying:
self.carrying.add(item)
for sprite in self.level.sprites.on(*[self.get_pos_in_dir(direction) for direction in range(4)]):
if sprite.type == "Rover" and sprite not in self.followers:
self.followers.add(sprite)
Sprite.after_move(self)
