X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=koperkapel%2Fgamelib%2Flevel.py;h=aed4cccb880b88b48cd33ba33e6afe0abcba9663;hb=e8bb29ef4dd7789bf9c20aebfb87f3911bb0d7b9;hp=bcc319a2886d2f5e530d30ff70be6f7f50478a09;hpb=5729b7d89f992ba3d5b3f779ecc1094a48a85459;p=koperkapel.git

diff --git a/koperkapel/gamelib/level.py b/koperkapel/gamelib/level.py
index bcc319a..aed4ccc 100644
--- a/koperkapel/gamelib/level.py
+++ b/koperkapel/gamelib/level.py
@@ -1,5 +1,7 @@
 """ Class holding the level info """
 
+import random
+
 
 class Level(object):
 
@@ -84,3 +86,32 @@ class Level(object):
 
     def get_exit_level(self):
         return self.exit["next level"]
+
+    def enemy_at(self, x, y):
+        for enemy in self.enemies:
+            if (x, y) == enemy.game_pos:
+                return True
+        return False
+
+    def get_enemy(self, x, y):
+        for enemy in self.enemies:
+            if (x, y) == enemy.game_pos:
+                return enemy
+        return None
+
+    def remove_enemy(self, enemy):
+        self.enemies.remove(enemy)
+
+    def add_enemy(self, enemy, x, y):
+        """Add an enemy to an empty floor space near x, y"""
+        added = False
+        while not added:
+            if self.can_walk(x, y, 'floor'):
+                if not self.friend_at(x, y):
+                    if not self.enemy_at(x, y):
+                        added = True
+                        enemy.game_pos = (x, y)
+                        self.enemies.append(enemy)
+                        added = True
+            x += random.randint(-1, 1)
+            y += random.randint(-1, 1)