24 lines
605 B
Python
24 lines
605 B
Python
|
from domain.entities.entity import Entity
|
||
|
|
from domain.world import World
|
||
|
|
|
||
|
|
|
||
|
|
class Vacuum(Entity):
|
||
|
|
def __init__(self, x: int, y: int, world: World):
|
||
|
|
super().__init__(x, y, 'VACUUM')
|
||
|
|
self.world = world
|
||
|
|
self.battery = 100
|
||
|
|
# TODO add more properties
|
||
|
|
|
||
|
|
def move(self, dx, dy):
|
||
|
|
end_x = self.x + dx
|
||
|
|
end_y = self.y + dy
|
||
|
|
|
||
|
|
if end_x > self.world.width - 1 or end_y > self.world.height - 1 or end_x < 0 or end_y < 0:
|
||
|
|
return
|
||
|
|
|
||
|
|
if self.world.is_obstacle_at(end_x, end_y):
|
||
|
|
return
|
||
|
|
|
||
|
|
self.x = end_x
|
||
|
|
self.y = end_y
|